package com.moral.service.impl; import static com.moral.common.bean.Constants.NULL_VALUE; import static org.apache.commons.lang3.time.DateUtils.addDays; 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.parseDate; import static org.apache.commons.lang3.time.DateUtils.truncate; import static org.springframework.util.ObjectUtils.isEmpty; import java.text.ParseException; import java.util.Calendar; import java.util.Collections; import java.util.Date; import java.util.HashMap; import java.util.HashSet; import java.util.LinkedHashMap; import java.util.List; import java.util.Map; import java.util.Set; 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.exception.BusinessException; import com.moral.common.util.CalculateUtils; import com.moral.common.util.ResourceUtil; 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 getDayAQIByDevice(Map parameters) { ValidateUtil.notNull(parameters.get("mac"), "param.is.null"); Map resultMap = new HashMap(); Date date = new Date(); // 昨日00:00:00 parameters.put("start", truncate(addDays(date, -1), Calendar.DATE)); // 今日00:00:00 parameters.put("end", truncate(date, Calendar.DATE)); String[] IAQIValues = ResourceUtil.getArrValue("IAQI"); Map average = historyMapper.getDayAQIByDevice(parameters); if (isEmpty(average)) { resultMap.put("AQI", NULL_VALUE); } else { Set IAQIs = new HashSet(); for (Map.Entry entry : average.entrySet()) { double minMacKey = 0, maxMacKey = 0, minIAQI = 0, maxIAQI = 0; String[] macKeyValues = ResourceUtil.getArrValue(entry.getKey()); Double avg = entry.getValue(); if (isEmpty(avg)) { IAQIs.add(null); } else { int index = -1; for (int i = 0; i < macKeyValues.length; i++) { if (avg <= Double.valueOf(macKeyValues[i])) { if (i == 0) { index = i; } else { index = i - 1; } break; } } if (index == -1) { IAQIs.add(Double.MAX_VALUE); } else { minMacKey = Double.valueOf(macKeyValues[index]); maxMacKey = Double.valueOf(macKeyValues[index + 1]); minIAQI = Double.valueOf(IAQIValues[index]); maxIAQI = Double.valueOf(IAQIValues[index + 1]); Double result = CalculateUtils.calculateIAQI(maxIAQI, minIAQI, maxMacKey, minMacKey, avg); IAQIs.add(result); } } } IAQIs.remove(null); if (isEmpty(IAQIs)) { resultMap.put("AQI", NULL_VALUE); } else { Double AQI = Collections.max(IAQIs); if (AQI == Double.MAX_VALUE) { resultMap.put("AQI", IAQIValues[IAQIValues.length - 1]); } else { resultMap.put("AQI", String.format("%.0f", AQI)); } } } return resultMap; } @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; } @Override public List> getSensorsAverageByDevice4Report(Map parameters,List> sensors) { Object type = parameters.get("type"); if ("hour".equals(type)) { parameters.put("type", "%Y-%m-%d %H:00"); } else if ("minute".equals(type)) { parameters.put("type", "%Y-%m-%d %H:%i:00"); } else { throw new BusinessException("type参数输入错误!"); } try { Date start = parseDate((String)parameters.get("time"), "yyyy-MM-dd"); parameters.put("start", start); parameters.put("end", addDays(start, 1)); } catch (ParseException e) { e.printStackTrace(); throw new BusinessException("time参数输入错误!"); } String queryColumns = ""; for (int i = 0; i < sensors.size(); i++) { String sensorKey = (String) sensors.get(i).get("key"); if (i == sensors.size() - 1) { queryColumns += "AVG(value -> '$." + sensorKey + "') " + sensorKey; } else { queryColumns += "AVG(value -> '$." + sensorKey + "') " + sensorKey +","; } } parameters.put("queryColumns", queryColumns); return historyMapper.getSensorsAverageByDevice4Report(parameters); } }