Golang TCP并发连接池实现示例-Golang-PHP中文网

Golang TCP并发连接池实现示例

P粉602998670

发布： 2025-09-29 12:37:02

原创

780人浏览过

答案：Go语言通过自定义TCPConnPool结构体实现并发连接池，利用channel缓存空闲连接并控制容量，配合sync.Mutex保证关闭操作的线程安全，通过NewTCPConnPool初始化池，Get方法优先从channel获取连接，若为空则新建，实现连接复用以提升高并发性能。

golang tcp并发连接池实现示例

在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：标识连接池是否已关闭

初始化与连接获取

使用工厂函数创建连接池实例：

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;"><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通信场景。对于高频或长连接服务，可扩展支持心跳检测、最大空闲时间、连接预热等机制。关键是控制并发访问安全，避免连接泄漏。

以上就是Golang TCP并发连接池实现示例的详细内容，更多请关注php中文网其它相关文章！