MongoDB+Spring Data嵌套数组匹配查询及结果返回问题

你的问题核心是既要保留完整的群组文档结构，又要精准过滤嵌套数组中的元素，同时避免无匹配时返回null。当前的实现用了两次MatchOperation，这是问题所在：第二个Match会直接过滤掉整个文档（如果没有符合条件的toys），而且它不会修改数组内容，只是判断文档是否存在匹配元素，完全达不到你要的数组过滤效果。

正确思路：用$map + $filter聚合操作处理嵌套数组 #

不需要用unwind（拆分数组后还要合并，反而复杂），直接通过投影（Project）阶段，利用MongoDB的$map遍历外层数组，结合$filter筛选内层数组元素，同时保留原文档的所有结构。这样即使没有符合条件的toys，也会返回空数组而非null，完整保留群组基础信息。

完整代码实现 #

// 第一步：获取当前用户的入群时间（确保用户在群组内）
Aggregation getUserAgg = newAggregation(
    match(Criteria.where("_id").is(groupId).and("groupUsers.userId").is(userId))
);
GroupUser groupUser = mongoTemplate.aggregate(getUserAgg, Group.class, GroupUser.class).getUniqueMappedResult();

if (groupUser == null) {
    // 处理用户不在该群组的情况，比如返回提示或空对象
    return null;
}

// 第二步：构建聚合查询，过滤嵌套的toys数组
MatchOperation matchTargetGroup = match(Criteria.where("_id").is(groupId));

// 投影阶段：遍历groupUsers，保留原用户信息，同时过滤toys数组
ProjectOperation filterToysProject = project()
    // 明确列出需要保留的顶层字段（比如_id、name，有其他字段都要加上）
    .andInclude("_id", "name")
    .and(AggregationExpressions.map("groupUsers", "user",
        // 合并原用户字段与过滤后的toys数组
        AggregationExpressions.mergeObjects(
            AggregationExpressions.field("user"),
            // 筛选toys中createdAt >= 用户入群时间的元素
            AggregationExpressions.filter("user.toys", "toy",
                AggregationExpressions.gte(AggregationExpressions.field("toy.createdAt"), groupUser.getCreatedAt())
            ).as("toys")
        )
    )).as("groupUsers");

// 执行聚合
Aggregation finalAggregation = newAggregation(matchTargetGroup, filterToysProject);
AggregationResults<Group> result = mongoTemplate.aggregate(finalAggregation, Group.class, Group.class);

// 即使没有匹配的toys，也会返回完整群组文档（toys数组为空）
Group group = result.getUniqueMappedResult();
if (group == null) {
    // 处理群组不存在的情况
    return new Group(); // 或者按业务需求返回默认对象
}
return group;

代码说明 #

1. $map遍历外层数组：遍历groupUsers中的每个用户元素，保留原有的所有字段（比如name、createdAt等）。
2. $filter筛选内层数组：对每个用户的toys数组，只保留createdAt大于等于用户入群时间的元素；如果没有符合条件的元素，会返回空数组而非删除用户或整个文档。
3. $mergeObjects合并字段：确保用户的其他原始字段不会丢失，仅替换toys为过滤后的数组。
4. 避免返回null：即使所有toys都不符合条件，最终仍会返回完整的群组文档，只是对应的toys数组为空；如果groupId不存在，可以手动返回默认的Group对象。

旧版Spring Data兼容写法（如果不支持Lambda表达式） #

如果你的Spring Data MongoDB版本较低（比如2.x），可以用DBObject手动构建聚合表达式：

ProjectOperation filterToysProject = project()
    .andInclude("_id", "name")
    .and(new AggregationExpression() {
        @Override
        public DBObject toDbObject(AggregationOperationContext context) {
            return new BasicDBObject("$map", new BasicDBObject("input", "$groupUsers")
                .append("as", "user")
                .append("in", new BasicDBObject("$mergeObjects", Arrays.asList(
                    "$$user",
                    new BasicDBObject("toys", new BasicDBObject("$filter", new BasicDBObject("input", "$$user.toys")
                        .append("as", "toy")
                        .append("cond", new BasicDBObject("$gte", Arrays.asList("$$toy.createdAt", groupUser.getCreatedAt())))
                    ))
                ))
            );
        }
    }).as("groupUsers");

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