Java实现MongoDB无循环查询嵌套数组匹配单条记录

实现思路 #

全程使用MongoDB原生查询能力完成数组过滤，不在Java业务层做任何循环遍历逻辑，两种可选实现方案如下：

方案1：聚合管道实现（兼容所有3.2+版本MongoDB） #

聚合管道逐阶段在服务端完成数据处理，最终直接返回匹配的子文档完整结构：

1. 第一阶段通过$match前置筛选包含目标邮箱子记录的父文档，减少后续计算的数据量
2. 第二阶段通过$filter投影操作，过滤affiliationChildren数组，仅保留provider.email匹配目标值的元素
3. 第三阶段通过$unwind把过滤后仅存单个元素的数组拆解为单条文档
4. 第四阶段通过$replaceRoot将目标子文档提升为根节点，直接返回子文档完整结构

Java代码实现（基于官方MongoDB Java同步驱动）：

import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import org.bson.Document;
import java.util.Arrays;
import static com.mongodb.client.model.Aggregates.*;
import static com.mongodb.client.model.Filters.eq;
import static com.mongodb.client.model.Projections.computed;
import static com.mongodb.client.model.Projections.fields;

public class MongoAffiliationQuery {
    public Document getTargetChildRecord(MongoDatabase db) {
        MongoCollection<Document> collection = db.getCollection("你的业务集合名");
        String targetEmail = "wy.providers+si5@gmail.com";

        Document matchedChild = collection.aggregate(Arrays.asList(
                // 前置过滤减少数据量
                match(eq("affiliationChildren.provider.email", targetEmail)),
                // 过滤数组仅保留匹配元素
                project(fields(computed("matchedChild", new Document("$filter", new Document()
                        .append("input", "$affiliationChildren")
                        .append("cond", new Document("$eq", Arrays.asList("$$this.provider.email", targetEmail)))
                )))),
                // 拆分数组
                unwind("$matchedChild"),
                // 提升子文档为根节点
                replaceRoot("$matchedChild")
        )).first(); // 取单条匹配结果

        return matchedChild;
    }
}

方案2：投影+elemMatch实现（MongoDB 3.6+版本适用，代码更轻量） #

利用MongoDB投影支持的$elemMatch操作符，直接在find查询的投影阶段返回数组中第一个匹配的元素，无需多阶段聚合：

import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;
import org.bson.Document;
import static com.mongodb.client.model.Filters.eq;
import static com.mongodb.client.model.Projections.*;

public class MongoAffiliationQueryLight {
    public Document getTargetChildRecord(MongoDatabase db) {
        MongoCollection<Document> collection = db.getCollection("你的业务集合名");
        String targetEmail = "wy.providers+si5@gmail.com";

        Document queryResult = collection.find(
                eq("affiliationChildren.provider.email", targetEmail)
        ).projection(fields(
                // 只返回数组中匹配的第一个元素
                elemMatch("affiliationChildren", eq("provider.email", targetEmail)),
                excludeId() // 不需要父文档_id可保留该行配置
        )).first();

        // 取出匹配到的子文档，结构与预期返回完全一致
        return queryResult.get("affiliationChildren", Document.class);
    }
}

注意事项 #

• 两种方案的数组过滤逻辑全部在MongoDB服务端执行，无Java层循环遍历，性能远高于全量拉取文档后做内存过滤
• 建议提前给affiliationChildren.provider.email字段创建单键索引，可大幅提升大集合下的查询速度
• 如果存在多条匹配记录，两种方案默认返回第一条，符合单条匹配的需求；如果需要返回所有匹配项，只需调整聚合管道去掉$replaceRoot后追加$group阶段聚合结果即可

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