工业级运维app手机api
沈斌
2018-02-28 03cdc1111aa07041d635840eb30433cbb980accf
src/main/java/com/moral/monitor/service/impl/ScreenServiceImpl.java
@@ -1,10 +1,14 @@
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;
@@ -18,9 +22,14 @@
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.mongodb.BasicDBObject;
import com.moral.monitor.dao.AccountEntityMapper;
import com.moral.monitor.dao.HistoryEntityMapper;
import com.moral.monitor.entity.AccountEntity;
@@ -43,6 +52,9 @@
   @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);
@@ -120,11 +132,11 @@
         resultMap.put("average", String.format("%.2f", average.get("average")));
      }
      
      List<Double> IAQIs = new ArrayList<Double>();
      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 {
@@ -224,4 +236,163 @@
      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;
   }
}