From 2adb8c4177e6233339f5c0b113e82320c9ef6c7a Mon Sep 17 00:00:00 2001 From: Zhichang Yu Date: Fri, 30 Nov 2018 10:06:27 +0800 Subject: [PATCH] added (*AddressBasedPool).ForEach --- conn_pool.go | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/conn_pool.go b/conn_pool.go index d2f0e0c..615f8d8 100644 --- a/conn_pool.go +++ b/conn_pool.go @@ -269,3 +269,11 @@ func (pool *AddressBasedPool) createConn(addr string) IOSession { pool.Unlock() return conn } + +func (pool *AddressBasedPool) ForEach(visitor func(addr string, conn IOSession)) { + pool.Lock() + for addr, conn := range pool.conns { + visitor(addr, conn) + } + pool.Unlock() +}