Merge pull request #4 from yuzhichang/add_foreach

added (*AddressBasedPool).ForEach
fagongzi · Nov 30, 2018 · 3d2943b · 3d2943b
2 parents fbef9f6 + 2adb8c4
commit 3d2943b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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()
+}