Spring Boot基于MongoDB开发多集合ID关联合并返回的GET API问题

你可以通过Spring Data MongoDB的聚合查询能力实现需求，全程仅做读操作，不会修改原有数据库的任何数据和集合结构： #

1. 定义对应实体类 #

主集合实体 #

// MainCollection.java
@Document(collection = "MainCollection")
@Data
public class MainCollection {
    @Id
    private Long id;
    private String name;
    private Integer value;
    private SubCollectionDetails subCollectionDetails;

    @Data
    public static class SubCollectionDetails {
        // 主集合原有字段
        private String subCollectionId;
        private String subCollectionOtherValues;
        // 子集合关联后填充的字段
        private String subCollectionName;
        private Boolean subCollectionPresent;
    }
}

子集合实体 #

// SubCollection.java
@Document(collection = "SubCollectionId")
@Data
public class SubCollection {
    @Id
    private String id;
    private String subCollectionName;
    private Boolean subCollectionPresent;
}

2. 编写聚合查询逻辑 #

关联逻辑全部在MongoDB查询侧完成，不会碰原有存储数据。你可以先运行原生MongoDB语句验证结果符合预期：

// 原生MongoDB聚合查询语句
db.MainCollection.aggregate([
  {
    $lookup: {
      from: "SubCollectionId",
      localField: "subCollectionDetails.subCollectionId",
      foreignField: "_id",
      as: "matchedSubData"
    }
  },
  {
    $project: {
      name: 1,
      value: 1,
      subCollectionDetails: {
        $cond: {
          if: { $gte: [{ $size: "$matchedSubData" }, 1] },
          then: { $mergeObjects: ["$subCollectionDetails", { $arrayElemAt: ["$matchedSubData", 0] }] },
          else: {}
        }
      }
    }
  }
])

对应的Spring Data MongoDB代码实现如下：

// 注入MongoTemplate
@Autowired
private MongoTemplate mongoTemplate;

public List<MainCollection> getMergedData() {
    // 1. 关联子集合
    LookupOperation lookupOperation = LookupOperation.newLookup()
            .from("SubCollectionId")
            .localField("subCollectionDetails.subCollectionId")
            .foreignField("_id")
            .as("matchedSubData");

    // 2. 合并字段，无匹配则返回空subCollectionDetails
    AggregationExpression mergeExpr = ConditionalOperators.when(ArrayOperators.Size.lengthOfArray("matchedSubData").gte(1))
            .then(ObjectOperators.MergeObjects.merge(
                    "subCollectionDetails",
                    ArrayOperators.ArrayElemAt.arrayOf("matchedSubData").elementAt(0)
            ))
            .otherwise(new Document());

    AggregationOperation projectOperation = Aggregation.project("name", "value")
            .and(mergeExpr).as("subCollectionDetails");

    // 3. 执行查询返回结果
    Aggregation aggregation = Aggregation.newAggregation(lookupOperation, projectOperation);
    AggregationResults<MainCollection> results = mongoTemplate.aggregate(aggregation, "MainCollection", MainCollection.class);
    return results.getMappedResults();
}

如果不需要保留原subCollectionId字段，可在project阶段直接排除该字段即可，和示例响应格式完全对齐。

3. 暴露GET接口 #

在Controller层直接调用上述方法返回结果即可：

@RestController
@RequestMapping("/api/data")
public class DataController {
    @Autowired
    private DataService dataService;

    @GetMapping
    public ResponseEntity<List<MainCollection>> getMergedData() {
        return ResponseEntity.ok(dataService.getMergedData());
    }
}

关键说明 #

• 全程为只读查询操作，不会修改原有集合的任何数据和schema结构，完全符合需求
• 关联逻辑在MongoDB侧完成，比在Java内存中循环匹配效率更高，适配大数据量场景
• 无匹配子集合数据时自动返回空的subCollectionDetails对象，和你给出的示例响应完全一致

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