更新 查询

xufenglei

2018-01-15 2e2861fd0bbba403a15a212c550bed9f8331f467

 src/main/java/com/moral/monitor/service/impl/ScreenServiceImpl.java

@@ -1,22 +1,35 @@
package com.moral.monitor.service.impl;



import java.util.ArrayList;

import static org.apache.commons.lang3.time.DateUtils.addDays;

import static org.apache.commons.lang3.time.DateUtils.truncate;

import static org.springframework.util.ObjectUtils.isEmpty;



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 java.util.concurrent.Callable;

import java.util.concurrent.CompletionService;

import java.util.concurrent.ExecutorCompletionService;

import java.util.concurrent.ExecutorService;

import java.util.concurrent.Executors;



import org.apache.commons.lang3.StringUtils;

import org.apache.commons.lang3.time.DateUtils;

import org.springframework.beans.factory.annotation.Autowired;

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.alibaba.fastjson.JSON;

import com.mongodb.BasicDBObject;

import com.moral.monitor.dao.AccountEntityMapper;

import com.moral.monitor.dao.HistoryEntityMapper;

import com.moral.monitor.entity.AccountEntity;

@@ -27,6 +40,7 @@
import com.moral.monitor.util.ResourceUtil;



@Service

@SuppressWarnings("unused")

public class ScreenServiceImpl implements ScreenService {



   @Autowired

@@ -38,94 +52,102 @@
   @Autowired

   private OrganizationService organizationService;



   @Autowired

   private MongoTemplate mongoTemplate;

	
   public List<AccountEntity> getAccountLists(String account, String password) {

      AccountEntityExample example = new AccountEntityExample();

      example.or().andAccountEqualTo(account).andPasswordEqualTo(password);

      return accountMapper.selectByExample(example);

   }



