如何使用MongoDB Java Driver 3.6在聚合管道中合并文档？

嗨，我来帮你搞定这个合并文档的需求！既然每个VIN恰好对应两条匹配记录，我们可以用MongoDB聚合里的$group阶段，按VIN分组来把两条记录合并成一个结果，这样你就不用处理两条独立的结果了。

核心思路 #

核心就是通过$group以CAR.VIN作为分组键，再用合适的累加操作符把两条记录整合到一起。这里分两种常见场景：

场景1：把两条记录存入数组（保留完整原始文档） #

这种方式适合需要保留两条记录所有细节的情况，不会丢失任何字段。假设你的文档结构大概是这样：

doc1:

{
  "CAR": { "VIN": "VIN-123" },
  "engineData": { "hp": 200 },
  "timestamp": "2024-01-01"
}

doc2:

{
  "CAR": { "VIN": "VIN-123" },
  "sensorData": { "temp": 90 },
  "timestamp": "2024-01-02"
}

对应的Java Driver 3.6代码实现：

import com.mongodb.client.MongoCollection;
import com.mongodb.client.model.Aggregates;
import com.mongodb.client.model.Bson;
import static com.mongodb.client.model.Accumulators.push;
import static com.mongodb.client.model.Filters.eq;
import java.util.Arrays;
import org.bson.Document;

// 获取你的集合对象
MongoCollection<Document> collection = mongoDatabase.getCollection("your_collection_name");

// 构建聚合管道
List<Bson> pipeline = Arrays.asList(
    // 可选：如果需要过滤特定VIN，添加$match阶段；不需要的话可以移除
    Aggregates.match(eq("CAR.VIN", "VIN-123")),
    // 按CAR.VIN分组，将所有匹配的文档存入records数组
    Aggregates.group("$CAR.VIN", push("records", "$$ROOT"))
);

// 执行聚合并处理结果
for (Document mergedResult : collection.aggregate(pipeline)) {
    // 这里可以处理合并后的结果，比如打印或转成自定义对象
    System.out.println(mergedResult.toJson());
}

执行后得到的结果示例：

{
  "_id": "VIN-123",
  "records": [
    {"CAR": {"VIN": "VIN-123"}, "engineData": {"hp": 200}, "timestamp": "2024-01-01"},
    {"CAR": {"VIN": "VIN-123"}, "sensorData": {"temp": 90}, "timestamp": "2024-01-02"}
  ]
}

场景2：直接合并成单个文档（字段无冲突时推荐） #

如果两条文档的字段不重叠（或者你希望后面的字段覆盖前面的），可以用$mergeObjects操作符把两条记录的字段合并成一个文档。MongoDB 3.6及以上支持这个操作符，对应的Java Driver 3.6也能兼容。

调整后的$group阶段代码：

import static com.mongodb.client.model.Accumulators.accumulator;

// 修改聚合管道里的$group部分
Aggregates.group("$CAR.VIN",
    accumulator("mergedDoc",
        new Document(), // 初始值为空文档
        new Document("$mergeObjects", Arrays.asList("$$value", "$$ROOT"))
    )
)

执行后得到的结果示例：

{
  "_id": "VIN-123",
  "mergedDoc": {
    "CAR": {"VIN": "VIN-123"},
    "engineData": {"hp": 200},
    "sensorData": {"temp": 90},
    "timestamp": "2024-01-02" // 注意：后面的文档timestamp覆盖了前面的
  }
}

额外注意事项 #

• 如果不需要过滤特定VIN，直接移除$match阶段即可，这样所有VIN对应的两条记录都会被分组合并。
• 如果两条文档有同名字段，$mergeObjects会用后处理的文档字段值覆盖先处理的，所以如果要保留所有同名字段，还是用场景1的数组方式更合适。
• Java Driver 3.6里使用accumulator来定义$mergeObjects的累加逻辑，这是因为3.6版本的驱动还没有直接封装$mergeObjects的累加器方法，手动构建accumulator是正确的方式。

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