NodeJS+MongoDB注册时高效校验用户名/邮箱是否存在的方法

你当前的两次独立查询实现确实存在效率问题，并非该场景的标准写法，完全可以通过MongoDB的查询运算符合并为单次数据库请求。

优化方案 #

使用$or运算符，一次查询匹配「用户名与提交值一致」或「邮箱与提交值一致」的用户记录，将两次串行查询的耗时压缩为单次查询耗时，代码实现如下：

// 单次DB查询完成用户名、邮箱存在性校验
try {
    const existingUser = await User.findOne({
        $or: [
            { username: req.body.username },
            { email: req.body.email }
        ]
    });

    if (existingUser) {
        const usernameConflict = existingUser.username === req.body.username;
        const emailConflict = existingUser.email === req.body.email;

        if (usernameConflict && emailConflict) return res.status(400).send("Username and email already taken");
        if (emailConflict) return res.status(400).send("Email already taken");
        if (usernameConflict) return res.status(400).send("Username already taken");
    }
    // 无冲突，继续执行后续密码加密、写入用户数据等注册流程
} catch (e) {
    return res.status(500).send("Error querying DB to check if username/email exists or not");
}

额外注意事项 #

• 业务层的查询校验无法完全规避并发场景下的竞态问题：如果两个注册请求同时到达，都在查询步骤判定无冲突，就会出现重复写入的问题。
• 最稳妥的实现是在数据库层给username、email字段分别创建唯一索引，从底层阻止重复值写入：

// Mongoose Schema中配置唯一索引
const userSchema = new mongoose.Schema({
    username: {
        type: String,
        required: true,
        unique: true // 添加唯一索引约束
    },
    email: {
        type: String,
        required: true,
        unique: true // 添加唯一索引约束
    },
    // 其余账号字段...
});
const User = mongoose.model('User', userSchema);

配置唯一索引后，即使并发请求绕过了业务层校验，写入重复数据时MongoDB会直接抛出重复键错误，你只需要在写入逻辑中捕获该错误返回对应提示即可，彻底避免脏数据产生。

内容的提问来源于stack exchange，提问作者Matthew Trent