From 03cdc1111aa07041d635840eb30433cbb980accf Mon Sep 17 00:00:00 2001 From: 沈斌 <bluelazysb@hotmail.com> Date: Wed, 28 Feb 2018 16:44:48 +0800 Subject: [PATCH] test --- src/main/java/com/moral/monitor/service/impl/ScreenServiceImpl.java | 349 ++++++++++++++++++++++++++++++++++++++++++++------------- 1 files changed, 269 insertions(+), 80 deletions(-) diff --git a/src/main/java/com/moral/monitor/service/impl/ScreenServiceImpl.java b/src/main/java/com/moral/monitor/service/impl/ScreenServiceImpl.java index e01d765..50be408 100644 --- a/src/main/java/com/moral/monitor/service/impl/ScreenServiceImpl.java +++ b/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,90 +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", 90.0384 + 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)); } - - // 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)); - 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"); - } - 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) { - 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)); - } + average = historyMapper.getMonthAverageBySensor(map); + if (ObjectUtils.isEmpty(average)) { + resultMap.put("average", "N/A"); + } else { + resultMap.put("average", String.format("%.2f", average.get("average"))); } - double AQI = Collections.max(IAQIs); - // ������������������ - if (AQI == 0.0) { - AQI = 231.670; + Set<Double> IAQIs = new HashSet<Double>(); + for (String macKey : macKeys) { + IAQIs.add(cs.take().get()); } - if (!resultMap.containsKey("AQI")) { - resultMap.put("AQI", String.format("%.2f", AQI)); + 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; } @@ -129,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")); @@ -206,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; + } } -- Gitblit v1.8.0