Spring Boot+MongoTemplate多条件批量读取Mongo记录并关联条件归属问询

我之前做项目时刚好碰到过一模一样的需求，你说的这两个问题确实是MongoTemplate里的常见痛点——OR组合没法溯源条件，BulkOperations又只支持写操作。给你两个实用的方案：

方案一：分条件查询后手动标记（直观易实现） #

这个思路很直接：把每个条件单独执行find查询，给每批查询结果打上对应条件的标记，最后合并所有结果。这样你能清晰知道每条记录满足哪个条件。

举个代码例子：

假设你有个实体类User，还有个DTO用来封装带条件标记的结果：

@Data
public class UserWithCondition {
    private User user;
    // 用来标记满足的条件，比如用枚举或者字符串
    private String conditionTag;
}

然后处理多个条件：

// 定义你的多个查询条件
List<Criteria> conditions = Arrays.asList(
    Criteria.where("age").gt(30),
    Criteria.where("city").is("Beijing"),
    Criteria.where("role").is("ADMIN")
);

List<UserWithCondition> resultList = new ArrayList<>();

for (int i = 0; i < conditions.size(); i++) {
    Criteria criteria = conditions.get(i);
    // 执行单个条件查询
    List<User> users = mongoTemplate.find(Query.query(criteria), User.class);
    // 给每条记录打上对应条件的标记
    String tag = "condition_" + (i + 1); // 或者用更有意义的标签，比如"age_gt_30"
    users.forEach(user -> {
        UserWithCondition dto = new UserWithCondition();
        dto.setUser(user);
        dto.setConditionTag(tag);
        resultList.add(dto);
    });
}

这个方案的优点是逻辑简单，容易调试；缺点是如果条件很多，会发起多次MongoDB请求，性能上可能不如单次聚合查询。

方案二：用Aggregation聚合查询（高效推荐） #

如果你的条件数量较多，或者追求更好的性能，推荐用MongoDB的聚合管道来实现——通过$addFields给每条满足条件的记录加上标记，一次查询就能拿到所有带标记的结果。

代码示例：

// 定义条件和对应的标签映射
Map<Criteria, String> conditionTagMap = new HashMap<>();
conditionTagMap.put(Criteria.where("age").gt(30), "age_gt_30");
conditionTagMap.put(Criteria.where("city").is("Beijing"), "city_beijing");
conditionTagMap.put(Criteria.where("role").is("ADMIN"), "role_admin");

// 构建$addFields阶段的表达式：判断记录满足哪个条件，给conditionTag赋值
AggregationExpression conditionTagExpr = new AggregationExpression() {
    @Override
    public Document toDocument(AggregationOperationContext context) {
        List<Document> caseClauses = new ArrayList<>();
        // 给每个条件添加一个$case分支
        for (Map.Entry<Criteria, String> entry : conditionTagMap.entrySet()) {
            Document whenClause = entry.getKey().getCriteriaObject();
            caseClauses.add(new Document("when", whenClause).append("then", entry.getValue()));
        }
        // 默认值可以设为null或者其他标识，表示不满足任何条件（不过我们后面会过滤掉这些）
        Document defaultClause = new Document("default", null);
        return new Document("$switch", new Document("branches", caseClauses).append("default", defaultClause));
    }
};

// 构建聚合管道：先加标记，再过滤掉不满足任何条件的记录
Aggregation aggregation = Aggregation.newAggregation(
    Aggregation.addFields().addFieldWithValue("conditionTag", conditionTagExpr).build(),
    Aggregation.match(Criteria.where("conditionTag").ne(null))
);

// 执行聚合查询，结果映射到带标记的DTO（注意要让DTO包含conditionTag字段）
AggregationResults<UserWithCondition> results = mongoTemplate.aggregate(aggregation, "user_collection", UserWithCondition.class);
List<UserWithCondition> resultList = results.getMappedResults();

这个方案只需要一次MongoDB请求，性能更优，适合条件较多的场景。需要注意的是，如果有记录同时满足多个条件，$switch会返回第一个匹配的标签——如果你的业务允许一条记录对应多个条件，可以把$switch改成$filter或者$map来收集所有满足的条件标签，不过这种情况比较少见。

补充说明 #

• 你提到的BulkOperations确实只支持批量写操作（insert、update、delete），没有批量读取的相关方法，所以不用在这上面浪费时间啦。
• OR组合查询之所以没法关联条件，是因为MongoDB返回的结果不会携带匹配的条件信息，必须通过我们上面的方式手动标记。

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