   public Map<String, Object> getMonthDataByEquipment(Map<String, Object> parameters) {

   public Map<String, Object> getMonthDataByEquipment(final Map<String, Object> parameters) throws Exception {

      Map<String, Object> resultMap = new HashMap<String, Object>();



      // 1.0 实际标准值

      resultMap.put("standard", ResourceUtil.getValue(parameters.get("macKey") + "-standard"));



      //resultMap.put("average", 90.0384 + Calendar.getInstance().get(Calendar.DAY_OF_MONTH) / 5);

		
      //resultMap.put("AQI", 236.71 + Calendar.getInstance().get(Calendar.DAY_OF_MONTH) / 5);



		
      // 2.0 本月平均值

      // 2.0 AQI 指数

      Date date = new Date();

      // 昨日00:00:00

      parameters.put("start", DateUtils.truncate(DateUtils.addDays(date, -1), Calendar.DATE));

      // 今日00:00:00

      parameters.put("end", DateUtils.truncate(date, Calendar.DATE));

      String[] macKeys = { "e1", "e2", "e10", "e11", "e15", "e16" };

      final String[] IAQIValues = ResourceUtil.getArrValue("IAQI");

      ExecutorService threadPool = Executors.newCachedThreadPool();

      CompletionService<Double> cs = new ExecutorCompletionService<Double>(threadPool);

      for (final String macKey : macKeys) {

         cs.submit(new Callable<Double>() {

            @Override

            public Double call() throws Exception {

               Double result = null;

               double avg = 0, minMacKey = 0, maxMacKey = 0, minIAQI = 0, maxIAQI = 0;

               Map<String, Object> map = new HashMap<String, Object>(parameters);

               map.put("macKey", macKey);

               Map<String, Object> average = historyMapper.getMonthAverageBySensor(map);

               if (!ObjectUtils.isEmpty(average)) {

                  avg = (Double) average.get("average");

                  String[] macKeyValues = ResourceUtil.getArrValue(macKey);

                  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) {

                     result = 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]);

                     result = calculateIAQI(maxIAQI, minIAQI, maxMacKey, minMacKey, avg);

                  }

               }

               return result;

            }

         });

      }



      // 3.0 本月平均值

      Map<String, Object> average = null;

      Map<String, Object> map = new HashMap<String, Object>(parameters);

      // 每月一日的数据取上月的数据

      if (1 == Calendar.getInstance().get(Calendar.DAY_OF_MONTH)) {

         // 上个月1日00:00:00

         parameters.put("start", DateUtils.truncate(DateUtils.addMonths(date, -1), Calendar.MONTH));

      }

      // 这个月1日00:00:00

      parameters.put("start", DateUtils.truncate(date, Calendar.MONTH));

      // 今日00:00:00

      parameters.put("end", DateUtils.truncate(date, Calendar.DATE));



      Map<String, Object> average = historyMapper.getMonthAverageBySensor(parameters);



      if (ObjectUtils.isEmpty(average)) {

         // 空值临时处理

         resultMap.put("average", 50.3467 + Calendar.getInstance().get(Calendar.DAY_OF_MONTH) / 5);

         map.put("start", DateUtils.truncate(DateUtils.addMonths(date, -1), Calendar.MONTH));

      } else {

         resultMap.putAll(average);

         // 这个月1日00:00:00

         map.put("start", DateUtils.truncate(date, Calendar.MONTH));

      }

      resultMap.put("average", String.format("%.2f", resultMap.get("average")));

      average = historyMapper.getMonthAverageBySensor(map);

      if (ObjectUtils.isEmpty(average)) {

         resultMap.put("average", "N/A");

      } else {

         resultMap.put("average", String.format("%.2f", average.get("average")));

      }

      

      // 3.0 AQI 指数

      String[] macKeys = { "e1", "e2", "e10", "e11", "e15", "e16" };

      String[] IAQIValues = ResourceUtil.getArrValue("IAQI");

      List<Double> IAQIs = new ArrayList<Double>();

      // 昨日00:00:00

      parameters.put("start", DateUtils.truncate(DateUtils.addDays(date, -1), Calendar.DATE));

      Set<Double> IAQIs = new HashSet<Double>();

      for (String macKey : macKeys) {

         double avg = 0, minMacKey = 0, maxMacKey = 0, minIAQI = 0, maxIAQI = 0;

         parameters.put("macKey", macKey);

         average = historyMapper.getMonthAverageBySensor(parameters);

         if (!ObjectUtils.isEmpty(average)) {

            avg = (Double) average.get("average");

            if ("e11".equals(macKey) || "e15".equals(macKey) || "e16".equals(macKey)) {

               avg *= 1000.0;

            }

         }

         String[] macKeyValues = ResourceUtil.getArrValue(macKey);

         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(cs.take().get());

      }

      IAQIs.remove(null);

      if (ObjectUtils.isEmpty(IAQIs)) {

         resultMap.put("AQI", "N/A");

      } else {

         Double AQI = Collections.max(IAQIs);

         if (AQI == Double.MAX_VALUE) {

            resultMap.put("AQI", IAQIValues[IAQIValues.length - 1]);

            break;

         } else {

            minMacKey = Double.valueOf(macKeyValues[index]);

            maxMacKey = Double.valueOf(macKeyValues[index + 1]);

            minIAQI = Double.valueOf(IAQIValues[index]);

            maxIAQI = Double.valueOf(IAQIValues[index + 1]);

            IAQIs.add(calculateIAQI(maxIAQI, minIAQI, maxMacKey, minMacKey, avg));

            resultMap.put("AQI", String.format("%.0f", AQI));

         }

      }

		
      double AQI = Collections.max(IAQIs);

      // 空值临时处理

      if (AQI == 0.0) {

         AQI = 60.670;

      }

      if (!resultMap.containsKey("AQI")) {

         resultMap.put("AQI", String.format("%.0f", AQI));

      }

		 


      return resultMap;

   }



