本文介绍如何在 go 语言中设计一个带唯一标识符（id）的 socket 连接集合，便于后续按 id 精准推送通知；通过结构体封装连接与 id，并结合 goroutine 安全地维护连接列表。
本文介绍如何在 go 语言中设计一个带唯一标识符（id）的 socket 连接集合，便于后续按 id 精准推送通知；通过结构体封装连接与 id，并结合 goroutine 安全地维护连接列表。
在构建实时通知服务（如聊天系统、设备状态推送等）时，常需将每个客户端连接与其逻辑 ID 关联起来，以便后续定向发送消息。Go 语言中，net.Conn 表示一个活跃的网络连接，但原生不携带业务标识。因此，推荐使用自定义结构体将连接与 ID 绑定，并用切片（slice）或更优的 map 存储管理。
✅ 推荐结构设计
首先定义一个清晰、可扩展的连接元数据结构：
type ConnDetails struct {
ID int // 业务唯一标识（如用户ID、设备ID）
Conn net.Conn // TCP/UDP 连接句柄
Created time.Time // 可选：记录连接时间，便于超时清理
}
⚠️ 注意：net.Conn 是接口类型，不可直接复制或跨 goroutine 并发读写。务必确保每个 Conn 仅由一个 goroutine 负责读/写（通常一读一写分离），避免竞态。
? 连接注册：为每个新连接分配 ID 并存入集合
在 accept 循环中，为每个新连接生成唯一 ID（如递增计数器 + sync.AtomicInt64），并安全添加到共享集合中：
var (
connections = make(map[int]ConnDetails) // 更推荐：O(1) 查找 & 删除
nextID = int64(1)
mu sync.RWMutex
)
func handleConnection(conn net.Conn) {
id := int(atomic.AddInt64(&nextID, 1))
details := ConnDetails{
ID: id,
Conn: conn,
Created: time.Now(),
}
mu.Lock()
connections[id] = details
mu.Unlock()
log.Printf("Client %d connected from %s", id, conn.RemoteAddr())
// 启动独立 goroutine 处理该连接（含心跳、读取消息、异常关闭等）
go serveConn(details)
}
? 按 ID 发送通知的示例
当需要向特定客户端推送消息时，只需查找对应 ID 并写入：
func NotifyByID(id int, msg []byte) error {
mu.RLock()
connDetail, ok := connections[id]
mu.RUnlock()
if !ok {
return fmt.Errorf("client %d not found", id)
}
_, err := connDetail.Conn.Write(msg)
return err
}
// 使用示例：
// NotifyByID(123, []byte("ALERT: system rebooting"))
? 关键注意事项
- 并发安全：切片本身非线程安全，若坚持用 []ConnDetails，需配合 sync.Mutex 全局锁；而 map[int]ConnDetails 配合读写锁（sync.RWMutex）更高效。
-
连接生命周期管理：务必在连接关闭（conn.Close()）后从集合中移除对应项，防止内存泄漏和无效写入：
defer func() { mu.Lock() delete(connections, id) mu.Unlock() log.Printf("Client %d disconnected", id) }() - 资源释放：net.Conn 实现了 io.Closer，务必调用 Close() 释放底层文件描述符。
- ID 分配策略：生产环境建议使用 UUID 或分布式 ID（如 snowflake），避免单点 ID 冲突；简单场景可用原子递增。
✅ 总结
将 net.Conn 与业务 ID 封装进结构体是 Go 服务端开发的常见模式。比起裸 slice，使用带锁的 map 能提供更可靠的查找、删除与并发支持。结合 goroutine 分离连接生命周期处理，即可构建出稳定、可扩展的通知分发系统。
