如何在Spring Data MongoDB中实现带$reduce的$addFields聚合？

问题原因 #

在Spring Data MongoDB中构建$reduce操作时，如果没有通过聚合表达式API正确定义in阶段的计算逻辑，而是直接传入简单值/字符串，就会导致生成的聚合语句仅保留initialValue，丢失核心的in逻辑。

原生MongoDB聚合示例（参考） #

假设你要实现的原生聚合逻辑如下：

[
  {
    "$addFields": {
      "totalAmount": {
        "$reduce": {
          "input": "$orders",
          "initialValue": 0,
          "in": { "$add": ["$$value", "$$item.amount"] }
        }
      }
    }
  }
]

正确的Spring Data MongoDB Kotlin实现 #

方式一：分步构建（清晰直观） #

import org.springframework.data.mongodb.core.aggregation.Aggregation
import org.springframework.data.mongodb.core.aggregation.ReduceOperation
import org.springframework.data.mongodb.core.aggregation.Variables
import org.springframework.data.mongodb.core.aggregation.AggregationExpression

// 定义$reduce的in计算逻辑：累加$$value和$$item.amount
val reduceInExpr = AggregationExpression { context ->
    context
        .operator("$add")
        .add(Variables.VALUE) // 对应$$value
        .add(Variables.ITEM.dot("amount")) // 对应$$item.amount
        .toDocument()
}

// 构建完整的$reduce操作
val reduceOp = ReduceOperation.builder()
    .input("orders") // 指定输入数组字段
    .initialValue(0) // 初始值
    .`in`(reduceInExpr) // 关键：传入定义好的in表达式
    .build()

// 构建$addFields阶段
val addFieldsStage = Aggregation.addFields()
    .field("totalAmount")
    .withValue(reduceOp)
    .build()

// 组装完整聚合管道
val aggregation = Aggregation.newAggregation(addFieldsStage)

方式二：简洁链式写法 #

利用Spring Data提供的静态工具方法，可大幅简化代码：

import org.springframework.data.mongodb.core.aggregation.Aggregation.*
import org.springframework.data.mongodb.core.aggregation.Expressions.*

val aggregation = newAggregation(
    addFields()
        .field("totalAmount")
        .withValue(
            reduce(
                "orders", // input字段
                0, // initialValue
                add(variable("value"), field("$$item.amount")) // in阶段的累加逻辑
            )
        )
        .build()
)

关键注意事项 #

1. 必须用聚合表达式定义in逻辑：不能直接给.in()传入数字、字符串等简单值，必须通过AggregationExpression或Expressions工具类（如add()、multiply()）构建计算逻辑。
2. 正确引用变量：$$value对应Variables.VALUE或variable("value")，$$item对应Variables.ITEM或field("$$item.xxx")。
3. 验证生成的聚合语句：可通过aggregation.toMongoDocument()打印出生成的JSON结构，确认$reduce是否包含完整的input、initialValue和in字段。

错误写法示例（避免） #

以下写法会丢失in逻辑，因为直接给.in()传入了固定值：

// 错误示例：in阶段用固定值代替表达式
val wrongReduceOp = ReduceOperation.builder()
    .input("orders")
    .initialValue(0)
    .`in`(0) // 此处错误，应传入AggregationExpression
    .build()

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