AWS DocDB 4.0：如何在文档缺失字段时正确分组过滤数据？

问题分析 #

原聚合管道通过$oldAccount字段分组识别关联账户组，但当旧账户文档缺失oldAccount字段时，该文档会被分到_id: null的独立组，无法与指向它的新账户（含oldAccount字段）归为同一组，导致CLOSED状态的旧账户无法被正确过滤。

解决方案 #

核心思路是统一关联组的分组键：对无oldAccount的文档，用自身_id作为分组键；有oldAccount的文档，用oldAccount作为分组键。这样关联的新旧账户会被归为同一组，后续过滤逻辑即可正常生效。

修改后的聚合管道如下：

db.getCollection("subscription").aggregate([
    // 筛选当前用户的订阅
    { $match: { user: "self" } },
    // 保留所需字段（原业务字段保留逻辑不变）
    {
        $project: {
            _id: 1,
            accountState: 1,
            oldAccount: 1,
            // 其他业务字段...
        }
    },
    // 生成统一分组键：存在oldAccount则用它，否则用自身_id
    {
        $addFields: {
            groupKey: { $ifNull: ["$oldAccount", "$_id"] }
        }
    },
    // 按统一分组键聚合关联账户组
    { 
        $group: { 
            _id: "$groupKey", 
            relationshipCnt: { $sum: 1 }, 
            tmp: { $push: "$$ROOT" } 
        }
    },
    { $unwind: { path: "$tmp", preserveNullAndEmptyArrays: true } },
    { $replaceRoot: { newRoot: { $mergeObjects: ["$tmp", { relationshipCnt: "$relationshipCnt" }] } } },
    // 过滤逻辑：保留非CLOSED状态，或CLOSED状态但组内仅自身（无关联有效账户）
    {
        $match: {
            $or: [
                { accountState: { $ne: "CLOSED" } },
                { 
                    accountState: { $regex: /CLOSED/, $options: "i" }, 
                    relationshipCnt: { $eq: 1 } 
                }
            ]
        }
    },
    // 可选：移除临时生成的groupKey字段
    { $project: { groupKey: 0 } }
]);

关键修改说明 #

1. 新增$addFields生成groupKey：通过$ifNull函数实现分组键统一，确保关联账户被归入同一组。
2. 调整$group的_id为$groupKey：替换原有的$oldAccount分组逻辑，解决缺失字段导致的分组分散问题。
3. 保留原有过滤逻辑：当组内存在非CLOSED账户时，CLOSED账户因relationshipCnt>1会被排除；仅当组内只有CLOSED账户时，才会保留该记录。

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