scalability improvement: use per-server peripconn pool

1 files changed, 6 insertions, 7 deletions

diff --git a/peripconn.go b/peripconn.go
index a8cb75e..afd2a92 100644
--- a/peripconn.go
+++ b/peripconn.go
@@ -7,6 +7,7 @@ import (
 )
 
 type perIPConnCounter struct {
+	pool sync.Pool
 	lock sync.Mutex
 	m    map[uint32]int
 }
@@ -45,25 +46,23 @@ type perIPConn struct {
 }
 
 func acquirePerIPConn(conn net.Conn, ip uint32, counter *perIPConnCounter) *perIPConn {
-	v := perIPConnPool.Get()
+	v := counter.pool.Get()
 	if v == nil {
-		v = &perIPConn{}
+		v = &perIPConn{
+			perIPConnCounter: counter,
+		}
 	}
 	c := v.(*perIPConn)
 	c.Conn = conn
 	c.ip = ip
-	c.perIPConnCounter = counter
 	return c
 }
 
 func releasePerIPConn(c *perIPConn) {
 	c.Conn = nil
-	c.perIPConnCounter = nil
-	perIPConnPool.Put(c)
+	c.perIPConnCounter.pool.Put(c)
 }
 
-var perIPConnPool sync.Pool
-
 func (c *perIPConn) Close() error {
 	err := c.Conn.Close()
 	c.perIPConnCounter.Unregister(c.ip)