Added getting statistics of all connection pools

Author: nvorobev

CHANGELOG.md

@@ -1,3 +1,7 @@
+## v2.1.0 (2022-07-12)
+
+- Added getting statistics of all connection pools
+
 ## v2.0.1 (2022-06-16)
 
 - Fixed error return value for MGetWithGD

ha_conn_factory.go

@@ -86,6 +86,12 @@ func NewHAConnFactory(cfg *HAConfig) (*HAConnFactory, error) {
 	return factory, nil
 }
 
+func (factory *HAConnFactory) stats() map[string]*redis.PoolStats {
+	return map[string]*redis.PoolStats{
+		factory.master.redisCli.Options().Addr: factory.master.redisCli.PoolStats(),
+	}
+}
+
 func (factory *HAConnFactory) close() {
 	factory.master.redisCli.Close()
 	factory.slaves.close()

pool.go

@@ -25,6 +25,7 @@ var (
 type ConnFactory interface {
 	getSlaveConn(key ...string) (*redis.Client, error)
 	getMasterConn(key ...string) (*redis.Client, error)
+	stats() map[string]*redis.PoolStats
 	close()
 }
 
@@ -160,12 +161,8 @@ func NewShard(cfg *ShardConfig) (*Pool, error) {
 	}, nil
 }
 
-func (p *Pool) Stats() (*redis.PoolStats, error) {
-	conn, err := p.connFactory.getMasterConn()
-	if err != nil {
-		return nil, err
-	}
-	return conn.PoolStats(), nil
+func (p *Pool) Stats() map[string]*redis.PoolStats {
+	return p.connFactory.stats()
 }
 
 func (p *Pool) Close() {

shard_conn_factory.go

@@ -63,6 +63,19 @@ func NewShardConnFactory(cfg *ShardConfig) (*ShardConnFactory, error) {
 	return factory, nil
 }
 
+func (factory *ShardConnFactory) stats() map[string]*redis.PoolStats {
+	results := make(map[string]*redis.PoolStats, len(factory.shards))
+
+	for _, shard := range factory.shards {
+		result := shard.stats()
+		for addr, stats := range result {
+			results[addr] = stats
+		}
+	}
+
+	return results
+}
+
 func (factory *ShardConnFactory) close() {
 	for _, shard := range factory.shards {
 		shard.close()