使用 Mongoose 更新用户的好友列表：最佳实践指南-js教程-PHP中文网

使用 Mongoose 更新用户的好友列表：最佳实践指南

心靈之曲

发布： 2025-09-28 17:09:14

原创

328人浏览过

使用 mongoose 更新用户的好友列表：最佳实践指南

本文旨在指导开发者如何在使用 Mongoose 构建社交应用时，正确地更新用户的好友列表。文章将探讨如何安全有效地处理好友请求的接受流程，并讨论维护用户好友列表的不同策略，包括直接在 User Schema 中维护以及通过查询 FriendRequest Schema 间接获取。同时，本文将强调数据一致性的重要性，并建议使用事务来确保数据库操作的原子性。

在构建社交应用时，管理用户之间的好友关系是一个常见的需求。本文将深入探讨如何使用 Mongoose 在用户接受好友请求后，更新双方用户的好友列表。我们将讨论安全性、数据一致性以及不同的实现策略。

确保请求发送者的身份验证

首先，务必确保好友请求的发送者身份已通过验证。不要依赖客户端发送的 sender ID，因为这可能被篡改。使用服务器端已经验证过的用户 ID，通常可以通过中间件获取。

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

  FriendRequest.create({
    sender: req.user._id, // 从经过身份验证的用户获取 sender ID
    receiver,
  })
    .then(() => {
      res.status(204).send();
    })
    .catch((error) => {
      res.status(500).send("Error sending friend request");
    });
});

登录后复制

维护好友列表的策略：直接 vs. 间接

有两种主要策略来维护好友列表：

直接方法： 在 User Schema 中维护一个 friends 数组，存储好友的 ObjectId。
间接方法： 依赖 FriendRequest Schema，通过查询 status 为 "accepted" 的请求来获取好友列表。

间接方法：查询 FriendRequest Schema

如果采用间接方法，你无需在 User Schema 中维护 friends 数组。你可以通过查询 FriendRequest 集合来动态获取好友列表。

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

登录后复制

这种方法的优点是避免了在多个地方更新数据，降低了数据不一致的风险。

直接方法：更新 User Schema 中的 friends 数组

如果选择在 User Schema 中维护 friends 数组，需要在接受好友请求时更新双方用户的 friends 数组。

堆友

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

306

查看详情

router.post("/requests/:id/accept", async (req, res) => {
  try {
    const request = await FriendRequest.findById(req.params.id);
    if (!request) {
      return res.status(404).send("Friend request not found");
    }

    request.status = "accepted";
    await request.save();

    const currentUser = await User.findById(request.receiver);
    const friendUser = await User.findById(request.sender);

    if (!currentUser || !friendUser) {
      return res.status(404).send("User not found");
    }

    currentUser.friends.push(request.sender);
    friendUser.friends.push(request.receiver);

    await currentUser.save();
    await friendUser.save();

    res.redirect("/requests");
  } catch (error) {
    console.error(error);
    res.status(500).send("Error accepting friend request");
  }
});

登录后复制

重要注意事项：使用事务

当同时更新多个集合（例如 FriendRequest 和 User）时，必须使用事务来保证数据的一致性。如果服务器在更新过程中崩溃，事务可以回滚所有更改，防止数据损坏。

以下是如何使用 Mongoose 事务的示例：

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);
        if (!request) {
            await session.abortTransaction();
            session.endSession();
            return res.status(404).send("Friend request not found");
        }

        request.status = "accepted";
        await request.save({ session });

        const currentUser = await User.findById(request.receiver).session(session);
        const friendUser = await User.findById(request.sender).session(session);

        if (!currentUser || !friendUser) {
            await session.abortTransaction();
            session.endSession();
            return res.status(404).send("User not found");
        }

        currentUser.friends.push(request.sender);
        friendUser.friends.push(request.receiver);

        await currentUser.save({ session });
        await friendUser.save({ session });

        await session.commitTransaction();
        session.endSession();

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

登录后复制

代码解释: