如何在MERN应用中基于Socket.io实现帖子所有者的点赞评论通知

1. 如何追踪帖子所有者并向其发送通知？ #

核心是维护在线用户与Socket实例的映射关系，结合帖子数据中的所有者ID精准推送：

• 前端连接Socket时携带当前用户ID（从登录态中获取）：

  // React前端代码
  import { io } from 'socket.io-client';
  const userId = localStorage.getItem('currentUserId');
  const socket = io('http://your-backend-url', { query: { userId } });
  

• 后端在用户连接时，将用户ID与socket.id绑定，存入映射表：

  // Node.js后端代码
  const onlineUsers = new Map(); // 键：用户ID，值：socket.id
  
  io.on('connection', (socket) => {
    const userId = socket.handshake.query.userId;
    if (userId) onlineUsers.set(userId, socket.id);
  
    // 用户断开连接时清理映射
    socket.on('disconnect', () => onlineUsers.delete(userId));
  });
  

• 触发点赞/评论事件时，从数据库获取帖子所有者ID，通过映射表找到对应socket推送：

  // 后端处理点赞事件
  socket.on('likePost', async (data) => {
    const { postId, likerId } = data;
    const post = await Post.findById(postId); // 从MongoDB获取帖子数据
    const ownerId = post.owner.toString();
  
    // 检查所有者是否在线，在线则推送通知
    const ownerSocketId = onlineUsers.get(ownerId);
    if (ownerSocketId) {
      io.to(ownerSocketId).emit('newNotification', {
        type: 'like',
        from: likerId,
        postId,
        content: '有人点赞了你的帖子'
      });
    }
  });
  

2. 是否需要为每个帖子创建Socket房间？ #

不需要。针对"仅向帖子所有者发通知"的场景，直接通过用户ID映射socket的方案更轻量、高效。

• 帖子房间适合批量通知场景（比如帖子被管理员置顶，需要通知所有关注该帖子的用户），此时可以创建post-${postId}命名的房间，用户关注帖子时加入、取消关注时退出。但仅针对所有者的通知，无需额外创建房间。

3. 其他基于Socket.io的技术方案 #

分布式场景下的Redis扩展 #

如果你的应用是多实例部署，单进程的onlineUsers映射表会失效，可结合Redis存储在线用户信息并实现跨实例通知：

// 后端引入Redis
const redis = require('redis');
const redisClient = redis.createClient();
const redisPub = redisClient.duplicate();
const redisSub = redisClient.duplicate();

// 用户连接时将socket.id存入Redis
io.on('connection', async (socket) => {
  const userId = socket.handshake.query.userId;
  await redisClient.set(`user:${userId}`, socket.id);
  redisSub.subscribe(`notification:${userId}`); // 订阅用户专属通知频道
});

// 发送通知时通过Redis发布
socket.on('commentPost', async (data) => {
  const post = await Post.findById(data.postId);
  const ownerId = post.owner.toString();
  await redisPub.publish(`notification:${ownerId}`, JSON.stringify({
    type: 'comment',
    from: data.commenterId,
    postId,
    content: '有人评论了你的帖子'
  }));
});

// 接收Redis消息并推送给对应socket
redisSub.on('message', (channel, message) => {
  const userId = channel.split(':')[1];
  redisClient.get(`user:${userId}`).then(socketId => {
    if (socketId) io.to(socketId).emit('newNotification', JSON.parse(message));
  });
});

离线通知兜底 #

Socket.io仅能处理在线用户，对于离线用户，需将通知存入数据库，用户下次登录时拉取：

• 后端判断用户不在线时，将通知写入Notification集合：

  if (!ownerSocketId) {
    await Notification.create({
      userId: ownerId,
      type: 'like',
      from: likerId,
      postId,
      isRead: false
    });
  }
  

• 前端登录后，调用API获取未读通知并展示。

内容的提问来源于stack exchange，提问作者prathamesh mungekar