在 Mongoose 中实现用户添加好友功能-js教程-PHP中文网

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

花韻仙語

发布： 2025-09-28 16:33:14

原创

933人浏览过

在 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");
});

登录后复制

注意事项：

堆友

Alibaba Design打造的设计师全成长周期服务平台，旨在成为设计师的好朋友

306

查看详情

事务： 在两个不同的集合中进行更改（更新 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");
  }
});

登录后复制

代码解释：