xufenglei
2019-02-13 1bd6a5e7678477f5b1fab449c6d618aae9041611
src/main/java/com/moral/service/impl/HistoryServiceImpl.java
@@ -1,40 +1,39 @@
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.time.Instant;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.ZoneId;
import java.time.ZoneOffset;
import java.time.temporal.ChronoUnit;
import java.time.temporal.TemporalAdjusters;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
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 org.springframework.util.ObjectUtils;
import com.mongodb.BasicDBObject;
import com.moral.common.exception.BusinessException;
import com.moral.common.util.CalculateUtils;
import com.moral.common.util.ResourceUtil;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.moral.common.util.ValidateUtil;
import com.moral.entity.Device;
import com.moral.entity.Profession;
import com.moral.entity.Sensor;
import com.moral.mapper.DeviceMapper;
import com.moral.mapper.HistoryMapper;
import com.moral.mapper.SensorMapper;
import com.moral.service.AccountService;
import com.moral.service.HistoryService;
import com.moral.service.ProfessionService;
@Service
public class HistoryServiceImpl implements HistoryService {
@@ -46,169 +45,254 @@
   private HistoryMapper historyMapper;
   @Resource
    private MongoTemplate mongoTemplate;
   private DeviceMapper deviceMapper;
   @Resource
   private SensorMapper sensorMapper;
    @Resource
    private ProfessionService professionService;
   @Override
   public Map<String, Object> getAllSensorAverageByDevice(Map<String, Object> parameters) {
   public Map<String, Object> getAllSensorAverageByDevice(Map<String, Object> parameters) throws Exception {
      ValidateUtil.notNull(parameters.get("areaCode"), "param.is.null");
      ValidateUtil.notNull(parameters.get("accountId"), "param.is.null");
      Map<String,Object> resultMap = new LinkedHashMap<String, Object>();
      accountService.setOrgIdsByAccount(parameters);
      Date date = new Date();
      LocalDateTime time = LocalDateTime.now();
      // 当前时间 -10分钟
      parameters.put("start", addMinutes(date, -10));
      parameters.put("start", time.minusMinutes(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("end", time.minusMinutes(5));
      List<Sensor> Sensors = sensorMapper.getSensorsByCriteria(parameters);
      List<String> sensorKeys = new ArrayList<String>();
      for (Sensor sensor : Sensors) {
         sensorKeys.add(sensor.getSensorKey());
      }
      parameters.put("queryColumns", queryColumns);
      parameters.put("macKey", "all");
      parameters.put("sensorKeys", sensorKeys);
      List<Map<String, Object>> list = historyMapper.getAreaAllDataByAccount(parameters);
      return list.get(0);
   }
   @Override
   public Map<String, Object> getDeviceRankingBySensorAverage(Map<String, Object> parameters) {
      Map<String, Object> result = new LinkedHashMap<String, Object>();
      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<Map<String, Object>> list = historyMapper.getAreaAllDataByAccount(parameters);
      for (Map<String, Object> map : list) {
         result.put((String) map.get("name"), map.get("avg"));
      }
      return result;
   }
   @Override
   public Map<String, Object> getDayAQIByDevice(Map<String, Object> parameters) {
      Map<String, Object> resultMap = new HashMap<String, Object>();
      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<String, Double> average = historyMapper.getDayAQIByDevice(parameters);
      if (isEmpty(average)) {
         resultMap.put("AQI", NULL_VALUE);
      } else {
         Set<Double> IAQIs = new HashSet<Double>();
         for (Map.Entry<String, Double> 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));
            }
         }
      }
      resultMap.putAll(list.get(0));
      return resultMap;
   }
   @Override
   public Map<String, Object> getMonthAverageBySensor(Map<String, Object> parameters) {
      Map<String, Object> result = new HashMap<String, Object>();
      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<BasicDBObject> results = mongoTemplate.aggregate(aggregation, "data", BasicDBObject.class);
      List<BasicDBObject> 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")));
   public Map<String, Object> getDeviceRankingBySensorAverage(Map<String, Object> parameters) {
      ValidateUtil.notNull(parameters.get("areaCode"), "param.is.null");
      ValidateUtil.notNull(parameters.get("accountId"), "param.is.null");
      Object macKey = parameters.get("macKey");
      ValidateUtil.notNull(macKey, "param.is.null");
      Map<String, Object> result = new LinkedHashMap<String, Object>();
      accountService.setOrgIdsByAccount(parameters);
      LocalDateTime time = LocalDateTime.now();
      // 当前时间 -1小时
      parameters.put("start", time.minusHours(1));
      parameters.put("end", time);
      List<String> sensorKeys = new ArrayList<String>();
      sensorKeys.add(macKey.toString());
      parameters.put("sensorKeys", sensorKeys);
      List<Map<String, Object>> list = historyMapper.getAreaAllDataByAccount(parameters);
      for (Map<String, Object> map : list) {
         result.put((String) map.get("name"), map.get(macKey.toString()));
      }
      return result;
   }
   @Override
   public String queryValueByMacAndTime(String mac, Date time){
      return  historyMapper.selectValueByMacAndTime(mac, time);
   }
   @Override
   public List<Map<String, Object>> getSensorsAverageByDevice4Report(Map<String, Object> parameters,List<Map<String, Object>> 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;
   public List<Map<String, Object>> getRegionRankingData(Map<String, Object> parameters) {
      ValidateUtil.notNull(parameters.get("regionCode"), "param.is.null");
      ValidateUtil.notNull(parameters.get("dimension"), "param.is.null");
      ValidateUtil.notNull(parameters.get("sensorKey"), "param.is.null");
      ValidateUtil.notNull(parameters.get("accountId"), "param.is.null");
      String regionCode = parameters.get("regionCode").toString();
      parameters.put("provinceCode", regionCode.substring(0,2).concat("0000"));
      String regionType = "village";
      String nextLevel = "";
      if (regionCode.length() == 6) {
         if (regionCode.endsWith("0000")) {
            regionType = "province";
            nextLevel = "city";
         } else if (regionCode.endsWith("00")) {
            regionType = "city";
            nextLevel = "area";
         } else {
            queryColumns += "AVG(value -> '$." + sensorKey + "') " + sensorKey +",";
            regionType = "area";
            nextLevel = "town";
         }
      } else {
         if (regionCode.endsWith("000")) {
            regionType = "town";
            nextLevel = "village";
         }
      }
      parameters.put("queryColumns", queryColumns);
      parameters.put("regionType", regionType);
      parameters.put("nextLevel", nextLevel);
      LocalDate localDate = LocalDate.now();
      Object timeType = parameters.get("timeType");
      String table = "history";
      String column = "value";
      if ("month".equals(timeType)) {
         if (1 != localDate.getDayOfMonth()) {
            table = "history_minutely";
            column = "json";
         }
         parameters.put("start", localDate.with(TemporalAdjusters.firstDayOfMonth()));
      } else if ("day".equals(timeType)) {
         parameters.put("start", localDate);
      } else if ("hour".equals(timeType)) {
         parameters.put("start", LocalDateTime.now().truncatedTo(ChronoUnit.HOURS));
      } else {
         parameters.put("start", LocalDateTime.now().minusMinutes(5));
      }
      parameters.put("table", table);
      parameters.put("column", column);
      
      return historyMapper.getSensorsAverageByDevice4Report(parameters);
      String dimension = "";
      if ("discharge".equals(parameters.get("dimension"))) {
         dimension = "discharge";
         parameters.put("dimension","monitorPoint");
      }
      accountService.setOrgIdsByAccount(parameters);
      List<Map<String, Object>> result = historyMapper.getRegionRankingData(parameters);
      for (Map<String, Object> map : result) {
         if (ObjectUtils.isEmpty(map.get("avg"))) {
            map.put("avg", 0);
         }
      }
      if ("profession".equals(parameters.get("dimension"))) {
         List<Profession> professions = professionService.getProfessiontList(parameters);
         for (Map<String, Object> map : result) {
            for (Profession profession : professions) {
               if (map.get("name").equals(profession.getName())) {
                  professions.remove(profession);
//                  if (ObjectUtils.isEmpty(map.get("avg"))) {
//                     map.put("avg", 0);
//                  }
                  break;
               }
            }
         }
         for (Profession profession : professions) {
            Map<String, Object> map = new HashMap<String, Object>();
            map.put("name", profession.getName());
            map.put("avg", 0);
            result.add(map);
         }
      } else if ("region".equals(parameters.get("dimension"))) {
         List<String> regions = historyMapper.selectNextLevelRegion(parameters);
         for (Map<String, Object> map : result) {
            if (regions.contains(map.get("name"))) {
               regions.remove(map.get("name"));
//               if (ObjectUtils.isEmpty(map.get("avg"))) {
//                  map.put("avg", 0);
//               }
            }
         }
         for (String region : regions) {
            Map<String, Object> map = new HashMap<String, Object>();
            map.put("name", region);
            map.put("avg", 0);
            result.add(map);
         }
      }
      if ("discharge".equals(dimension)) {
         parameters.put("year", LocalDate.now().getYear());
         List<Map<String, Object>> discharges = historyMapper.getDischargeByRegion(parameters);
         for (Map<String, Object> map : result) {
            for (Map<String, Object> discharge : discharges) {
               if (map.get("name").equals(discharge.get("name"))) {
                  map.put("value", discharge.get("value"));
                  discharges.remove(discharge);
                  break;
               }
            }
         }
         Collections.sort(result, new Comparator<Map<String, Object>>() {
            @Override
            public int compare(Map<String, Object> o1, Map<String, Object> o2) {
               return Double.compare(((Double) o2.get("avg"))- Double.valueOf((String) o2.get("value")),
                     ((Double) o1.get("avg")) - Double.valueOf((String) o1.get("value")));
            }
         });
      }
      return result;
   }
    @Override
   public List<Map<String, Object>> getSensorsDataByDevice(Map<String, Object> parameters) {
      ValidateUtil.notNull(parameters.get("deviceVersionId"), "param.is.null");
      ValidateUtil.notNull(parameters.get("mac"), "param.is.null");
      ValidateUtil.notNull(parameters.get("deviceId"), "param.is.null");
       List<Map<String, Object>> sensors = sensorMapper.getSensorsByDeviceVersionId(parameters);
       parameters.put("size", 1);
       List<Map<String, Object>> values = historyMapper.getValueByMacAndSize(parameters);
       JSONObject map = null;
       if (!ObjectUtils.isEmpty(values)) {
          map = JSON.parseObject(values.get(0).get("value").toString());
      }
       Device device = deviceMapper.selectByPrimaryKey(Integer.valueOf((String) parameters.get("deviceId")));
       for (Map<String, Object> sensor : sensors) {
          if (!ObjectUtils.isEmpty(map)) {
             sensor.put("value", map.remove(sensor.get("sensor_key")));
         }
          sensor.put("state", device.getState());
          sensor.put("name", sensor.get("description"));
       }
      return sensors;
   }
   @Override
   public List<Map<String, Object>> getSensorDataBySensorKey(Map<String, Object> parameters) {
      ValidateUtil.notNull(parameters.get("sensorKey"), "param.is.null");
      ValidateUtil.notNull(parameters.get("mac"), "param.is.null");
      ValidateUtil.notNull(parameters.get("size"), "param.is.null");
      parameters.put("size", Integer.valueOf(parameters.remove("size").toString()));
      List<Map<String, Object>> values = historyMapper.getValueByMacAndSize(parameters);
      for (Map<String, Object> value : values) {
         JSONObject json = JSON.parseObject(value.remove("value").toString());
         value.put("value", json.get(parameters.get("sensorKey")));
      }
      return values;
   }
   @Override
   public Map<String, Object> getActualDataByRegion(Map<String, Object> parameters) {
      Object sensorKey = parameters.get("sensorKey");
      ValidateUtil.notNull(sensorKey, "param.is.null");
      ValidateUtil.notNull(parameters.get("accountId"), "param.is.null");
      ValidateUtil.notNull(parameters.get("interval"), "param.is.null");
      Integer interval = Integer.valueOf(parameters.get("interval").toString());
      LocalDateTime start, end;
      if (!parameters.containsKey("time")) {
         end = LocalDateTime.now();
         int second = end.getSecond() / interval * interval;
         end = end.truncatedTo(ChronoUnit.MINUTES).plusSeconds(second);
         start = end.minusSeconds(interval);
      } else {
         //start = LocalDateTime.ofEpochSecond(Long.valueOf(parameters.get("time").toString()) / 1000, 0, ZoneOffset.ofHours(8));
         start = LocalDateTime.ofInstant(Instant.ofEpochMilli(Long.valueOf(parameters.get("time").toString())),ZoneId.of("+8"));
         end = start.plusSeconds(interval);
      }
      parameters.put("start", start);
      parameters.put("end", end);
      parameters.put("sensorKeys", Arrays.asList(((String) sensorKey).split(",")));
      accountService.setOrgIdsByAccount(parameters);
      Map<String, Object> result = historyMapper.getActualDataByRegion(parameters);
      if (ObjectUtils.isEmpty(result)) {
         result = new HashMap<String, Object>();
      }
      result.put("time", end.toInstant(ZoneOffset.of("+8")).toEpochMilli());
      return result;
   }
}