jinpengyong
2020-04-30 d1f1b489eea7730bb1cb78cdae24c626db835298
src/main/java/com/moral/controller/ReportController.java
@@ -1,13 +1,11 @@
package com.moral.controller;
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 java.text.SimpleDateFormat;
import java.util.*;
import javax.annotation.Resource;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
@@ -20,6 +18,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 +31,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;
@@ -81,6 +83,110 @@
            exportColumn[index + 1] = new String[]{name, "10", key};
        }
        OutputStream outputStream = exportData(response, "Excel数据", list, exportColumn);
        outputStream.flush();
        outputStream.close();
        return new ResultBean<Boolean>(true);
    }
    @GetMapping("newExcel")
    public ResultBean<Boolean> getNExcelReport(HttpServletRequest request, HttpServletResponse response) throws Exception {
        Map<String, Object> parameters = getParametersStartingWith(request, null);
        List<Map<String, Object>> list = historyMinutelyService.getMonitorPointOrDeviceAvgData(parameters);
        System.out.println(list);
        String type = parameters.get("type").toString();
        String time = parameters.get("time").toString();
        System.out.println(parameters);
        String timeb = "";
        if (parameters.get("timeb") == null) {
            timeb = parameters.get("time").toString();
        } else {
            timeb = parameters.get("timeb").toString();
        }
        String[] endTimes = timeb.split("-");
        String dateFormat = "";
        String beginTime = "";
        String endTime = "";
        List<String> monthList1 = new ArrayList<>();
        monthList1.add("01");
        monthList1.add("03");
        monthList1.add("05");
        monthList1.add("07");
        monthList1.add("08");
        monthList1.add("10");
        monthList1.add("12");
        List<String> monthList2 = new ArrayList<>();
        monthList2.add("04");
        monthList2.add("06");
        monthList2.add("09");
        monthList2.add("11");
        Integer year = Integer.valueOf(endTimes[0]);
        int i = 0;
        if (type.equals("year")) {
            dateFormat = "yyyy-MM";
            beginTime = time + "-01";
            endTime = timeb + "-12";
            i = Calendar.MONTH;
        } else if (type.equals("month")) {
            dateFormat = "yyyy-MM-dd";
            beginTime = time + "-01";
            i = Calendar.DAY_OF_MONTH;
            if (monthList1.contains(endTimes[1])) {
                endTime = timeb + "-31";
            } else if (monthList2.contains(endTimes[1])) {
                endTime = timeb + "-30";
            } else {
                if (year % 4 == 0 && year % 100 != 0 || year % 400 == 0) {
                    endTime = timeb + "-29";
                } else {
                    endTime = timeb + "-28";
                }
            }
        } else if (type.equals("day")) {
            dateFormat = "yyyy-MM-dd HH";
            beginTime = time + " 00";
            endTime = timeb + " 23";
            i = Calendar.HOUR_OF_DAY;
        }
        SimpleDateFormat sdf = new SimpleDateFormat(dateFormat);
        Calendar cal = Calendar.getInstance();
        cal.setTime(sdf.parse(beginTime));
        ArrayList<Map<String, Object>> resultList = new ArrayList<>();
        for (long d = cal.getTimeInMillis(); d <= sdf.parse(endTime).getTime(); cal.set(i, cal.get(i) + 1), d = cal.getTimeInMillis()) {
            String format = sdf.format(d);
            Map<String, Object> map = new HashMap<>();
            map.put("time", format);
            resultList.add(map);
        }
        for (Map<String, Object> map : resultList) {
            for (Map<String, Object> resultMap : list) {
                if (resultMap.get("time").equals(map.get("time"))) {
                    map.putAll(resultMap);
                }
            }
        }
        List<String> sensors = (List<String>) parameters.get("sensors");//[e1,e2]
        String[][] exportColumn = new String[sensors.size() + 1][];
        exportColumn[0] = new String[]{"时间", "20", "time"};
        for (
                int index = 0; index < sensors.size(); index++) {
            String[] split = sensors.get(index).split("-");
            String name = split[1];
            String key = split[0];
            String unit = split[2];
            if (!ObjectUtils.isEmpty(unit) && !"null".equals(unit)) {
                name += "-" + unit;
            }
            exportColumn[index + 1] = new String[]{name, "10", key};
        }
        OutputStream outputStream = exportData(response, "Excel数据", resultList, exportColumn);
        outputStream.flush();
        outputStream.close();
        return new ResultBean<Boolean>(true);
@@ -139,9 +245,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 +275,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);
    }
}