what if those conns are tracked and closed on Close?

Author: ndyakov

internal/pool/pool.go

@@ -68,6 +68,9 @@ type Pooler interface {
 	AddPoolHook(hook PoolHook)
 	RemovePoolHook(hook PoolHook)
 
+	TrackConn(*Conn)
+	UntrackConn(*Conn)
+
 	Close() error
 }
 
@@ -736,3 +739,20 @@ func (p *ConnPool) isHealthyConn(cn *Conn, now time.Time) bool {
 	}
 	return true
 }
+
+func (p *ConnPool) TrackConn(cn *Conn) {
+	p.connsMu.Lock()
+	p.conns = append(p.conns, cn)
+	p.connsMu.Unlock()
+}
+
+func (p *ConnPool) UntrackConn(cn *Conn) {
+	p.connsMu.Lock()
+	for i, c := range p.conns {
+		if c == cn {
+			p.conns = append(p.conns[:i], p.conns[i+1:]...)
+			break
+		}
+	}
+	p.connsMu.Unlock()
+}

redis.go

@@ -1071,10 +1071,12 @@ func (c *Client) pubSub() *PubSub {
 				return nil, err
 			}
 
+			c.pool.TrackConn(cn)
 			return cn, nil
 		},
 		closeConn: func(cn *pool.Conn) error {
 			_ = cn.Close()
+			c.pool.UntrackConn(cn)
 			return nil
 		},
 		pushProcessor: c.pushProcessor,