spring集成mongodb聚合操作记录

深藏阁楼爱情的钟 2021-09-24 02:08 328阅读 0赞

导入mongodbjar，配置xml文件

控制器聚合操作记录

@GetMapping("/selectCountExcel")
        @RequiresPermissions("sms:view")
        public void  selectCountExcel() {
            try {
            
                Aggregation aggregation = null;
    //            设置时间查询范围
                Criteria criteria = Criteria.where("time").gt(startTime).lte(endTime);
                if (countType == 2) {
    //            不包含陕西省
                    criteria.and("province").ne("陕西省");
                    aggregation = Aggregation.newAggregation(
    //条件放入aggregation管道
                            Aggregation.match(criteria),
    //设置根据city分组查询去第一个给别名cityName，再取出想显示的省名称给别名 provinceName，分组后的统计值设置别名num
                       Aggregation.group("city").first("city").as("cityName").first("province").as("provinceName").count().as("num"),
    //                     Aggregation.project("city", "num","province"),
    //根据统计值倒叙排序
                            Aggregation.sort(Sort.by(Sort.Order.desc("num")))
                    );
                } else if (countType == 3) {
    //            全国省分组排名
                    aggregation = Aggregation.newAggregation(
                            Aggregation.match(criteria),
                            Aggregation.group("province").first("province").as("provinceName").first("city").as("cityName").count().as("num"),
    //                     Aggregation.project("province", "num"),
                            Aggregation.sort(Sort.by(Sort.Order.desc("num")))
                    );
                } else {
    //            陕西省内的分组统计
                    criteria.and("province").is("陕西省");
                    aggregation = Aggregation.newAggregation(
                            Aggregation.match(criteria),
                            Aggregation.group("city").first("city").as("cityName").first("province").as("provinceName").count().as("num"),
    //                        Aggregation.project("city", "num","province"),
                            Aggregation.sort(Sort.by(Sort.Order.desc("num")))
                    );
                }
    //根据管道参数，查询send_对应的表名
                AggregationResults<BasicDBObject> outputTypeCount = outputTypeCount = mongoTemplate.aggregate(aggregation, "send_"+smsId, BasicDBObject.class);
                List<MongodbPhoneCount> mongodbPhoneCountList = new ArrayList<>();
                for (Iterator<BasicDBObject> iterator = outputTypeCount.iterator(); iterator.hasNext(); ) {
    //                DBObject obj = iterator.next();
                    String s = JSON.toJSONString(iterator.next());
                    JSONObject jsStr = JSONObject.parseObject(s);
    //获取到的数据转成json最终转成对象
                    MongodbPhoneCount mongodbPhoneCount = (MongodbPhoneCount) JSONObject.toJavaObject(jsStr, MongodbPhoneCount.class);
    //添加到一个list中
                    mongodbPhoneCountList.add(mongodbPhoneCount);
                }
                
               return mongodbPhoneCountList;
            }catch (Exception e){
                log.error("导出失败", e);
            }
        }