本文旨在指导开发者如何在 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");
});注意事项：
部分功能简介：商品收藏夹功能热门商品最新商品分级价格功能自选风格打印结算页面内部短信箱商品评论增加上一商品，下一商品功能增强商家提示功能友情链接用户在线统计用户来访统计用户来访信息用户积分功能广告设置用户组分类邮件系统后台实现更新用户数据系统图片设置模板管理CSS风格管理申诉内容过滤功能用户注册过滤特征字符IP库管理及来访限制及管理压缩，恢复，备份数据库功能上传文件管理商品类别管理商品添加/修改/
- 事务： 在两个不同的集合中进行更改（更新 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 中实现用户添加好友的功能。记住，数据一致性和安全性是至关重要的，因此请务必使用事务并验证用户输入。
