MongoDB聚合管道中如何根据动态status值获取对应statusLog日期?
问题
我有如下结构的MongoDB文档:
{ "status": "in_progress", "statusLog": { "received": "2010-01-01", "in_progress": "2011-02-02" } }
需要将其转换为以下格式:
{ "status": "in_progress", "date": "2011-02-02" }
其中status的值是动态可变的,对应的日期存储在同文档的statusLog属性中,属性名和status的值一致。
我已经有一条较长的聚合管道,希望在末尾添加$addFields阶段来新增date字段,但找不到合适的操作符。预期代码结构如下:
collection.aggregate([ { $addFields: { "date": {} // 此处应使用什么MongoDB操作符? } } ])
补充背景:之前status值是固定的RECEIVED、IN_PROGRESS、CLOSED,我用$switch实现,但现在status可以是任意内容,原代码如下:
$addFields: { statusDate: { $switch: { branches: [ { case: { $eq: ['$status', 'RECEIVED'] }, then: { $ifNull: ['$statusLog.RECEIVED.datetime', null] }, }, { case: { $eq: ['$status', 'IN_PROGRESS'] }, then: { $ifNull: ['$statusLog.IN_PROGRESS.datetime', null] }, }, { case: { $eq: ['$status', 'CLOSED'] }, then: { $ifNull: ['$statusLog.CLOSED.datetime', null] }, }, ], default: null, }, }, },
解决方案
用$objectToArray+$filter组合就能实现动态匹配,无需枚举所有可能的status值,具体实现如下:
核心代码
collection.aggregate([ { $addFields: { "date": { $ifNull: [ { $getField: { field: "v", input: { $first: { $filter: { input: { $objectToArray: "$statusLog" }, cond: { $eq: ["$$this.k", "$status"] } } } } } }, null ] } } } ])
逻辑说明
$objectToArray:将statusLog对象转为键值对数组,格式为[{k: 键名, v: 对应值}, ...]。$filter:遍历数组,只保留键名k与当前文档status值相等的元素。$first:取筛选后的第一个元素(因键名唯一,只会有一个匹配项)。$getField:提取该元素的v字段,即目标日期。$ifNull:若status在statusLog中无对应键,返回null避免空值报错。
适配嵌套日期场景(对应原代码的.datetime)
如果statusLog中的日期是嵌套结构,只需修改$getField的field参数:
"date": { $ifNull: [ { $getField: { field: "v.datetime", input: { $first: { $filter: { input: { $objectToArray: "$statusLog" }, cond: { $eq: ["$$this.k", "$status"] } } } } } }, null ] }
内容的提问来源于stack exchange,提问作者Pipe
相关产品推荐
相关产品推荐

