本文介绍使用 go 语言构建 websocket 服务时，如何将某客户端发送的消息精准反射（广播）给另一指定客户端（如定位追踪场景），核心在于连接管理、会话标识与消息路由。
本文介绍使用 go 语言构建 websocket 服务时，如何将某客户端发送的消息精准反射（广播）给另一指定客户端（如定位追踪场景），核心在于连接管理、会话标识与消息路由。
在基于 WebSocket 的实时协作或位置追踪系统（例如司机与乘客双向定位）中，常见需求是：客户端 A 发送一条包含经纬度、状态等信息的 JSON 消息，服务端需仅将其转发给与之关联的客户端 B（而非全量广播）。这要求服务端具备“连接-身份-关系”三层映射能力，而非简单回显或群发。
WebSocket 8.18.2
下载
WebSocket 8.18.2 是该协议规范的一个重要迭代版本，主要优化了连接稳定性与数据传输效率。它通过全双工通信机制，允许客户端与服务器在单一长连接上实时交换数据，大幅降低传统 HTTP 轮询的开销。该版本增强了心跳保活、自动重连及二进制帧传输能力，适用于即时通讯、在线游戏及金融行情推送等低延迟场景，为开发者提供更可靠的实时网络交互基础。
✅ 关键设计原则
-
唯一标识每个连接：为每个 WebSocket 连接分配唯一 ID（如
connID），并关联其认证信息（token）和业务上下文（jobId）； -
建立业务维度索引：以
jobId为键，维护一个map[string][]*Client，记录当前参与同一任务的所有在线客户端； -
消息路由而非广播：当收到某客户端消息时，解析其
jobId，查表获取目标客户端列表，逐个推送（可进一步按token精确匹配单个接收方）。
? 示例代码结构（精简版）
type Client struct {
ID string // 连接唯一ID（如 uuid.NewString()）
Token string // 认证凭证
JobID string // 业务关联ID（如订单/任务ID）
Conn *websocket.Conn
}
var (
// 按 jobId 分组的在线客户端池
jobClients = sync.Map{} // map[string][]*Client
)
func handleWebSocket(conn *websocket.Conn) {
// 1. 解析 URL 查询参数
query := conn.Request().URL.Query()
token := query.Get("token")
jobID := query.Get("jobId")
if token == "" || jobID == "" {
http.Error(conn.HTTPResponse, "missing token or jobId", http.StatusBadRequest)
return
}
// 2. 创建并注册客户端
client := &Client{
ID: uuid.NewString(),
Token: token,
JobID: jobID,
Conn: conn,
}
// 加入 jobId 对应的客户端池
if clients, ok := jobClients.Load(jobID); ok {
jobClients.Store(jobID, append(clients.([]*Client), client))
} else {
jobClients.Store(jobID, []*Client{client})
}
defer func() {
// 连接关闭时清理
if clients, ok := jobClients.Load(jobID); ok {
filtered := []*Client{}
for _, c := range clients.([]*Client) {
if c.ID != client.ID {
filtered = append(filtered, c)
}
}
if len(filtered) == 0 {
jobClients.Delete(jobID)
} else {
jobClients.Store(jobID, filtered)
}
}
}()
// 3. 消息处理循环
for {
_, msg, err := conn.ReadMessage()
if err != nil {
break // 连接断开
}
// 4. 解析并路由：仅转发给同 jobId 的其他 client（可选：排除自身）
if targets, ok := jobClients.Load(jobID); ok {
for _, target := range targets.([]*Client) {
if target.ID == client.ID { // 跳过自己（可选）
continue
}
if err := target.Conn.WriteMessage(websocket.TextMessage, msg); err != nil {
log.Printf("failed to send to %s: %v", target.ID, err)
}
}
}
}
}⚠️ 注意事项
-
并发安全：
sync.Map适用于读多写少场景；若高频增删，建议改用sync.RWMutex+ 常规map； -
连接有效性检查：发送前建议调用
target.Conn.WriteDeadline()或捕获websocket.IsCloseError避免向已断连客户端写入； -
Token 校验前置：应在
handleWebSocket初期验证token合法性（如 JWT 解析、DB 查询），防止未授权接入； -
扩展性考虑：单机部署下上述方案可行；若需横向扩展，应将
jobClients迁移至 Redis Pub/Sub 或分布式缓存，并引入连接网关层。
通过以上设计，即可在 Go WebSocket 服务中稳健实现“一客发、一客收”的定向消息反射，满足实时定位、协同操作等典型业务需求。
