在Next.js与Node.js中用Socket.io实现仅向指定用户发送消息

1. 服务器端改造 #

核心是维护一个用户ID与Socket ID的映射表，用来记录在线用户对应的连接标识，以此实现定向消息发送。

修改后的服务器代码：

// 用Map存储用户ID和Socket ID的映射，避免普通对象的键类型限制
const userSocketMap = new Map();

io.on('connection', (socket) => {    
  // 客户端连接后，接收用户ID并绑定对应Socket ID
  socket.on('user-connected', (usuarioID) => {
    userSocketMap.set(usuarioID, socket.id);
  });

  // 处理消息发送请求，仅发给目标用户
  socket.on('mensaje', (data) => {
    // 从映射表中获取目标用户的Socket ID
    const targetSocketId = userSocketMap.get(data.destinoUsuarioID);
    if (targetSocketId) {
      // 定向发送消息到目标用户的连接
      io.to(targetSocketId).emit('mensaje-recibido', {
        contenido: data.contenido,
        remitenteID: data.remitenteID
      });
    }
  });

  // 用户断开连接时，清理映射表中的无效记录
  socket.on('disconnect', () => {
    for (const [usuarioID, socketId] of userSocketMap.entries()) {
      if (socketId === socket.id) {
        userSocketMap.delete(usuarioID);
        break;
      }
    }
  });
});

2. 客户端改造 #

需要在连接成功后先绑定用户ID，发送消息时必须传递目标用户ID和消息内容：

修改后的客户端代码：

// 连接建立后，立即发送当前用户ID给服务器绑定
socket.on('connect', () => {
  socket.emit('user-connected', usuarioActual.usuarioID);
});

// 发送消息的函数，传入目标用户ID和消息内容
const enviarMensaje = (destinoUsuarioID, contenido) => {
  socket.emit('mensaje', {
    destinoUsuarioID,
    contenido,
    remitenteID: usuarioActual.usuarioID // 可选：携带发送者ID用于前端展示
  });
};

// 监听定向发来的消息
socket.on('mensaje-recibido', (mensaje) => {
  // 这里处理收到的消息，比如渲染到页面
  console.log('收到消息:', mensaje);
});

关键注意点 #

• userSocketMap必须定义在io.on('connection')外部，确保所有连接实例共享同一映射表。
• 使用io.to(socketId).emit()可以精准定位到目标用户的Socket连接，不会广播给其他用户。
• 必须处理断开连接事件，避免映射表残留无效的Socket ID导致消息发送失败。

内容的提问来源于stack exchange，提问作者Lucian