需求描述:
数据在mongo中,要求按月统计最近一年内的数据;
问题描述:
1.每天都会有数据,但是数据每天都是全量的,也就是说2号的数据包含1号的全部数据; 所以,按月统计,每月只能取最后一天的数据进行加和;
2.时间字段只有一个,所以需要对时间格式化成日和月的形式:日用来过滤每月最后一天; 月用来统计每月数据量;
话不多说,上代码
@Autowired
private MongoTemplate mongoTemplate;
@Override
public Map<String, Object> getIncrementTrend() {
//获取每个月1号
LocalDate localDate = LocalDate.now().withDayOfMonth(1);
String[] months = new String[12];
//使用set,自动排序;
Set<String> monthSet = new TreeSet<>();
//获取最近一年内每个月最后一天;
for (int i = 0; i < 12; i++) {
//每月最后一天;
LocalDate lastDayOfMonth = localDate.minusMonths(i).minusDays(1);
//格式化最后一天;"yyyy-MM-dd"
months[i] = lastDayOfMonth.format(DateTimeFormatter.ISO_LOCAL_DATE);
//格式化成年月: "yyyy-MM",返回前端 月列表;
monthSet.add(lastDayOfMonth.format(DateTimeFormatter.ofPattern("yyyy-MM")));
}
List<AggregationOperation> operations = new ArrayList<>();
//将需要查询的字段格式化为需要的形式,我这里需要将collectTime格式化为日期和月的形式;
//project:需要显示的字段;
//andExpression:表达式;需要格式化日期为需要的形式
operations.add(Aggregation.project()
.andExpression("{$dateToString:{ format:'%Y-%m-%d',date: '$collectTime', timezone: 'Asia/Shanghai'}}").as("$collectDate")
.andExpression("{$dateToString:{ format:'%Y-%m',date: '$collectTime', timezone: 'Asia/Shanghai'}}").as("$collectMonth"));
//查询条件,字段在我指定的日期中;
operations.add(Aggregation.match(Criteria.where("collectDate").in(months)));
//按月分组统计;group分组,会将分组字段放入_id中,如果需要显示该字段,可以用聚合函数中的其中一个标记一下,我这里用了max,as()给字段命名。
operations.add(Aggregation.group("collectMonth").count().as("countNum").max("collectMonth").as("collectMonth"));
//排序;
operations.add(Aggregation.sort(Sort.by(new Sort.Order(Sort.Direction.ASC, "collectMonth"))));
Aggregation aggregation = Aggregation.newAggregation(operations);
AggregationResults<HashMap> aggregate = mongoTemplate.aggregate(aggregation, "CollectionName", HashMap.class);
List<HashMap> results = aggregate.getMappedResults();
HashMap<String,Object> map = new HashMap<>();
HashMap<String,Object> dataMap = new HashMap<>();
results.forEach( maps -> dataMap.put((String)maps.get("collectMonth"),maps.get("countNum")));
map.put("months",monthSet);
map.put("data",dataMap);
return map;
}
最后,在补充一下:
分组指的是将满足相同条件的数据变为一条(组)数据
比如createTime 创建时间为 2020年的11月有10条数据,若根据createTime 为2020年11月分组,分组后createTime字段唯一,其他字段不一定唯一,其他字段怎么显示呢?这里要用到聚合函数
聚合指的是多个数据如何取值
常见的聚合函数有:sum(求和),max(取最大值),min(最小值),avg(平均值),first(第一个),last(最后一个),count(记录条数)等
本例聚合的所有collectMonth都相同,所以使用first、last、max、min都可以