router: add RefreshBackend method for Router interface (#287)

pingcap · May 23, 2023 · 74113ab · 74113ab
1 parent 2f5bfd8
commit 74113ab
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/pkg/manager/router/router.go b/pkg/manager/router/router.go
@@ -38,6 +38,7 @@ type Router interface {
 	ConnEventReceiver
 
 	GetBackendSelector() BackendSelector
+	RefreshBackend()
 	RedirectConnections() error
 	ConnCount() int
 	// ServerVersion returns the TiDB version.

diff --git a/pkg/manager/router/router_score.go b/pkg/manager/router/router_score.go
@@ -186,6 +186,11 @@ func (router *ScoreBasedRouter) adjustBackendList(be *glist.Element[*backendWrap
 	}
 }
 
+// RefreshBackend implements Router.GetBackendSelector interface.
+func (router *ScoreBasedRouter) RefreshBackend() {
+	router.observer.Refresh()
+}
+
 // RedirectConnections implements Router.RedirectConnections interface.
 // It redirects all connections compulsively. It's only used for testing.
 func (router *ScoreBasedRouter) RedirectConnections() error {

diff --git a/pkg/manager/router/router_static.go b/pkg/manager/router/router_static.go
@@ -50,6 +50,8 @@ func (r *StaticRouter) GetBackendSelector() BackendSelector {
 	}
 }
 
+func (r *StaticRouter) RefreshBackend() {}
+
 func (r *StaticRouter) RedirectConnections() error {
 	return nil
 }