Node.js 编程实战：即时聊天应用 —— 群聊与私聊功能实现

程序员爱钓鱼
58 阅读4分钟

在上一篇文章中，我们已经使用 WebSocket 在 Node.js 中实现了一个基础的实时聊天功能，能够让多个客户端建立连接并实时接收消息。但在真实的即时聊天应用中，消息并不是无差别地广播给所有人，而是需要区分 群聊 和 私聊。

本文将基于 WebSocket，进一步扩展聊天系统，实现 多人群聊 和 点对点私聊 功能，并介绍其核心设计思路。

一、群聊与私聊的设计思路

在实现功能之前，首先要明确两种聊天模式的区别。

群聊的特点是：

  • 一个消息发送到一个群
  • 群内所有在线用户都能收到
  • 群通常有唯一标识（如 roomId）

私聊的特点是：

  • 消息只在两个用户之间传递
  • 不对其他用户可见
  • 需要精准定位接收者

这两种模式的本质差异，在于消息的投递范围。

二、服务器端数据结构设计

为了实现群聊和私聊，服务端需要维护一些核心数据结构。

1. 用户与连接映射

const users = new Map(); // userId -> ws

在客户端连接成功后，将用户 ID 与 WebSocket 实例绑定，便于后续消息定向发送。

2. 群聊房间结构

const rooms = new Map(); // roomId -> Set<userId>

每个群聊房间维护一个用户列表，记录当前在线成员。

三、客户端连接与身份绑定

客户端建立连接后，第一步通常是发送用户身份信息：

{
  "type": "join",
  "userId": "u1001",
  "roomId": "room1"
}

服务器接收后：

  • 记录用户与连接的关系
  • 将用户加入指定群聊房间
ws.on('message', (msg) => {
  const data = JSON.parse(msg);

  if (data.type === 'join') {
    users.set(data.userId, ws);

    if (!rooms.has(data.roomId)) {
      rooms.set(data.roomId, new Set());
    }
    rooms.get(data.roomId).add(data.userId);
  }
});

四、群聊消息实现

1. 群聊消息格式

{
  "type": "groupMessage",
  "roomId": "room1",
  "from": "u1001",
  "content": "大家好"
}

2. 群聊消息转发逻辑

服务器收到群聊消息后：

  1. 根据 roomId 找到群成员
  2. 向群内每个在线用户发送消息
function sendGroupMessage(roomId, message) {
  const members = rooms.get(roomId);
  if (!members) return;

  members.forEach(userId => {
    const client = users.get(userId);
    if (client && client.readyState === WebSocket.OPEN) {
      client.send(JSON.stringify(message));
    }
  });
}

这样即可实现一个基础的多人群聊功能。

五、私聊消息实现

1. 私聊消息格式

{
  "type": "privateMessage",
  "from": "u1001",
  "to": "u1002",
  "content": "你好"
}

2. 私聊消息转发逻辑

私聊的关键是精准投递：

function sendPrivateMessage(message) {
  const target = users.get(message.to);
  if (target && target.readyState === WebSocket.OPEN) {
    target.send(JSON.stringify(message));
  }
}

这种方式确保消息只会被目标用户接收。

六、用户上下线处理

当用户断开连接时，需要清理相关数据：

  • 移除用户连接映射
  • 将用户从群聊房间中移除
  • 通知群内其他用户（可选）
ws.on('close', () => {
  users.forEach((value, key) => {
    if (value === ws) {
      users.delete(key);
      rooms.forEach(room => room.delete(key));
    }
  });
});

良好的上下线处理能避免“幽灵用户”问题。

七、常见问题与优化方向

在实际项目中，群聊和私聊还需要考虑更多细节。

1. 离线消息

  • 用户不在线时保存消息
  • 用户上线后推送历史消息

通常需要结合数据库或缓存系统。

2. 多房间支持

一个用户可能同时加入多个群，需要：

  • 维护用户与多个房间的关系
  • 精确控制消息投递范围

3. 权限控制

  • 是否允许用户加入某个群
  • 是否允许发送群消息

这些都需要结合业务逻辑处理。

八、与真实项目的差距

示例代码演示了核心思路，但在真实生产环境中还需要：

  • 使用 JWT 进行身份校验
  • 使用 Redis 存储在线状态
  • 多进程或集群部署
  • 使用消息队列进行跨节点通信

这些内容可以作为后续进阶优化方向。

九、总结

通过 WebSocket，我们可以在 Node.js 中轻松实现群聊和私聊功能。关键在于 连接管理、消息路由和状态维护。在掌握这些基础能力后，即时聊天系统就具备了真实产品的雏形。

在《Node.js 编程实战》系列中，即时聊天项目是理解实时通信、事件驱动和高并发处理的一个非常好的案例。

avatar
文章
阅读
粉丝