如何在MongoDB aggregate聚合查询中使用JavaScript对象映射字段值

你可以通过以下两种方式实现需求，根据你使用的MongoDB版本选择即可：

方案1：MongoDB 5.0及以上版本（推荐） #

直接使用$literal将外部JS对象传入聚合上下文，再通过$getField根据tId取值即可，修改后的聚合语句如下：

const obj = {'id_1': 'data_1', 'id_2': 'data_2'/* 其余键值对 */}

models.dummy.aggregate(
  [
    { $match: findQuery },
    {
      $project: {
        id: 1,
        tId: 1,
        quantity: 1,
        status: 1,
        notes: {
          // 若tId不存在对应值可加$ifNull设置默认值，示例默认设为null
          $ifNull: [
            { $getField: { input: { $literal: obj }, field: "$tId" } },
            null
          ]
        }
      }
    }
  ]);

• 原理说明：$literal会直接把传入的JS对象作为聚合运算的常量值，不会解析其中的字段表达式，$getField可以根据指定的字段名（这里就是文档的tId值）从对象中取出对应值。

方案2：MongoDB 5.0以下版本 #

低版本不支持$getField，可以把外部对象的键值对转为$switch的分支判断逻辑实现匹配，代码示例如下：

const obj = {'id_1': 'data_1', 'id_2': 'data_2'/* 其余键值对 */}
// 生成switch的分支规则
const switchBranches = Object.entries(obj).map(([tId, noteVal]) => ({
  case: { $eq: ["$tId", tId] },
  then: noteVal
}))

models.dummy.aggregate(
  [
    { $match: findQuery },
    {
      $project: {
        id: 1,
        tId: 1,
        quantity: 1,
        status: 1,
        notes: {
          $switch: {
            branches: switchBranches,
            default: null // 未匹配到对应tId时的默认值
          }
        }
      }
    }
  ]);

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