From 2e2861fd0bbba403a15a212c550bed9f8331f467 Mon Sep 17 00:00:00 2001
From: xufenglei <xufenglei>
Date: Mon, 15 Jan 2018 17:14:13 +0800
Subject: [PATCH] 更新 查询

---
 src/main/java/com/moral/monitor/service/impl/ScreenServiceImpl.java |  320 ++++++++++++++++++++++++++++++++++++++++------------
 1 files changed, 243 insertions(+), 77 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 eb35a0d..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,96 +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"));
 
-		// 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));
+			map.put("start", DateUtils.truncate(DateUtils.addMonths(date, -1), Calendar.MONTH));
 		} else {
 			// ���������1���00:00:00
-			parameters.put("start", DateUtils.truncate(date, Calendar.MONTH));
+			map.put("start", DateUtils.truncate(date, Calendar.MONTH));
 		}
-		// ������00:00:00
-		parameters.put("end", DateUtils.truncate(date, Calendar.DATE));
-
-		Map<String, Object> average = null;
-		average = historyMapper.getMonthAverageBySensor(parameters);
-
+		average = historyMapper.getMonthAverageBySensor(map);
 		if (ObjectUtils.isEmpty(average)) {
-			// ������������������
-			//resultMap.put("average", 50.3467 + Calendar.getInstance().get(Calendar.DAY_OF_MONTH) / 5);
 			resultMap.put("average", "N/A");
 		} else {
-			//resultMap.putAll(average);
-			//resultMap.put("average", String.format("%.2f", resultMap.get("average")));
 			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)) {
-				continue;
-			} else {
-				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));
-			}
+			IAQIs.add(cs.take().get());
 		}
-		
-		if (!resultMap.containsKey("AQI")) {
-			if (ObjectUtils.isEmpty(IAQIs)) {
-				resultMap.put("AQI", "N/A");
-			}else {
-				double AQI = Collections.max(IAQIs);
-				// ������������������
-				/*if (AQI == 0.0) {
-					AQI = 60.670;
-				}*/
+		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;
 	}
 
@@ -142,19 +162,6 @@
 		// ������������ -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"));
@@ -229,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;
+	}
 }

--
Gitblit v1.8.0