使用redis与socket.io和NodeJs构建实时聊天

leo*_*dev 2 sockets redis node.js socket.io

我想为我的项目构建一个实时聊天系统,但实际上我在使用 Redis 时遇到了一些问题,因为我希望我的数据存储得尽可能好。

我的问题:

我想使用 Socket Io 在封闭组(两个人)中进行实时聊天,但如何存储消息?

Redis 是一个键值存储,这意味着如果我想存储某些内容,我需要在存储之前向我的数据添加一个唯一的键。

如果同一用户发布多条消息,我将在 redis 中使用哪些键?我正在考虑将唯一 id 作为唯一键,但因为我希望能够在用户登录聊天页面时获取此评论,但如果我这样做,我需要编写另一个数据库,将聊天 id 与发布该评论的用户相关联信息

我是不是忘记了什么?有没有最好的方法来做到这一点?

对不起,我的英语不好。

Zla*_*tko 5

Redis 不仅仅是键值存储。

所以你想要以下内容:

  • 聊天消息,
  • 两人讨论,
  • 您没有提到时间限制,所以我们假设您在一段时间后存档邮件,
  • 你也没有说你是否想要两个人之间单独的“线程”,例如论坛或连续消息,例如Facebook。我假设是连续的。

对于每个用户,你必须存储他发送的消息。这么说吧APP_NAMESPACE:MESSAGES:<USER_ID>:<MESSAGE_ID>。我们在这里添加 userId,以便我们可以轻松检索单个用户发送的所有消息。

并且,对于每两个用户,您需要跟踪他们的对话。作为密钥,您可以简单地使用他们的 userids APP_NAMESPACE:CONVERSATIONS:<USER1_ID>-<USER2_ID>。为了确保两个用户始终获得相同的共享对话,您可以按字母顺序对他们的 ID 进行排序,以便用户 132 和 145 都将 132:145 作为对话键

那么“对话”中要存储什么内容呢?让我们使用一个列表:[messageKey, messageKey, messageKey]

好的,但是现在的 messageKey 是什么?上面的 userId 和 messageId 的组合(这样我们就可以获取实际的消息)。

所以基本上,你需要两件事:

  1. 存储消息并为其指定 ID
  2. 将此消息的引用存储到相关对话中。

对于节点和标准 redis/hiredis 客户端,这将类似于(我将跳过明显的错误等检查,并且我将编写 ES6。如果您还无法读取 ES6,只需将其粘贴到babel):

 // assuming the init connects to redis and exports a redisClient
import redisClient from './redis-init';
import uuid from `node-uuid`;


export function storeMessage(userId, toUserId, message) {

  return new Promise(function(resolve, reject) {

    // give it an id.
    let messageId = uuid.v4(); // gets us a random uid.
    let messageKey = `${userId}:${messageId}`;
    let key = `MY_APP:MESSAGES:${messageKey}`;
    client.hmset(key, [
      "message", message,
      "timestamp", new Date(),
      "toUserId", toUserId
    ], function(err) {
      if (err) { return reject(err); }

      // Now we stored the message. But we also want to store a reference to the messageKey
      let convoKey = `MY_APP:CONVERSATIONS:${userId}-${toUserId}`; 
      client.lpush(convoKey, messageKey, function(err) {
        if (err) { return reject(err); }
        return resolve();
      });
    });
  });
}

// We also need to retreive the messages for the users.

export function getConversation(userId, otherUserId, page = 1, limit = 10) {
  return new Promise(function(resolve, reject) {
    let [userId1, userId2] = [userId, otherUserId].sort();
    let convoKey = `MY_APP:CONVERSATIONS:${userId1}-${userId2}`;
    // lets sort out paging stuff. 
    let start = (page - 1) * limit; // we're zero-based here.
    let stop = page * limit - 1;
    client.lrange(convoKey, start, stop, function(err, messageKeys) {

      if (err) { return reject(err); }
      // we have message keys, now get all messages.
      let keys = messageKeys.map(key => `MY_APP:MESSAGES:${key}`);
      let promises = keys.map(key => getMessage(key));
      Promise.all(promises)
      .then(function(messages) {
         // now we have them. We can sort them too
         return resolve(messages.sort((m1, m2) => m1.timestamp - m2.timestamp));
      })
      .catch(reject);
    }); 
  });
}

// we also need the getMessage here as a promise. We could also have used some Promisify implementation but hey.
export function getMessage(key) {
  return new Promise(function(resolve, reject)  {
    client.hgetall(key, function(err, message) {
      if (err) { return reject(err); }
      resolve(message);
    });
  });
}
Run Code Online (Sandbox Code Playgroud)

虽然这很粗糙且未经测试,但这就是如何做到这一点的要点。