如何判断Mongoose的$or查询触发了哪个匹配条件？

嘿，这个问题我之前做注册功能时也碰到过，完全不用多跑两次查询，有两种靠谱的方法能搞定！

方法一：查询后直接对比字段 #

这是最直接简单的方式，拿到查询结果后，把返回用户的字段和你传入的参数做对比，就能精准判断匹配的是哪个条件，甚至还能处理「邮箱和用户名同时被占用」的极端情况。

代码示例：

User.findOne({$or:[ {'local.email':email}, {'local.name': req.body.name} ]}, function(err, existingUser) {
  if (err) { 
    // 这里处理数据库查询错误
    return;
  }
  if (existingUser) {
    let tipMessage = '';
    // 判断是否匹配邮箱条件
    if (existingUser.local.email === email) {
      tipMessage += '邮箱已被占用';
    }
    // 判断是否匹配用户名条件
    if (existingUser.local.name === req.body.name) {
      tipMessage = tipMessage ? `${tipMessage}，` : '';
      tipMessage += '用户名已被占用';
    }
    // 把tipMessage返回给前端即可
    console.log(tipMessage);
  } else {
    // 没有匹配项，可以执行注册逻辑
  }
});

方法二：用聚合查询标记匹配类型 #

如果你想在数据库查询阶段就直接得到匹配类型，可以用MongoDB的聚合管道，通过$addFields和$switch给结果添加一个标记字段，拿到结果后直接判断这个字段就行。

代码示例：

User.aggregate([
  {
    $match: {
      $or: [ {'local.email': email}, {'local.name': req.body.name} ]
    }
  },
  {
    $addFields: {
      matchedBy: {
        $switch: {
          branches: [
            { 
              case: { $and: [ { $eq: ['$local.email', email] }, { $eq: ['$local.name', req.body.name] } ] }, 
              then: 'both' 
            },
            { case: { $eq: ['$local.email', email] }, then: 'email' },
            { case: { $eq: ['$local.name', req.body.name] }, then: 'name' }
          ],
          default: 'none'
        }
      }
    }
  }
])
.then(results => {
  if (results.length > 0) {
    const existingUser = results[0];
    switch(existingUser.matchedBy) {
      case 'email':
        console.log('邮箱已被占用');
        break;
      case 'name':
        console.log('用户名已被占用');
        break;
      case 'both':
        console.log('邮箱和用户名都已被占用');
        break;
    }
  } else {
    // 没有匹配项，执行注册逻辑
  }
})
.catch(err => {
  // 处理聚合查询错误
});

要不要执行两次查询？ #

完全没必要！两次查询会额外增加数据库负载，而且上面两种方法已经能完美解决问题，效率更高。除非你遇到了极端到几乎不可能发生的场景（比如查询结果返回后，用户数据在极短时间内被修改），否则根本不需要多跑一次查询。

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