From b1935a09641bc915c61f1868d911adf455d79833 Mon Sep 17 00:00:00 2001
From: ZhuDongming <773644075@qq.com>
Date: Tue, 28 Apr 2020 09:33:41 +0800
Subject: [PATCH] update大屏本月平均值查询表

---
 src/main/java/com/moral/controller/ReportController.java |   99 +++++++++++++++++++++++++++++++++++++++----------
 1 files changed, 78 insertions(+), 21 deletions(-)

diff --git a/src/main/java/com/moral/controller/ReportController.java b/src/main/java/com/moral/controller/ReportController.java
index c86ffbb..b4ef1d7 100644
--- a/src/main/java/com/moral/controller/ReportController.java
+++ b/src/main/java/com/moral/controller/ReportController.java
@@ -2,12 +2,12 @@
 
 import java.io.OutputStream;
 import java.util.ArrayList;
-import java.util.Arrays;
 import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
 
 import javax.annotation.Resource;
+import javax.servlet.ServletOutputStream;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 
@@ -20,6 +20,7 @@
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
+import com.alibaba.fastjson.JSONObject;
 import com.moral.common.bean.ResultBean;
 import com.moral.common.util.ParameterUtils;
 import com.moral.common.util.WebUtils;
@@ -32,6 +33,9 @@
 import com.moral.service.HistoryDailyService;
 import com.moral.service.HistoryMinutelyService;
 import com.moral.service.MonitorPointService;
+
+import cn.hutool.core.io.IoUtil;
+import cn.hutool.poi.excel.ExcelWriter;
 
 import static com.moral.common.util.ExportExcelUtils.exportData;
 import static com.moral.common.util.WebUtils.getParametersStartingWith;
@@ -139,9 +143,26 @@
     public ResultBean<Boolean> getCustomMadeExcelReport(HttpServletRequest request, HttpServletResponse response) throws Exception {
         Map<String, Object> parameters = getParametersStartingWith(request, null);
         ParameterUtils.getTimeType4Time(parameters);
-        Object sensorKey = parameters.get("sensorKey");
-        String sensorInfo = parameters.get("sensorInfo").toString();
-        parameters.put("sensorKeys", Arrays.asList(sensorKey));
+        String sensorsInfo = parameters.get("sensors").toString();
+        String[] sensorsStr = sensorsInfo.split(",");
+        List<String> sensorKeys = new ArrayList<>();
+        List<String> sensorKeysNames = new ArrayList<>();
+        List<String> sensorKeysUnits = new ArrayList<>();
+        for (int index = 0; index < sensorsStr.length; index++) {
+            String[] split = sensorsStr[index].split("-");
+            String key;
+            if (index == 0) {
+                key = split[0].replace("\"", "").substring(1);
+            } else {
+                key = split[0].replace("\"", "");
+            }
+            String name = split[1].replace("\"", "");
+            String Unit = split[2].replace("\"", "");
+            sensorKeys.add(key);
+            sensorKeysNames.add(name);
+            sensorKeysUnits.add(Unit);
+        }
+        parameters.put("sensorKeys", sensorKeys);
         List<String> macList = new ArrayList<>();
         List<MonitorPoint> monitorPoints = monitorPointService.getMonitorPointsAndDevicesByRegion(parameters);
         for (MonitorPoint m : monitorPoints) {
@@ -152,27 +173,63 @@
         parameters.put("macs", macList);
         List<Map<String, Object>> list = new ArrayList<>();
         if (!CollectionUtils.isEmpty(macList)) {
-            list = historyMinutelyService.getDevicesAvgDataToExcel(parameters);
+            list = historyMinutelyService.getDevicesSensorsAvgDataToExcel(parameters);
         }
-        Map<String, Object> map = list.get(0);
-        Map<Integer, String> mapNew = new LinkedHashMap<>();
-        int i = 0;
-        for (String key : map.keySet()) {
-            if (!("monitorPointName".equals(key) || "name".equals(key))) {
-                mapNew.put(i, key);
-                i++;
+        List<List<Map<String, Object>>> sheets = new ArrayList<>();
+        for (int i = 0; i < sensorKeys.size(); i++) {
+            List<Map<String, Object>> sheet = new ArrayList<>();
+            for (int j = 0; j < list.size(); j++) {
+                Map<String, Object> data = new LinkedHashMap<>();
+                for (Map.Entry<String, Object> kv : list.get(j).entrySet()) {
+                    if ("monitorPointName".equals(kv.getKey())) {
+                        data.put("������������(������:" + sensorKeysUnits.get(i) + ")", kv.getValue());
+                    } else if ("name".equals(kv.getKey())) {
+                        data.put("������������", kv.getValue());
+                    } else {
+                        String sensorsValue = kv.getValue().toString();
+                        JSONObject jsonObject = JSONObject.parseObject(sensorsValue);
+                        if (jsonObject != null) {
+                            List<Object> sensorsValueList = (List<Object>) jsonObject.get(sensorKeys.get(i));
+                            if (sensorsValueList != null) {
+                                data.put(kv.getKey(), sensorsValueList.get(0));
+                            } else {
+                                data.put(kv.getKey(), "");
+                            }
+                        } else {
+                            data.put(kv.getKey(), "");
+                        }
+                    }
+                }
+                sheet.add(data);
             }
+            sheets.add(sheet);
         }
-        String[][] exportColumn = new String[map.size()][];
-        exportColumn[0] = new String[]{"������������-" + sensorInfo, "30", "monitorPointName"};
-        exportColumn[1] = new String[]{"������������", "20", "name"};
-        for (int index = 0; index < map.size() - 2; index++) {
-            exportColumn[index + 2] = new String[]{mapNew.get(index), "20", mapNew.get(index)};
+        if (!CollectionUtils.isEmpty(sheets)) {
+            ExcelWriter writer = new ExcelWriter(false, sensorKeysNames.get(0));
+            writer.write(sheets.get(0), true);
+            writer.autoSizeColumnAll();
+            writer.setColumnWidth(0, 25);
+            writer.setColumnWidth(1, 25);
+            if (sheets.size() >= 2) {
+                for (int i = 1; i < sheets.size(); i++) {
+                    writer.setSheet(sensorKeysNames.get(i));
+                    writer.write(sheets.get(i), true);
+                    writer.autoSizeColumnAll();
+                    writer.setColumnWidth(0, 25);
+                    writer.setColumnWidth(1, 25);
+                }
+            }
+            response.setContentType("application/vnd.ms-excel;charset=utf-8");
+            String codedFileName = java.net.URLEncoder.encode("Excel������", "UTF-8");
+            response.setHeader("Content-Disposition", "attachment;filename=" + codedFileName + ".xls");
+            ServletOutputStream out = response.getOutputStream();
+            writer.flush(out, true);
+            // ������writer���������������
+            writer.close();
+            //������������������������Servlet���
+            IoUtil.close(out);
         }
-        OutputStream outputStream = exportData(response, "Excel������", list, exportColumn);
-        outputStream.flush();
-        outputStream.close();
-        return new ResultBean<Boolean>(true);
+        return new ResultBean<>(true);
     }
 
 }

--
Gitblit v1.8.0