Question

我正在尝试建立一个实时聊天，我使用mongodb和socket.io来存储消息和用户。

创建新用户时，用户存储在mongodb和套接字对象中。

如果用户刷新页面，则用户将从套接字对象中删除，这意味着为了让该用户重新进入，他们必须创建一个新用户名并生成一个新套接字。

这是我的服务器端代码的样子

/*
|--------------------------------------------------------------------------
| Live socket communication with front end:
|--------------------------------------------------------------------------
|
|
*/

var users = {};

io.sockets.on('connection', function (socket) {

    // Listen to a new user then emit the user to all clients
    socket.on('new user', function (data) {

        // Check mongodb to see if the user exists and emit proper message
        models.Message.findOne({username:data},function(err,user){
            if(err){
                console.log('something went wrong')
            }
            else if(user){
                socket.emit('username taken', 'something');
            }
            else{
                socket.emit('create user', data);
                socket.userName = data;
                socket.connected = true;
                users[socket.userName] = socket;
                io.sockets.emit('user name', Object.keys(users));
            }
        });

    });

    socket.on('facebook id', function(data) {

        models.User.findOne({username:data.name}, function(err, user) {
            if (user) {
                console.log('User already exists');
                socket.userName = data.name;
                socket.facebook_id = data.id;
                socket.connected = true;
                users[socket.userName] = socket;
                io.sockets.emit('user name', Object.keys(users));
            } 
            else {
                var newUser = new models.User({
                    username: data.name,
                    facebook_id: data.id
                });
                newUser.save(function(err, user) {
                    console.log('successfully inserted user/user: ' + user._id);
                });
            }
        });

    });

    // Listen to a new message then emit the message to all clients
    socket.on('send message', function (data, callback) {

        io.sockets.emit('new message', {message: data, username: socket.userName, facebook_id: socket.facebook_id});

    });

    // Logic when client disconnects
    socket.on('disconnect', function (data) {

        if(!socket.userName) return;
        seeder.disconnect(socket.userName);
        delete users[socket.userName]
        io.sockets.emit('user disconnected', Object.keys(users));

    });

});

您在我的断开连接中看到我从users对象中删除了套接字。

我的问题是有没有办法在断开连接时保存套接字信息然后如果同一套接字尝试连接它是否识别用户并继续？

Additonal：我想我可能需要首先专注于使用mongodb创建用户登录，然后在会话数据中使用该日志并将其传递给套接字，使用当前数据库创建套接字对象细节？这听起来像是更有意义的东西，这可能吗？

Answer 1

您可以使用Cookie来识别用户。生成随机哈希，将其放入cookie中，因此在客户端和服务器之间建立连接时将传输此数据。

客户端代码可能如下所示：

function generateHash(len) {
  var symbols = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890';
  var hash = '';
  for (var i = 0; i < len; i++) {
    var symIndex = Math.floor(Math.random() * symbols.length);
    hash += symbols.charAt(symIndex);
  }
  return hash;
}
if (!/\buser_id=/.test(document.cookie)) { //if no 'user_id' in cookies
  document.cookie = 'user_id=' + generateHash(32);  //add cookie 'user_id'
}

//here goes establishing connection to server via `io.connect`

在服务器端，您可以写：

io.sockets.on('connection', function (socket) {
  var cookie = socket.handshake.headers.cookie;
  var match = cookie.match(/\buser_id=([a-zA-Z0-9]{32})/);  //parse cookie header
  var userId = match ? match[1] : null;
  //...

因此，您拥有userId变量，该变量对每个用户都是唯一的。然后你可以评论这一行：

delete users[socket.userName]

因为您应该保留用户数据。

您现在可以将users对象以userId（非用户名）存储为密钥，并在每个连接上检查users[userId] != null。如果此类用户存在，请使用他们的套接字信息

刷新后Socket.io断开连接？

1 个答案: