答案：Go语言通过自定义TCPConnPool结构体实现并发连接池，利用channel缓存空闲连接并控制容量，配合sync.Mutex保证关闭操作的线程安全，通过NewTCPConnPool初始化池，Get方法优先从channel获取连接，若为空则新建，实现连接复用以提升高并发性能。
在Go语言中实现TCP并发连接池，主要是为了复用已建立的TCP连接，避免频繁创建和销毁连接带来的开销，提升高并发场景下的性能。虽然标准库net没有直接提供连接池支持，但我们可以借助sync.Pool或自定义结构管理连接队列。下面是一个轻量级、线程安全的TCP连接池实现示例。
连接池核心结构设计
连接池需要维护一组空闲连接，限制最大连接数，并保证并发安全。
定义一个TCPConnPool结构体：
type TCPConnPool struct {
addr string
capacity int
connections chan net.Conn
mu sync.Mutex
closed bool
}
字段说明：
立即学习“go语言免费学习笔记（深入）”；
- addr：目标服务地址，如"127.0.0.1:8080"
- capacity：连接池最大容量
- connections：缓存空闲连接的带缓冲channel
- mu：保护关闭状态的锁
- closed：标识连接池是否已关闭
初始化与连接获取
使用工厂函数创建连接池实例：
Golang Uber Dig
下载
使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等
func NewTCPConnPool(addr string, cap int) *TCPConnPool {
return &TCPConnPool{
addr: addr,
capacity: cap,
connections: make(chan net.Conn, cap),
}
}
从池中获取连接时，优先从channel中取，若为空则新建：
func (p *TCPConnPool) Get() (net.Conn, error) {
select {
case conn := <-p.connections:
if isHealthy(conn) {
return conn, nil
}
// 连接不健康，尝试重新建立
return p.dial()
default:
return p.dial()
}
}
<p>func (p *TCPConnPool) dial() (net.Conn, error) {
p.mu.Lock()
defer p.mu.Unlock()
if p.closed {
return nil, errors.New("connection pool is closed")
}
return net.Dial("tcp", p.addr)
}
isHealthy用于检测连接是否有效（例如通过写入心跳）：
func isHealthy(conn net.Conn) bool {
if conn == nil {
return false
}
conn.SetReadDeadline(time.Now().Add(10 * time.Millisecond))
var buf [1]byte
n, err := conn.Read(buf[:])
return n == 0 && err != nil
}
连接归还与资源释放
使用完连接后应归还到池中，而不是直接关闭：
func (p *TCPConnPool) Put(conn net.Conn) error {
p.mu.Lock()
defer p.mu.Unlock()
if p.closed {
return conn.Close()
}
select {
case p.connections <- conn:
return nil
default:
// 池已满，关闭连接
return conn.Close()
}
}
关闭连接池时需关闭所有现存连接：
func (p *TCPConnPool) Close() {
p.mu.Lock()
defer p.mu.Unlock()
if p.closed {
return
}
p.closed = true
close(p.connections)
for conn := range p.connections {
conn.Close()
}
}
使用示例
模拟多个goroutine并发使用连接池：
pool := NewTCPConnPool("localhost:9000", 10)
<p>var wg sync.WaitGroup
for i := 0; i < 20; i++ {
wg.Add(1)
go func(id int) {
defer wg.Done()
conn, err := pool.Get()
if err != nil {
log.Printf("Goroutine %d: %v", id, err)
return
}
defer pool.Put(conn)</p><pre class="brush:php;toolbar:false;"> // 发送数据
conn.Write([]byte("hello"))
// 接收响应
buf := make([]byte, 1024)
n, _ := conn.Read(buf)
log.Printf("Goroutine %d received: %s", id, buf[:n])
}(i)
} wg.Wait() pool.Close()
基本上就这些。这个连接池适合中低频TCP通信场景。对于高频或长连接服务，可扩展支持心跳检测、最大空闲时间、连接预热等机制。关键是控制并发访问安全，避免连接泄漏。
