如何用 Go 语言实现支持私聊的 WebSocket 聊天系统

本文详解如何使用 go 语言基于 gorilla/websocket 构建一个支持单对单私聊（而非仅群聊）的 websocket 实时聊天服务，涵盖连接管理、用户标识、消息路由与客户端交互等核心实现。

要实现“向指定用户发送消息”而非广播给所有人，关键在于为每个 WebSocket 连接绑定唯一用户身份，并建立可寻址的用户连接映射表——这正是私聊功能的基础。官方推荐的 Gary Burd 的 WebSocket Chat 示例 提供了极佳的起点，但其原始版本默认实现的是全局广播（即所有在线用户接收同一条消息）。我们需要在此基础上扩展用户注册、会话寻址与定向投递逻辑。

✅ 核心改造步骤

1. 用户登录与唯一标识
   客户端首次连接时，需通过初始 jsON 消息提交用户名（如 {“type”:”login”,”username”:”alice”}），服务端解析后将该 *Client 实例存入 map[String]*Client（以用户名为 key）。

2. 消息结构升级
   定义结构化消息类型，支持 broadcast（群发）、private（私聊）等模式：

   type Message struct {     Type      string `json:"type"`      // "private", "broadcast", "system"     From      string `json:"from"`      // 发送者用户名     To        string `json:"to"`        // 接收者用户名（私聊时必填）     Content   string `json:"content"`     Timestamp int64  `json:"timestamp"` }

3. 定向消息分发逻辑
   在 hub.broadcast 基础上新增 hub.sendToUser(to string, msg []byte) 方法：

   func (h *Hub) sendToUser(username string, message []byte) {     if client, ok := h.clients[username]; ok {         select {         case client.send <- message:         default:             close(client.send)             delete(h.clients, client.username)         }     } }

4. 客户端示例（javaScript）发送私聊

   const ws = new WebSocket("ws://localhost:8080/ws"); ws.onopen = () => {     ws.send(JSON.stringify({type: "login", username: "bob"})); }; // 向 alice 发送私聊 ws.send(JSON.stringify({     type: "private",     to: "alice",     content: "Hello! This is a private message." }));

⚠️ 注意事项

• 用户名唯一性校验：需在登录阶段检查重复，避免覆盖已有连接；
• 连接异常处理：defer hub.unregister
• 并发安全：map[string]*Client 读写需加 sync.RWMutex（原示例未包含，必须补充）；
• 生产环境建议：引入 JWT 鉴权、消息持久化（如 redis 存储离线消息）、心跳保活（websocket.SetPingHandler）。

通过以上设计，你将拥有一个轻量、可扩展、真正支持点对点通信的 Go WebSocket 聊天服务——它不止于演示，更具备落地私聊场景的完整能力。