如何在Socket对象中存储/添加数据？Socket ID关联用户信息咨询

能否将用户信息存储在Socket对象中？ #

完全可以，这是Socket类库（比如Socket.io）的常规用法。单个连接生命周期内需要的用户数据，直接挂载到Socket实例上是最直接高效的方式，既不用频繁查询数据库，也不会产生额外性能开销。

如何将用户信息添加至Socket对象？ #

以Socket.io为例，在用户连接并完成身份验证后，直接给当前Socket实例添加自定义属性即可：

io.on('connection', (socket) => {
  // 假设已通过身份验证拿到用户信息userInfo
  const userInfo = { userId: 1001, username: 'test_user' };
  // 将用户信息挂载到socket对象
  socket.user = userInfo;

  // 后续业务逻辑中可直接取用
  socket.on('send-message', (content) => {
    console.log(`用户${socket.user.username}发送消息：${content}`);
  });
});

其他Socket类库的逻辑类似，只要是当前连接对应的Socket实例，都可以添加自定义属性来存储用户数据。

集群模式下的适配问题 #

你提到集群模式无法使用全局对象，但挂载到单个Socket实例的信息是连接上下文独立的——每个Worker进程处理各自的连接，Socket实例本身属于当前进程的连接上下文，不会跨进程共享，所以完全适配集群环境，不存在同步问题。

跨进程查询的补充方案 #

如果后续需要通过Socket ID跨进程获取用户信息，仅靠Socket对象存储就不够了，这时可以用Socket.io自带的Redis Adapter来同步集群内的连接数据：

// 初始化Redis Adapter
const { createAdapter } = require('@socket.io/redis-adapter');
const { createClient } = require('redis');

const pubClient = createClient({ url: 'redis://localhost:6379' });
const subClient = pubClient.duplicate();

Promise.all([pubClient.connect(), subClient.connect()]).then(() => {
  io.adapter(createAdapter(pubClient, subClient));
});

io.on('connection', (socket) => {
  const userInfo = { userId: 1001, username: 'test_user' };
  socket.user = userInfo;
  // 将Socket ID与用户ID关联，存入Adapter
  io.of('/').adapter.remoteJoin(socket.id, `user:${userInfo.userId}`);
});

// 跨进程通过用户ID获取对应的Socket ID
async function getSocketIdByUserId(userId) {
  const sockets = await io.in(`user:${userId}`).allSockets();
  return sockets.values().next().value; // 假设单个用户仅一个在线连接
}

这种方式既保留了Socket对象存储的便捷性，又解决了集群下跨进程查询的需求，性能远优于频繁操作数据库。

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