MongoDB聚合计算数组平均值：$function在$group中报错问题

问题背景 #

我的MongoDB集合文档结构如下：

{
    _id: 5,
    responses: [
        {
            staff: false,
            timestamp: 1000
        },
        {
            staff: true,
            timestamp: 1500
        }
    ]
}

我编写了一段$function逻辑处理responses数组，用于计算特定条件下的时间差：

const diffs=[];let current;for (let i = 0; i < responses.length; i++) {  if (i+1>=responses.length) break;  if (!current && !responses[i].staff) current = responses[i];  if (!current) continue;  const next = responses[i+1];  if (!next.staff) continue;  diffs.push(next.timestamp - current.timestamp);  current = undefined;}return diffs;

这个函数会返回类似[500,1000,10]的数值数组，也可能返回空数组[]。我需要把所有文档返回的数组合并成一个一维数组，再用$avg计算平均值。

原本计划的步骤：

1. 为每个文档计算时间差数组，得到二维数组集合；
2. 用$concatArrays合并所有数组；
3. 计算合并后数组的平均值。

但尝试在$group阶段直接使用$function时，出现错误：MongoServerError: unknown group operator '$function'，不知道怎么正确实现第一步。

解决方案 #

$group阶段的字段定义不支持直接使用$function，必须先通过$addFields或$project为每个文档生成时间差数组，再进行后续分组操作。

完整聚合代码示例 #

db.collection.aggregate([
    // 步骤1：为每个文档生成时间差数组
    {
        $addFields: {
            docDiffs: {
                $function: {
                    body: function(responses) {
                        const diffs = [];
                        let current;
                        for (let i = 0; i < responses.length; i++) {
                            if (i + 1 >= responses.length) break;
                            if (!current && !responses[i].staff) current = responses[i];
                            if (!current) continue;
                            const next = responses[i + 1];
                            if (!next.staff) continue;
                            diffs.push(next.timestamp - current.timestamp);
                            current = undefined;
                        }
                        return diffs;
                    },
                    args: ["$responses"],
                    lang: "js"
                }
            }
        }
    },
    // 步骤2：收集所有文档的时间差数组为二维数组
    {
        $group: {
            _id: null, // 全局分组，合并所有文档结果
            allDiffs: { $push: "$docDiffs" }
        }
    },
    // 步骤3：将二维数组展平为一维数组
    {
        $addFields: {
            flattenedDiffs: {
                $reduce: {
                    input: "$allDiffs",
                    initialValue: [],
                    in: { $concatArrays: ["$$value", "$$this"] }
                }
            }
        }
    },
    // 步骤4：计算平均值
    {
        $project: {
            avgDiff: { $avg: "$flattenedDiffs" },
            _id: 0
        }
    }
])

关键细节说明 #

• 第一步的正确实现：用$addFields为每个文档新增docDiffs字段，存储当前文档的时间差数组，这一步可以正常使用$function；
• 合并数组：通过$group+$push把所有文档的docDiffs收集成一个二维数组，再用$reduce配合$concatArrays将其展平为一维数组（直接$concatArrays无法处理多个数组的批量合并）；
• 性能优化：如果用$unwind两次展平数组也能实现，但$reduce的方式更高效，避免了多次展开文档的性能损耗。

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