Nodejs express+socket.io，如何限制一个用户只能登录一次，一个用户只有一个连接到socket.io

Node.js Express + Socket.io: 如何限制一个用户只能登录一次，一个用户只有一个连接到Socket.io

服务端代码

在服务端，我们需要确保每个用户只能有一个活动的Socket连接。这可以通过使用socket.io的授权钩子（authorization hook）来实现。我们还需要维护一个用户列表，以跟踪哪些用户已经连接。

以下是修改后的服务端代码：

const io = require('socket.io')(server); // 假设 server 是你的 HTTP 服务器实例
const sessionStore = require('./sessionStore'); // 假设这是你的会话存储模块
const cookie = require('cookie');
const utils = require('express/node_modules/connect/lib/utils');

let connectedUsers = new Set(); // 使用 Set 来存储用户名

io.set('authorization', function (handshakeData, callback) {
  const cookies = cookie.parse(handshakeData.headers.cookie);
  const signedCookies = utils.parseSignedCookies(cookies, settings.cookie_secret);
  
  sessionStore.get(signedCookies['connect.sid'], function(err, session) {
    if (err) {
      callback(err, false);
      return;
    }
    
    const username = session.user && session.user.name;
    console.log('Authorization for user:', username);

    if (!username) {
      callback('未找到用户名', false);
      return;
    }

    if (connectedUsers.has(username)) {
      // 用户已连接
      callback('重复登录', false);
    } else {
      connectedUsers.add(username);
      handshakeData.username = username; // 设置 handshakeData 中的用户名
      callback(null, true);
    }
  });
});

io.of('/client').on('connection', function (socket) {
  const username = socket.handshake.query.username; // 获取用户名
  console.log(`User ${username} connected`);

  socket.on('disconnect', function () {
    console.log(`User ${username} disconnected`);
    connectedUsers.delete(username);
  });

  socket.on('someEvent', function (data) {
    console.log(`Received data from ${username}:`, data);
  });
});

客户端代码

客户端需要在连接时传递用户名，并处理连接失败的情况：

const username = 'your-username'; // 替换为实际的用户名
const client = io.connect('/client', { query: `username=${encodeURIComponent(username)}` });

client.on('connect', function () {
  console.log(`Connected as ${username}`);
});

client.on('connect_error', function (error) {
  console.error(`Connection error: ${error.message}`);
});

client.on('disconnect', function () {
  console.log(`Disconnected as ${username}`);
});

解释

1. 服务端：

   • 在authorization钩子中，我们从请求头中解析出Cookie，并通过会话存储获取用户的会话数据。
   • 如果用户已经存在于connectedUsers集合中，则拒绝新的连接。
   • 将用户名添加到connectedUsers集合中，并将其设置为handshakeData.username，以便在后续事件处理中使用。

2. 客户端：

   • 连接时传递用户名作为查询参数。
   • 监听connect_error事件来处理连接失败的情况。

这样，每个用户将只能有一个活跃的Socket连接。