package com.moral.service.impl; import static com.moral.common.bean.Constants.NULL_VALUE; import static org.apache.commons.lang3.time.DateUtils.addHours; import static org.apache.commons.lang3.time.DateUtils.addMinutes; import static org.apache.commons.lang3.time.DateUtils.addMonths; import static org.apache.commons.lang3.time.DateUtils.truncate; import static org.springframework.util.ObjectUtils.isEmpty; import java.util.Calendar; import java.util.Date; import java.util.HashMap; import java.util.LinkedHashMap; import java.util.List; import java.util.Map; import javax.annotation.Resource; import org.springframework.data.mongodb.core.MongoTemplate; import org.springframework.data.mongodb.core.aggregation.Aggregation; import org.springframework.data.mongodb.core.aggregation.AggregationResults; import org.springframework.data.mongodb.core.query.Criteria; import org.springframework.stereotype.Service; import com.mongodb.BasicDBObject; import com.moral.common.util.ValidateUtil; import com.moral.mapper.HistoryMapper; import com.moral.service.AccountService; import com.moral.service.HistoryService; @Service public class HistoryServiceImpl implements HistoryService { @Resource private AccountService accountService; @Resource private HistoryMapper historyMapper; @Resource private MongoTemplate mongoTemplate; @Override public Map getAllSensorAverageByDevice(Map parameters) { ValidateUtil.notNull(parameters.get("areaCode"), "param.is.null"); ValidateUtil.notNull(parameters.get("accountId"), "param.is.null"); accountService.setOrgIdsByAccount(parameters); Date date = new Date(); // 当前时间 -10分钟 parameters.put("start", addMinutes(date, -10)); // 当前时间 -5分钟 parameters.put("end", addMinutes(date, -5)); String queryColumns = ""; for (int i = 1; i < 20; i++) { if (i == 1) { queryColumns += "AVG(value -> '$.e" + i + "') e" + i; } else { queryColumns += " , AVG(value -> '$.e" + i + "') e" + i; } } parameters.put("queryColumns", queryColumns); parameters.put("macKey", "all"); List> list = historyMapper.getAreaAllDataByAccount(parameters); return list.get(0); } @Override public Map getDeviceRankingBySensorAverage(Map parameters) { ValidateUtil.notNull(parameters.get("areaCode"), "param.is.null"); ValidateUtil.notNull(parameters.get("accountId"), "param.is.null"); ValidateUtil.notNull(parameters.get("macKey"), "param.is.null"); Map result = new LinkedHashMap(); accountService.setOrgIdsByAccount(parameters); Date date = new Date(); // 当前时间 -1小时 parameters.put("start", addHours(date, -1)); parameters.put("end", date); parameters.put("macKey", "'$." + parameters.get("macKey") + "'"); List> list = historyMapper.getAreaAllDataByAccount(parameters); for (Map map : list) { result.put((String) map.get("name"), map.get("avg")); } return result; } @Override public Map getMonthAverageBySensor(Map parameters) { ValidateUtil.notNull(parameters.get("mac"), "param.is.null"); ValidateUtil.notNull(parameters.get("macKey"), "param.is.null"); Map result = new HashMap(); Date date = new Date(); Long end = truncate(date, Calendar.DATE).getTime(), start; // 每月一日的数据取上月的数据 if (1 == Calendar.getInstance().get(Calendar.DAY_OF_MONTH)) { // 上个月1日00:00:00 start = truncate(addMonths(date, -1), Calendar.MONTH).getTime(); } else { // 这个月1日00:00:00 start = truncate(date, Calendar.MONTH).getTime(); } Aggregation aggregation = Aggregation.newAggregation( Aggregation.match(Criteria.where("mac").is(parameters.get("mac"))), Aggregation.match(Criteria.where("time").gte(start)), Aggregation.match(Criteria.where("time").lt(end)), Aggregation.group("mac").avg((String) parameters.get("macKey")).as("average")); AggregationResults results = mongoTemplate.aggregate(aggregation, "data", BasicDBObject.class); List list = results.getMappedResults(); if (isEmpty(list)) { result.put("average", NULL_VALUE); } else { result = list.get(0); result.put("average", String.format("%.2f", result.get("average"))); } return result; } }