具有复杂对象的 Spring Data MongoDB addToSet()答案

【问题标题】：Spring Data MongoDB addToSet() with complex object具有复杂对象的 Spring Data MongoDB addToSet()
【发布时间】：2018-04-18 19:07:34
【问题描述】：

我想使用 Spring Data 通过 Java 应用程序从 MongoDB 获取数据。

我做了以下 MongoDB 查询并成功转换为 Java 代码

db.getCollection('financialMessage').aggregate([{
            $match:{ createdDate: {
                       $gte: ISODate("2017-11-03 00:00:00.000Z"), 
                       $lt: ISODate("2017-11-04 00:00:00") }}}, { 
            $group: { _id: {
                        consolidatedBatchId: "$consolidatedBatchId",
                        version: "$version"},
                      messages: { $addToSet: "$message" }}}, {
            $sort: {
                    "_id.consolidatedBatchId": 1,
                    "_id.version": 1}
            }])

结果如下：

{
    "_id" : {
        "consolidatedBatchId" : "5f4e1d16-2070-48ef-8369-00004ec3e8ee",
        "version" : 4
    },
    "messages" : [ 
        "message1", 
        "message2", 
        "message3"
    ]
}

上述查询的 Java 代码如下所示：

        Criteria filterCriteria = Criteria.where(CREATED_DATE)
            .gte(startDate)
            .lt(endDate);
        Sort sort = new Sort(Sort.Direction.DESC, "consolidatedBatchId" ,"version");

        Aggregation agg = Aggregation.newAggregation(
            Aggregation.match(filterCriteria),
            Aggregation.group("consolidatedBatchId", "version")
                .addToSet("message").as("messages"),
            Aggregation.sort(sort)
        );

        AggregationResults<FinancialMessageKey> aggregationResults =
            mongoTemplate.aggregate(agg, FinancialMessage.class, FinancialMessageKey.class);

        return aggregationResults.getMappedResults();

现在我找不到如何将以下 MongoDB 查询代码转换为 Java 代码：

db.getCollection('financialMessage').aggregate([{
        $match:{ createdDate: {
                            $gte: ISODate("2017-11-03 00:00:00.000Z"), 
            $lt: ISODate("2017-11-04 00:00:00")
                }}}, { 
        $group: { _id: {
            consolidatedBatchId: "$consolidatedBatchId",
            version: "$version"},
                            messages: { $addToSet: {message: "$message",
                                             createdDate: "$createdDate",
                                             sender: "$sender",
                                             receiver: "$receiver" }}}}, {
        $sort: {
                "_id.consolidatedBatchId": 1,
                "_id.version": 1}
        }])

输出如下：

{
    "_id" : {
        "consolidatedBatchId" : "5f4e1d16-2070-48ef-8369-00004ec3e8ee",
        "version" : 4
    },
    "messages" : [ 
        {
            "message" : "message1",
            "createdDate" : ISODate("2017-11-03T07:13:08.074Z"),
            "sender" : "sender",
            "receiver" : "receiver"
        }, 
        {
            "message" : "message2",
            "createdDate" : ISODate("2017-11-03T07:13:08.111Z"),
            "sender" : "sender",
            "receiver" : "receiver"
        }, 
        {
            "message" : "message3",
            "createdDate" : ISODate("2017-11-03T07:13:07.986Z"),
            "sender" : "sender",
            "receiver" : "receiver"
        }
    ]
}

如何用 Java 编写 addToSet() 以获得 List 而不是简单的 List ？

【问题讨论】：

标签： java mongodb aggregation-framework spring-data-mongodb

【解决方案1】：

在谷歌上研究了几分钟，终于找到了方法。

public List<FinancialMessageKey> findFinancialMessageKeys(FinancialMessageQueryDTO financialMessageQueryDTO) {

        Criteria filterCriteria = Criteria.where(CREATED_DATE)
            .gte(startDate)
            .lt(endDate);
        Sort sort = new Sort(Sort.Direction.DESC, "consolidatedBatchId", "version");

        Aggregation agg = Aggregation.newAggregation(
            Aggregation.match(filterCriteria),
            Aggregation.group(CONSOLIDATED_BATCH_ID, VERSION)
                .addToSet(new BasicDBObject() {
                              {
                                  put(CREATED_DATE, "$" + CREATED_DATE);
                                  put(SENDER, "$" + SENDER);
                                  put(RECEIVER, "$" + RECEIVER);
                                  put(MESSAGE, "$" + MESSAGE);
                              }
                          }
                ).as("messages"),
            Aggregation.sort(sort));

        AggregationResults<FinancialMessageKey> aggregationResults =
            mongoTemplate.aggregate(agg, FinancialMessage.class, FinancialMessageKey.class);

        return aggregationResults.getMappedResults();
    }

【讨论】：