MongoTemplate之分组分页复合条件查询

ongodbsql 文档地址: https://docs.mongodb.com/manual/reference/operator/update/

废话不多说,直接上代码

情景一:分组查询

 public Page<CourseDetail> listCourseDetails(QueryCourseDetailModel queryModel) {
​
        //条件一 where gradeId in (集合) and mark=true
        Criteria criteria = Criteria.where("gradeId").in(gradeIds).and("mark").is(true);
​
        if (null != status && status.size() > 0) {
            Criteria criteriaChild = new Criteria();
            //条件二  criteriaChild 条件查询 ( teacherCommentStatus = XX or parentsReplyTeacherStatus =XX  )
            criteriaChild.orOperator(Criteria.where("teacherCommentStatus").in(status), Criteria.where("parentsReplyTeacherStatus").is(replayStatus));
        }
​
        //拼接最后语句 where gradeId in (集合) and mark=true  and ( teacherCommentStatus = XX or parentsReplyTeacherStatus =XX  )
        criteria.andOperator(criteriaChild);

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

        //使用 mongoTemplate.group 分组查询
        GroupByResults groupByResults = mongoTemplate.
                group(criteria, "homework", groupBy, Homework.class);

        //获取结果
        BasicDBList list = (BasicDBList) groupByResults.getRawResults().get("retval");
        list.stream().map(map -> {
            BasicDBObject obj = (BasicDBObject) map;
            ......
            return obj;
        }).collect(Collectors.toList());
​
        return list;

        ​}

情景二: 分页查询

public Page<CourseDetail> listCourseDetails(QueryCourseDetailModel queryModel) {
​
        //创建排序模板Sort
        Sort sort = new Sort(Sort.Direction.DESC, "creationDate");​

        //创建分页模板Pageable 
        Pageable pageable = new PageRequest(queryModel.getPageStart(), queryModel.getPageSize(), sort);

        //创建查询条件对象 
        Query query = new Query();​

        //条件gradeId =XX
        Criteria criteria = Criteria.where("gradeId").is(queryModel.getGradeId());

        //条件 and time > XX 
        if (!StringUtil.isEmpty(queryModel.getStartCourseTime())) {
            criteria.and("time").gt(queryModel.getStartCourseTime());
        }​
        //条件 and time < XX if(! StringUtil.isEmpty(queryModel.getEndCourseTime())){
        criteria.and("time").lt(queryModel.getEndCourseTime());

        query.addCriteria(criteria);​

        //mongoTemplate.count计算总数 
        long total = mongoTemplate.count(query, CourseDetail.class);

        // mongoTemplate.find 查询结果集
        List<CourseDetail> items = mongoTemplate.find(query.with(pageable), CourseDetail.class);​
        items.forEach(courseDetail -> {
            courseDetail.setTeacherUnReplaySize(homeworkRepository.countByGradeIdAndParentsReplyTeacherStatusAndCourseStartTimeAndMarkTrue(courseDetail.getGradeId(), PARENTS_REPLY_TEACHER_STATUS.UNANSWERED, courseDetail.getCourseStartTime().getTime()));
        });
        
        ​return new PageImpl(items, pageable, total);
        
        ​}