mongodb - Mongo 聚合映射和查询以使用 Mongo 模板获取特定年份的选择性记录

Question

我正在使用 mongo 模板查询基于状态和邮政编码的文档，这是使用以下标准和查询方法实现的:

List<ModelClass> modelClass = null;
Criteria criteria = new Criteria();
if(request.getState()!=null) {
  criteria = criteria.and("state").is(request.getState());
}
if(request.getZipcode()!=null) {
 criteria = criteria.and("zipcode").is(request.getZipcode());
}
Query query = new Query(criteria);

modelClass = mongoTemplate.find(query, ModelClass.class, "collectionName");

以下是我的收藏结构:

{
  "id": "5c59508688c12c9a3dc2dfe2",
  "zipcode": "12345",
  "state": "NY",
  "date": "2019-02-05T08:59:50.703",
  "status": "pass"
},
{
   "id": "5c5957216f4b9613c0c0b1ed",
    "zipcode": "45678",
    "state": "CO",
    "date": "2018-01-05T08:59:50.703",
    "status": "fail"
},
etc....

仅对于过去的一年(例如 2019 年)，现在我需要检索相应的记录，从日期中提取月份部分并使用我的集合中的状态字段，以便为 UI 处理通过和失败的记录。有很多使用 mongoShell 直接查询的例子，但是需要帮助使用 mongo 模板和必要的 pojo 类映射来实现相同的语法。也可以通过mongo聚合查询本身实现每个月通过和失败记录的处理吗？

期待来自 mongo 专家的一些帮助/输入，以正确的方式在通过和失败场景的单次提取操作中完成。

Answer 1

最后，我已经能够使用 mongo 实现本身来实现上述要求。使用最新的 mongo 版本可能会有更好的解决方案。我使用 spring-data-mongodb-1.8.2.RELEASE，因此这可能会帮助其他寻找类似解决方案的人。

创建支持聚合操作的类:

import org.springframework.data.mongodb.core.aggregation.AggregationOperation;
import org.springframework.data.mongodb.core.aggregation.AggregationOperationContext;

import com.mongodb.DBObject;

public class CustomProjectAggregationOperation implements AggregationOperation {

    private DBObject operation;

    public CustomProjectAggregationOperation (DBObject operation) {
        this.operation = operation;
    }

    @Override
    public DBObject toDBObject(AggregationOperationContext context) {
        return context.getMappedObject(operation);
    }
}

实现:

Criteria criteria = new Criteria();
criteria = criteria.and("date").gte(fromDate).lte(endDate).exists(true);

DBObject operation = (DBObject)new BasicDBObject("$group",
                          new BasicDBObject("_id", new BasicDBObject("month", 
                            new BasicDBObject("$month", "$date")))
                            .append("passCount", new BasicDBObject("$sum", 
                               new BasicDBObject("$cond", new Object[]{
                                 new BasicDBObject("$eq", new Object[]{ "$status", 
                                                               "Pass"}),1,0}))
                            ).append("failCount", new BasicDBObject(    
                                "$sum", new BasicDBObject("$cond", new Object[]{ 
                                   new BasicDBObject("$eq", new Object[]{ "$status", 
                                                            "Fail"}),1,0}))
                            ));

Aggregation agg = newAggregation(match(criteria),new 
                          CustomProjectAggregationOperation(operation));

List<ModelClass> modelList= mongoTemplate.aggregate(agg,"collectionName",ModelClass.class).getMappedResults();

像我这样的任何 mongo 新手的模型类引用:

public class ModelClass{

    private String month;
    private int passCount;
    private int failCount;

    public String getMonth() {
        return month;
    }
    public void setMonth(String month) {
        this.month= month;
    }
    public int getPassCount() {
        return passCount;
    }
    public void setPassCount(int passCount) {
        this.passCount= passCount;
    }
    public int getFailCount() {
        return failCount;
    }
    public void setFailCount(int failCount) {
        this.failCount = failCount;
    }   
}

请确保使用查询中指定的相似名称创建模型 pojo 类，以便 mongo 映射正常工作。

等效的 Mongo Shell:

db.collectionName.aggregate([
    {
        "$match": {
            "date": {
                "$gte": {
                    "$date": "2018-12-31T18:30:00.000Z"
                },
                "$lte": {
                    "$date": "2019-12-31T13:43:17.106Z"
                },
                "$exists": true
            }
        }
    }, {
        "$group": {
            "_id": {
                "month": {
                    "$month": "$date"
                }
            },
            "passCount": {
                "$sum": {
                    "$cond": [{
                        "$eq": ["$status", "Pass"]
                    }, 1, 0]
                }
            },
            "failCount": {
                "$sum": {
                    "$cond": [{
                        "$eq": ["$status", "Fail"]
                    }, 1, 0]
                }
            }
        }
    }
]);