@@ -133,26 +155,13 @@
      Map<String, Object> result = new LinkedHashMap<String, Object>();



      setOrgIdsByAccount(parameters);



      parameters.put("macKey", "all");

      Date date = new Date();

      // 当前时间 -10分钟

      parameters.put("start", DateUtils.addMinutes(date, -10));

      // 当前时间 -5分钟

      parameters.put("end", DateUtils.addMinutes(date, -5));

      List<Map<String, Object>> averageByAll = historyMapper.getAverageByAll(parameters);



      // 临时方案，空值处理

      if (ObjectUtils.isEmpty(averageByAll)) {

         String macLog = historyMapper.getMacLogByLast();

         if (StringUtils.isNotBlank(macLog)) {

            Map<String, Object> map = JSON.parseObject(macLog);

            for (String key : map.keySet()) {

               if (key.startsWith("e")) {

                  result.put(key, map.get(key));

               }

            }

         }

      }



      for (Map<String, Object> map : averageByAll) {

         result.put((String) map.get("mac_key"), map.get("avg"));

@@ -210,4 +219,180 @@
      return (maxIAQI - minIAQI) * (avg - minMacKey) / (maxMacKey - minMacKey) + minIAQI;

   }



   @Override

   public Map<String, Object> getAverageBySensor(Map<String, Object> parameters) {

      Map<String, Object> result = new LinkedHashMap<String, Object>();



      setOrgIdsByAccount(parameters);

      Date date = new Date();

      // 当前时间 -1小时

      parameters.put("start", DateUtils.addHours(date, -1));

      parameters.put("end", date);

      List<Map<String, Object>> averageByAll = historyMapper.getAverageByAll(parameters);



      for (Map<String, Object> map : averageByAll) {

         result.put((String) map.get("name"), map.get("avg"));

      }

      return result;

   }



   @Override

   public Map<String, Object> getMonthAverageBySensor(Map<String, Object> parameters) {

      Map<String, Object> result = new HashMap<String, Object>();

      Date date = new Date();

      //Long end = DateUtils.truncate(date, Calendar.DATE).getTime(),start;

      Date end = DateUtils.truncate(date, Calendar.DATE),start;

		
      // 每月一日的数据取上月的数据

      if (1 == Calendar.getInstance().get(Calendar.DAY_OF_MONTH)) {

         // 上个月1日00:00:00

         //start = DateUtils.truncate(DateUtils.addMonths(date, -1), Calendar.MONTH).getTime();

         start = DateUtils.truncate(DateUtils.addMonths(date, -1), Calendar.MONTH);

      } else {

         // 这个月1日00:00:00

         //start =  DateUtils.truncate(date, Calendar.MONTH).getTime();

         start =  DateUtils.truncate(date, Calendar.MONTH);

      }

      start = DateUtils.addHours(start, 8);

      end = DateUtils.addHours(end, 8);

      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("avg")

         );

      //AggregationResults<BasicDBObject> results = mongoTemplate.aggregate(aggregation, "data",BasicDBObject.class);

      AggregationResults<BasicDBObject> results = mongoTemplate.aggregate(aggregation, "data_hourly",BasicDBObject.class);

      List<BasicDBObject> list = results.getMappedResults();

      if (!ObjectUtils.isEmpty(list)) {

         result = list.get(0);

      }

      return result;

   }



   @Override

   public Map<String, Object> getDayAQIBySensor(final Map<String, Object> parameters) throws Exception {

      Map<String, Object> resultMap = new HashMap<String, Object>();

      Date date = new Date();

      // 昨日00:00:00

      parameters.put("start", DateUtils.truncate(DateUtils.addDays(date, -1), Calendar.DATE));

      // 今日00:00:00

      parameters.put("end", DateUtils.truncate(date, Calendar.DATE));

      String[] macKeys = { "e1", "e2", "e10", "e11", "e15", "e16" };

      final String[] IAQIValues = ResourceUtil.getArrValue("IAQI");

      ExecutorService threadPool = Executors.newCachedThreadPool();

      CompletionService<Double> cs = new ExecutorCompletionService<Double>(threadPool);

      for (final String macKey : macKeys) {

         cs.submit(new Callable<Double>() {

            @Override

            public Double call() throws Exception {

               Double result = null;

               double avg = 0, minMacKey = 0, maxMacKey = 0, minIAQI = 0, maxIAQI = 0;

               Map<String, Object> map = new HashMap<String, Object>(parameters);

               map.put("macKey", macKey);

               Map<String, Object> average = historyMapper.getMonthAverageBySensor(map);

               if (!ObjectUtils.isEmpty(average)) {

                  avg = (Double) average.get("average");

                  String[] macKeyValues = ResourceUtil.getArrValue(macKey);

                  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) {

                     result = 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]);

                     result = calculateIAQI(maxIAQI, minIAQI, maxMacKey, minMacKey, avg);

                  }

               }

               return result;

            }

         });

      }

      Set<Double> IAQIs = new HashSet<Double>();

      for (String macKey : macKeys) {

         IAQIs.add(cs.take().get());

      }

      IAQIs.remove(null);

      if (ObjectUtils.isEmpty(IAQIs)) {

         resultMap.put("AQI", "N/A");

      } 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<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", "N/V");

      } 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 = calculateIAQI(maxIAQI, minIAQI, maxMacKey, minMacKey, avg);

                  IAQIs.add(result);

               }

            }

         }

         IAQIs.remove(null);

         if (isEmpty(IAQIs)) {

            resultMap.put("AQI", "N/V");

         } 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;

   }

}