如何使用MongoDB聚合查询生成指定编码的库存可用性状态？

问题1：仅用一条aggregate语句是否可以实现该需求？ #

可以，完全可以通过单条MongoDB聚合管道实现，核心思路是先把待查询的编码列表注入每个文档，保证每个待查询编码都能生成对应条目，不会因为原文档没有对应库存记录而丢失。

问题2：pymongo开发场景下，数据库端还是代码端处理更优？ #

可根据实际业务场景选择：

• 集合数据量较大（万条以上）：优先选择数据库端处理，可大幅减少MongoDB到应用端的数据传输量，避免拉取全量库存数据到本地遍历，降低网络开销和应用端内存占用。
• 集合数据量很小（千条以内）、且查询逻辑频繁变动：优先选择代码端处理，修改业务判断逻辑不需要调整聚合语句，开发调试成本更低。

问题3：现有聚合语句如何修改？ #

你原来的写法问题在于：先$unwind库存再match的逻辑，会直接过滤掉原文档中不存在的编码记录，没法生成缺省的out of stock条目。注意你存储的Stock.Code是字符串类型，待查询的数字列表[1,10]要先转成字符串["1","10"]避免类型不匹配，正确的管道逻辑如下：

// 你的查询编码列表，和Stock.Code类型保持一致
const queryCodes = ["1", "10"];
db.Collection.aggregate([
  // 步骤1：将待查询编码列表注入每个文档
  {
    "$addFields": {
      "QueryCode": queryCodes
    }
  },
  // 步骤2：展开查询编码，保证每个编码对应一条待处理记录
  {
    "$unwind": "$QueryCode"
  },
  // 步骤3：匹配当前查询编码对应的库存条目，计算可用性
  {
    "$project": {
      "_id": 0,
      "Id": 1,
      "Code": "$QueryCode",
      "Availability": {
        "$let": {
          "vars": {
            // 找到Stock中Code和当前查询编码一致的条目
            "matchedStock": {
              "$first": {
                "$filter": {
                  "input": "$Stock",
                  "cond": {"$eq": ["$$this.Code", "$QueryCode"]}
                }
              }
            }
          },
          "in": {
            "$cond": [
              // 条件：匹配到库存 且 库存数量>=3
              {"$and": [{"$ne": ["$$matchedStock", null]}, {"$gte": ["$$matchedStock.Qty", 3]}]},
              "in stock",
              "out of stock"
            ]
          }
        }
      }
    }
  }
])

如果你使用的MongoDB版本低于4.4，不支持$first操作符，可以把$first替换为$arrayElemAt，写法如下：

"matchedStock": {
  "$arrayElemAt": [
    {
      "$filter": {
        "input": "$Stock",
        "cond": {"$eq": ["$$this.Code", "$QueryCode"]}
      }
    },
    0
  ]
}

pymongo中直接把queryCodes作为Python列表传入，把上述JS语法替换为Python字典格式即可直接运行。

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