如何为WebSocket连接附加额外数据以实现断开时的清理操作

这问题我之前做多人卡牌游戏的时候也碰到过，其实思路和你想的完全一致——给WebSocket实例添加自定义的playerId属性，就能在close事件里轻松识别玩家并清理数据了。不过有几个细节需要调整，我帮你把代码修正并完善一下：

1. 修正close事件的参数（关键错误） #

你之前的close事件回调写法有问题：ws.on("close", function close(ws) {}) 里的参数不是WebSocket实例，而是断开的状态码code和原因reason，这个参数名会误导你。正确的回调参数应该是code和reason，WebSocket实例本身是外层的ws变量。

2. 在message事件中绑定playerId #

当收到包含玩家身份的消息时，直接给当前的WebSocket实例添加playerId属性，同时完成和player对象的双向关联：

wss.on("connection", function connection(ws) {
  console.log("新连接建立");
  
  // 先初始化playerId为null，避免后续判断undefined
  ws.playerId = null;

  ws.on("close", function close(code, reason) {
    console.log(`连接断开: ${code} - ${reason}`);
    // 只有当playerId存在时才执行清理逻辑
    if (ws.playerId) {
      // 解除player对象和ws的关联
      if (player[ws.playerId]) {
        player[ws.playerId].ws = null;
        
        // 清理游戏数据（示例：移除玩家所在房间的记录）
        if (player[ws.playerId].roomId && gameRooms[player[ws.playerId].roomId]) {
          const targetRoom = gameRooms[player[ws.playerId].roomId];
          targetRoom.players = targetRoom.players.filter(p => p.id !== ws.playerId);
          // 如果房间空了，销毁房间释放内存
          if (targetRoom.players.length === 0) {
            delete gameRooms[player[ws.playerId].roomId];
          }
        }
        
        // 删除玩家数据（根据你的业务逻辑调整）
        delete player[ws.playerId];
        console.log(`已清理玩家 ${ws.playerId} 的所有关联数据`);
      }
    }
  });

  ws.on("message", function incoming(message) {
    console.log("收到消息: %s", message);
    try {
      const obj = JSON.parse(message);
      switch (obj.action) {
        // 假设你有"login"或"joinGame"这类传递玩家ID的动作
        case "login":
        case "joinGame":
          // 给WebSocket实例绑定玩家ID
          ws.playerId = obj.playerId;
          // 关联player对象的ws属性
          player[obj.playerId] = player[obj.playerId] || {};
          player[obj.playerId].ws = ws;
          // 其他业务逻辑：比如给玩家发送进入游戏的确认消息
          ws.send(JSON.stringify({ action: "loginSuccess", playerId: obj.playerId }));
          break;
        // 其他动作的处理逻辑...
        default:
          console.log("未知动作类型");
      }
    } catch (err) {
      console.error("消息解析失败:", err);
      // 给客户端返回错误提示
      ws.send(JSON.stringify({ error: "无效的消息格式" }));
    }
  });
});

3. 额外注意事项 #

• 消息解析容错：一定要用try-catch包裹JSON.parse，避免客户端发送非法JSON导致服务器崩溃
• 未绑定playerId的连接处理：如果玩家刚连接上还没发送身份消息就断开，ws.playerId会是null，直接跳过清理逻辑即可
• 内存泄漏防范：清理时要确保解除所有双向关联（player的ws和ws的playerId），避免内存泄漏

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