router: add ConnCount (#156)

pingcap · Dec 22, 2022 · 8aee2ce · 8aee2ce
1 parent 6c986bb
commit 8aee2ce
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/pkg/manager/router/router.go b/pkg/manager/router/router.go
@@ -30,9 +30,12 @@ import (
 type Router interface {
 	Route(RedirectableConn) (string, error)
 	RedirectConnections() error
+	ConnCount() int
 	Close()
 }
 
+var _ Router = &ScoreBasedRouter{}
+
 var (
 	ErrNoInstanceToSelect = errors.New("no instances to route")
 )
@@ -416,6 +419,17 @@ func (router *ScoreBasedRouter) removeBackendIfEmpty(be *list.Element) bool {
 	return false
 }
 
+func (router *ScoreBasedRouter) ConnCount() int {
+	router.Lock()
+	defer router.Unlock()
+	j := 0
+	for be := router.backends.Front(); be != nil; be = be.Next() {
+		backend := be.Value.(*backendWrapper)
+		j += backend.connList.Len()
+	}
+	return j
+}
+
 // Close implements Router.Close interface.
 func (router *ScoreBasedRouter) Close() {
 	router.Lock()