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

Mongo Aggregation mapping and query to fetch selective records for particular year using Mongo Template

我正在使用 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");

下面是我的collection结构:

{
  "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 年),现在我需要检索相应的记录,从日期中提取月份部分并使用我的 collection 中的状态字段,以便处理 collection 的通过和失败记录=31=]。使用 mongoShell 直接查询获得了许多示例,但需要使用 mongo 模板和必要的 pojo class 映射来实现语法方面的帮助。也可以使用 mongo 聚合查询本身实现对每个月通过和失败记录的处理吗?

期待 mongo 专家的一些 help/inputs 以正确的方式在通过和失败场景的单次提取操作中做到这一点。

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

创建class支持聚合操作:

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();

型号Class供任何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 class,以便 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]
                }
            }
        }
    }
]);