本文旨在指导开发者如何在 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");
});注意事项：
HISUI 前端框架基础开发指南。HISUI 是基于 easyui 二次开发的 jQuery UI 组件库，专为医疗信息系统设计。当涉及 HISUI 组件开发、jQuery easyui 表格/表单/布局/窗口、医疗系统前端页面编写时使用此 skill。 涵盖 40 个组件的官方文档参考，支持声明式与编程式两种初始化方式，提供 $cm/$m 数据访问方法。
- 事务： 在两个不同的集合中进行更改（更新 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");
}
});代码解释：
- 启动会话： const session = await mongoose.startSession(); 启动一个新的 Mongoose 会话。
- 开始事务： session.startTransaction(); 开始一个事务。
- 执行操作： 在 try 块中，执行所有数据库操作。关键是使用 .session(session) 将会话与每个操作关联起来。
- 提交事务： await session.commitTransaction(); 如果所有操作都成功，则提交事务，将所有更改保存到数据库。
- 中止事务： await session.abortTransaction(); 如果任何操作失败，则中止事务，回滚所有更改。
- 结束会话： session.endSession(); 结束会话。
- 错误处理： 在 catch 块中，处理错误并中止事务。
总结：
- 为了确保数据一致性，建议使用事务。
- 在用户模式中显式维护好友数组可能不是最佳做法。可以通过查询 FriendRequest 集合来获取好友列表。
- 始终从已认证的用户信息中获取发送者 ID，以提高安全性。
通过以上步骤，你可以安全且有效地在 Mongoose 中实现用户添加好友的功能。记住，数据一致性和安全性是至关重要的，因此请务必使用事务并验证用户输入。
