如何在Spring Data MongoDB聚合管道中组合Document与AggregationOperation？

完全可以将手动构建的Document类型聚合阶段包装为AggregationOperation，与Spring Data MongoDB内置的聚合构建器生成的阶段无缝拼接使用，无需修改原有聚合执行逻辑。

具体实现方式 #

• 函数式接口包装（推荐，适用于Spring Data MongoDB 3.2+版本）
  AggregationOperation本身是函数式接口，你不需要编写专门的实现类，直接通过lambda表达式返回自定义构造的阶段Document即可。以带子查询逻辑的$lookup阶段为例：

// 手动构建符合MongoDB语法的自定义聚合阶段
Document customLookupDoc = new Document("$lookup",
        new Document("from", "orders")
                .append("let", new Document("uid", "$_id"))
                .append("pipeline", List.of(
                        new Document("$match", new Document("$expr",
                                new Document("$eq", List.of("$user_id", "$$uid"))
                        ))
                ))
                .append("as", "orderList")
);

// 一行代码包装为AggregationOperation
AggregationOperation customLookupOp = context -> customLookupDoc;

// 和内置构建器生成的阶段自由组合，可放在管道任意位置
Aggregation aggregation = Aggregation.newAggregation(
        Aggregation.match(Criteria.where("account_status").is(1)),
        customLookupOp,
        Aggregation.project().andExclude("password"),
        Aggregation.sort(Sort.by(Sort.Direction.DESC, "create_time"))
);

// 按原有方式执行聚合即可
List<UserVO> result = mongoTemplate.aggregate(aggregation, "users", UserVO.class).getMappedResults();

• 匿名类实现（兼容低版本）
  如果使用的是3.2之前的旧版本，无法用lambda写法，可以直接创建匿名实现类重写转换方法：

AggregationOperation customLookupOp = new AggregationOperation() {
    @Override
    public Document toDocument(AggregationOperationContext context) {
        return customLookupDoc;
    }
};

注意：自定义构建的Document需要严格遵循MongoDB的聚合阶段语法，Spring Data MongoDB不会对自定义阶段做语法校验，语法错误会在执行时由MongoDB服务端直接抛出异常。

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