响应式Spring Boot Webflux中MongoDB Aggregation的$in操作不生效问题

问题根因 #

你遇到的$in条件失效问题是Spring Data MongoDB针对带@MongoId注解的字段在聚合查询中的类型转换规则导致的，核心问题有两点：

1. 你手动将UUID转为字符串列表后，TypedAggregation绑定了User实体类，MongoConverter识别到_id对应实体的UUID类型id字段，会自动将你传入的字符串id列表二次转换为UUID类型，和数据库中存储的String类型_id无法匹配
2. 聚合查询中针对主键字段查询时，在TypedAggregation场景下应使用实体类的字段名id而非数据库原生字段名_id，避免映射规则冲突

修复方案 #

方案一：关闭类型映射，直接使用原生字段查询 #

直接指定查询集合名称，不绑定实体类的类型映射，避免自动类型转换干扰：

public Flux<User> findAllWithCompany(List<UUID> users, long skip, long limit, Sort.Direction direction) {
    List<AggregationOperation> aggs = new ArrayList<>();
    List<String> idsAsString = users.stream().map(UUID::toString).collect(Collectors.toList());
    aggs.add(match(Criteria.where("_id").in(idsAsString)));
    aggs.add(sort(direction, Fields.CREATED_ON.getValue()));
    aggs.add(Aggregation.skip(skip));
    aggs.add(Aggregation.limit(limit));
    // 创建普通聚合，直接指定集合名称，不做实体类型映射
    Aggregation aggregation = Aggregation.newAggregation(aggs);
    return mongoTemplate.aggregate(aggregation, "Users", User.class);
}

方案二：使用实体字段名构造查询，不手动转换UUID #

利用Spring Data的自动映射能力，直接传入UUID列表，用实体字段名id构造Criteria：

public Flux<User> findAllWithCompany(List<UUID> users, long skip, long limit, Sort.Direction direction) {
    List<AggregationOperation> aggs = new ArrayList<>();
    // 直接使用实体字段名id，传入原生UUID列表，由Spring自动完成到数据库String类型的映射
    aggs.add(match(Criteria.where("id").in(users)));
    aggs.add(sort(direction, Fields.CREATED_ON.getValue()));
    aggs.add(Aggregation.skip(skip));
    aggs.add(Aggregation.limit(limit));
    TypedAggregation<User> aggregation = Aggregation.newAggregation(User.class, aggs);
    return mongoTemplate.aggregate(aggregation, User.class);
}

验证方法 #

可以开启MongoDB查询日志，打印Spring实际执行的聚合语句，对比和你手动执行的命令行语句的差异，确认$in条件的参数类型是否符合预期：
在application配置中添加：

logging:
  level:
    org.springframework.data.mongodb.core.MongoTemplate: DEBUG

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