当前位置：首页 > 文章列表 > 文章 > 前端 > Mongoose好友添加功能实现详解

Mongoose好友添加功能实现详解

2025-10-05 10:18:36 0浏览收藏

本文详细讲解了如何使用 Mongoose 在 Node.js 应用中实现用户添加好友功能，并着重强调了安全性与数据一致性的重要性。文章深入探讨了处理好友请求的正确姿势，避免前端篡改，并推荐从已认证的用户信息中获取发送者 ID。同时，对比了在用户模式中维护好友数组与通过查询 FriendRequest 集合获取好友列表两种方案，推荐后者以简化数据结构。针对更新多个集合可能导致数据不一致的问题，本文还详细介绍了 Mongoose 事务的使用方法，确保好友关系变更的原子性。无论你是初学者还是经验丰富的开发者，本文都能帮助你安全、高效地实现用户好友功能，提升应用的用户体验。

在 Mongoose 中实现用户添加好友功能

本文旨在指导开发者如何在 Mongoose 中实现用户添加好友的功能，并探讨了维护用户好友关系的不同策略。重点介绍如何安全地处理好友请求，以及在用户接受好友请求后，如何正确更新用户模式中的好友数组。同时，也讨论了使用额外数组存储好友关系的必要性，并提出了更优的查询方案。

安全地处理好友请求

在处理好友请求时，安全性至关重要。不要从前端获取发送者 ID，因为这容易受到篡改。应该从已经认证的用户信息中获取发送者 ID。假设你已经创建了一个用于检查认证的中间件，可以这样修改 POST 请求处理程序：

router.post("/requests", (req, res) => {
  const { receiver } = req.body;

  // 创建一个新的 FriendRequest 实例
  const newRequest = new FriendRequest({
    sender: req.user._id, // 从认证用户获取 sender ID
    receiver,
  });

  // 将好友请求保存到数据库
  newRequest
    .save()
    .then(() => {
      // 重定向到成功页面或发送成功响应
      res.status(204).send();
    })
    .catch((error) => {
      // 适当处理错误
      res.status(500).send("Error sending friend request");
    });
});

更简洁的做法是使用 FriendRequest.create() 方法：

FriendRequest.create({
  sender: req.user._id,
  receiver
})
.then(() => {
  // 重定向到成功页面或发送成功响应
  res.status(204).send();
})
.catch((error) => {
  // 适当处理错误
  res.status(500).send("Error sending friend request");
});

更新用户模式中的好友数组 (可选，但不推荐)

虽然可以在用户模式中显式地维护一个好友数组，但这并非总是必要的。如果需要获取好友列表，可以通过查询 FriendRequest 集合来实现。

FriendRequest
  .find({sender: req.user._id, status: 'accepted'})
  .then((listOfFriends)=>{
    console.log(listOfFriends);
  })
  .catch((e)=>{
    // 处理错误
  })

如果仍然希望在用户模式中显式地维护一个好友数组，则需要在接受好友请求时更新该数组：

router.post("/requests/:id/accept", async (req, res) => {
  const request = await FriendRequest.findById(req.params.id);
  request.status = "accepted";
  await request.save();
  const currentUserId = req.user._id; // 获取当前用户的ID
  const currentUser = await User.findById(currentUserId);
  currentUser.friends.push(request.receiver);
  await currentUser.save();
  const otherUser = await User.findById(request.receiver);
  otherUser.friends.push(currentUserId);
  await otherUser.save();
  res.redirect("/requests");
});

注意事项：

事务： 在两个不同的集合中进行更改（更新 FriendRequest 和 User）可能导致数据不一致。例如，如果服务器在更新 FriendRequest 后崩溃，但 User 未更新，则数据库将处于不一致状态。为了确保数据一致性，应该使用事务。

使用事务确保数据一致性

为了确保数据一致性，需要使用事务。Mongoose 支持事务，可以使用 startSession() 方法启动一个会话，并在该会话中执行多个操作。如果在会话中的任何操作失败，则可以回滚所有操作。

以下是如何使用事务更新 FriendRequest 和 User 集合的示例：

const mongoose = require('mongoose');

router.post("/requests/:id/accept", async (req, res) => {
  const session = await mongoose.startSession();
  session.startTransaction();

  try {
    const request = await FriendRequest.findById(req.params.id).session(session);
    request.status = "accepted";
    await request.save({ session });

    const currentUserId = req.user._id; // 获取当前用户的ID
    const currentUser = await User.findById(currentUserId).session(session);
    currentUser.friends.push(request.receiver);
    await currentUser.save({ session });

    const otherUser = await User.findById(request.receiver).session(session);
    otherUser.friends.push(currentUserId);
    await otherUser.save({ session });

    await session.commitTransaction();
    session.endSession();
    res.redirect("/requests");
  } catch (error) {
    await session.abortTransaction();
    session.endSession();
    console.error("Transaction failed:", error);
    res.status(500).send("Error accepting friend request");
  }
});

代码解释：