AI创作问答(高级版)
在Go语言中，可以使用`net`和`sync`包来实现TCP连接池。下面是一个简单的示例代码：
```go
package main
import (
"fmt"
"net"
"sync"
)
type TCPConnPool struct {
pool chan net.Conn
mutex sync.Mutex
}
func NewTCPConnPool(addr string, size int) (*TCPConnPool, error) {
pool := make(chan net.Conn, size)
for i := 0; i < size; i++ {
conn, err := net.Dial("tcp", addr)
if err != nil {
return nil, err
}
pool <- conn
}
return &TCPConnPool{pool: pool}, nil
}
func (p *TCPConnPool) Get() (net.Conn, error) {
select {
case conn := <-p.pool:
return conn, nil
default:
conn, err := net.Dial("tcp", "your_tcp_server_address")
if err != nil {
return nil, err
}
return conn, nil
}
}
func (p *TCPConnPool) Put(conn net.Conn) {
p.pool <- conn
}
func main() {
pool, err := NewTCPConnPool("your_tcp_server_address", 10)
if err != nil {
fmt.Println("Failed to create TCP connection pool:", err)
return
}
conn, err := pool.Get()
if err != nil {
fmt.Println("Failed to get TCP connection from pool:", err)
return
}
// 使用连接进行操作...
pool.Put(conn)
}
```
这段代码定义了一个`TCPConnPool`结构体，其中包含一个大小为`size`的连接池。`NewTCPConnPool`函数用于创建连接池，并初始化一定数量的连接。`Get`方法从连接池获取可用的连接，如果连接池为空，则会创建新的连接。`Put`方法将使用完毕的连接放回连接池。
你可以根据自己的实际情况修改代码中的`your_tcp_server_address`为目标TCP服务器的地址。注意在实际使用中，你可能需要添加适当的错误处理和连接超时控制等。
希望以上解答对你有帮助！如果还有其他问题，请随时提问。
金宣小助手
官网微信公众号
