本文旨在指导开发者如何在使用 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 数组。
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"); } });
代码解释:
- mongoose.startSession(): 启动一个新的 Mongoose 会话，用于事务管理。
- session.startTransaction(): 开始事务。
- .session(session): 将会话传递给所有 Mongoose 操作 (例如 findById, save)。
- session.commitTransaction(): 提交事务，将所有更改保存到数据库。
- session.abortTransaction(): 如果出现错误，中止事务，回滚所有更改。
- session.endSession(): 结束会话。
总结
管理用户好友关系需要仔细考虑安全性、数据一致性和性能。 使用经过身份验证的用户 ID，选择适合你的应用需求的维护策略（直接或间接），并始终使用事务来保证数据一致性，这些都是至关重要的。 通过遵循这些最佳实践，你可以构建一个健壮且可靠的社交应用。