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

Love The Way You Lie 2022-04-12 02:46 928阅读 0赞

> 先来一个常见的错误信息：
> 
>     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查询的文章（[点击这里][Link 1]）

希望能够帮助到你

over

[Link 1]: https://blog.csdn.net/qq_41402200/article/details/82149110