本文介绍如何在 go 语言中设计结构化连接池，通过自定义结构体将 tcp 连接（net.conn）与唯一标识符（如用户 id 或会话 id）绑定，并安全地存入切片，从而支持按 id 精准推送通知。
本文介绍如何在 go 语言中设计结构化连接池，通过自定义结构体将 tcp 连接（net.conn）与唯一标识符（如用户 id 或会话 id）绑定，并安全地存入切片，从而支持按 id 精准推送通知。
在构建实时通知类服务（如聊天系统、状态广播或设备联动）时，服务器需维护多个活跃客户端连接，并能根据业务逻辑（例如用户 ID、设备编号）向特定连接发送消息。Go 的 net.Conn 接口本身不携带元数据，因此必须将其与业务标识符显式关联。推荐做法是定义一个组合结构体，封装连接与 ID，并使用切片统一管理。
定义连接结构体
type ConnDetails struct {
ID int // 唯一标识（可替换为 string、uuid.UUID 等）
Conn net.Conn // TCP 连接句柄
CreatedAt time.Time // 可选：便于超时清理或调试
}
该结构体清晰分离关注点：ID 用于路由，Conn 用于 I/O 操作，CreatedAt 提供生命周期上下文（非必需但强烈建议）。
初始化与安全添加连接
避免直接使用未加锁的全局切片（并发写入 panic 风险），应配合 sync.Mutex 或 sync.RWMutex：
var (
mu sync.RWMutex
connections []ConnDetails
)
// AddConnection 安全添加新连接
func AddConnection(id int, conn net.Conn) {
mu.Lock()
defer mu.Unlock()
connections = append(connections, ConnDetails{
ID: id,
Conn: conn,
CreatedAt: time.Now(),
})
}
// FindConnectionByID 查找指定 ID 的连接（读操作，用 RLock）
func FindConnectionByID(id int) (net.Conn, bool) {
mu.RLock()
defer mu.RUnlock()
for _, c := range connections {
if c.ID == id {
return c.Conn, true
}
}
return nil, false
}
实际使用示例：TCP 服务端集成
func handleConnection(conn net.Conn, nextID *int) {
// 分配唯一 ID（生产环境建议用原子操作或 UUID）
mu.Lock()
id := *nextID
*nextID++
mu.Unlock()
// 注册连接
AddConnection(id, conn)
defer func() {
mu.Lock()
// 清理：从切片中移除已断开连接（此处简化，实际需遍历+copy）
for i := range connections {
if connections[i].ID == id {
connections[i].Conn.Close()
connections = append(connections[:i], connections[i+1:]...)
break
}
}
mu.Unlock()
}()
log.Printf("Client %d connected", id)
// 后续可监听消息、转发通知等...
}
func main() {
listener, _ := net.Listen("tcp", ":8080")
defer listener.Close()
var nextID int
for {
conn, err := listener.Accept()
if err != nil {
log.Println("Accept error:", err)
continue
}
go handleConnection(conn, &nextID)
}
}
注意事项与最佳实践
- 并发安全是核心：切片的 append 和遍历均为非原子操作，务必使用互斥锁保护。
- 连接生命周期管理：net.Conn 关闭后需及时从切片中移除，否则会造成内存泄漏与无效写入 panic。
- ID 分配策略：避免简单整数递增（易冲突/暴露信息），推荐使用 atomic.Int64 或 github.com/google/uuid。
- 扩展性考虑：当连接数增长，切片线性查找效率下降，可升级为 map[int]net.Conn（仍需锁保护）或引入连接池（如 sync.Map + 自定义清理逻辑）。
- 错误处理不可省略：对 conn.Write() 调用必须检查返回错误（如 io.EOF 表示客户端已断开），避免静默失败。
通过这种结构化设计，你不仅能精准定位目标连接，还能为后续功能（如心跳检测、连接统计、权限校验）打下坚实基础。
