使用mongoTemplate实现多条件加分组查询

先来一个常见的错误信息：

Due to limitations of the com.mongodb.BasicDocument, you can't add a second '$and' expression specified as '$and :

错误原因：在一个 Criteria 对象中调用了多次 andOperator() 方法

使用mongoTemplate实现多条件查询

多个条件的查询只需要创建 Query 对象，然后把需要添加的条件使用 Query 对象的 addCriteria() 方法

// 场景：查询指定时间段内，状态为1的数据
// 入参条件 ：beginTime ，endTime ，statue
// mongodb字段：time ， state

// 存放条件的对象
Query condition= new Query();

// 判断时间是否为空
if(beginTime != null && endTime != null){
    // 添加大于开始时间小于结束时间的条件
    condition.addCriteria(Criteria.where("time").gte(beginTime).lte(endTime));
}else{
    // 其中一个为空 分别进行判断
    if(beginTime != null){
        condition.addCriteria(Criteria.where("time").gte(beginTime));
    }
    if(endTime != null){
        condition.addCriteria(Criteria.where("time").lte(endTime));
    }
}

// 添加状态为1条件
if(statue!=null){
    condition.addCriteria(Criteria.where("state").is(statue));
}

条件有了后再调用 mongoTemplate.find(condition,返回类型.class,collectionName)

但是...想要分组，得调用 mongoTemplate.group(Criteria criteria , String inputCollectionName , GroupBy groupBy , Class<T> entityClass) 方法 （不是说只能这样才能分组，而是我通过这种方法实现了分组查询）

朋友们，第一个参数条件只能入参 Criteria 对象，而不能入参 Query 对象

结果我发现 Criteria 对象有 andOperator(Criteria ... criteria) 方法

这个方法就厉害了，可以入参数组，也就是说

我们可以把查询条件先存放到一个集合里面（因为数组需要定义长度，如果条件个数不确定，就不能直接定义数组），然后把集合放入数组中，再把数组入参 andOperator(Criteria ... criteria) 方法

// 场景：查询指定时间段内，状态为1的数据
// 入参条件 ：beginTime ，endTime ，statue
// mongodb字段：time ， state

// 定义一个存放条件的集合
List<Criteria> criteriaList = new ArrayList<>();
// 定义一个存放条件的数组（暂时不给长度）
Criteria[] criteriaArray = {};

// 判断时间是否为空
if(beginTime != null && endTime != null){
    // 添加大于开始时间小于结束时间的条件
    Criteria between = Criteria.where("time").gte(beginTime).lte(endTime);
    criteriaList.add(between);
}else{
    // 其中一个为空 分别进行判断
    if(beginTime != null){
        Criteria gte = Criteria.where("time").gte(beginTime);
        criteriaList.add(gte);
    }
    if(endTime != null){
        Criteria lte = Criteria.where("time").lte(endTime);
        criteriaList.add(lte);
    }
}

// 添加状态为1条件
if(statue!=null){
    Criteria isState = Criteria.where("state").is(statue);
    criteriaList.add(isState);
}

// 如果有条件
if(criteriaList.size()>0){
  // 集合的个数就是数组的长度
  criteriaArray = new Criteria[criteriaList.size()];
  // 遍历添加到数组中
  for(int i = 0 ; i<criteriaList.size(); i++){
      criteriaArray[i] = criteriaList.get(i);
   }
}

这种就可以调用 mongoTemplate.group(Criteria criteria , String inputCollectionName , GroupBy groupBy , Class<T> entityClass) 方法进行分组查询了

 GroupBy groupBy = new GroupBy("分组字段")
                    .initialDocument("{ count: 0 }")
                    .reduceFunction("function (doc,pre){pre.count +=1 ;}");

// new Criteria().andOperator(criteriaArray) 这个是很关键的一步操作，把刚刚的条件数组放入进入
// groupByResults 这个对象里面内容很多，有兴趣的朋友可以断点进入看一下
GroupByResults groupByResults = mongoTemplate.
                    group(new Criteria().andOperator(criteriaArray), mongodb的collectionName, groupBy, 实体类.class);
// 获取分组后的数量
long resultCount = ((List)groupByResults.getRawResults().get("retval")).size();

到此多条件加分组查询已经结束

我还有一篇关于mongodb查询的文章（点击这里）

希望能够帮助到你

over

使用mongoTemplate实现多条件加分组查询

使用mongoTemplate实现多条件查询

猜你喜欢