From c20b17fee68aa391bbbdaed34d1ee22f1ed21448 Mon Sep 17 00:00:00 2001
From: 16639036659 <577530412@qq.com>
Date: 星期日, 07 四月 2024 13:23:58 +0800
Subject: [PATCH] 日报功能开启

---
 src/main/java/com/gkhy/fourierSpecialGasMonitor/schedule/DailyReportSchedule.java |   91 ++++++++++++++++++++++++++++++++-------------
 1 files changed, 64 insertions(+), 27 deletions(-)

diff --git a/src/main/java/com/gkhy/fourierSpecialGasMonitor/schedule/DailyReportSchedule.java b/src/main/java/com/gkhy/fourierSpecialGasMonitor/schedule/DailyReportSchedule.java
index 2086fb0..fdc1deb 100644
--- a/src/main/java/com/gkhy/fourierSpecialGasMonitor/schedule/DailyReportSchedule.java
+++ b/src/main/java/com/gkhy/fourierSpecialGasMonitor/schedule/DailyReportSchedule.java
@@ -1,8 +1,10 @@
 package com.gkhy.fourierSpecialGasMonitor.schedule;
 
 import com.alibaba.fastjson.JSON;
+import com.gkhy.fourierSpecialGasMonitor.Application;
 import com.gkhy.fourierSpecialGasMonitor.commons.enums.ForeignResultCode;
 import com.gkhy.fourierSpecialGasMonitor.commons.enums.ResultCode;
+import com.gkhy.fourierSpecialGasMonitor.commons.enums.SystemCacheKeyEnum;
 import com.gkhy.fourierSpecialGasMonitor.commons.exception.BusinessException;
 import com.gkhy.fourierSpecialGasMonitor.commons.exception.DataReceiveException;
 import com.gkhy.fourierSpecialGasMonitor.config.file.ReportFilePathConfig;
@@ -26,6 +28,8 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.core.io.ClassPathResource;
+import org.springframework.core.io.Resource;
 import org.springframework.scheduling.annotation.Async;
 import org.springframework.scheduling.annotation.Scheduled;
 import org.springframework.stereotype.Component;
@@ -39,6 +43,7 @@
 import javax.xml.parsers.ParserConfigurationException;
 import java.io.*;
 import java.lang.reflect.Field;
+import java.net.URL;
 import java.time.LocalDate;
 import java.time.LocalDateTime;
 import java.time.LocalTime;
@@ -64,6 +69,8 @@
     private static final DateTimeFormatter lineChartXDataDisplay = DateTimeFormatter.ofPattern("HH:mm");
 
     private final Logger logger = LoggerFactory.getLogger(this.getClass());
+
+    private static int serialNumber = 4;
 
     @Autowired
     private GasWarnLogService gasWarnLogService;
@@ -91,25 +98,46 @@
     public void generateDailyReport() {
         LocalDateTime now = LocalDateTime.now();
         String startTime = now.format(execformatter);
-        RBucket<List<GasCategory>> bucket = redissonClient.getBucket("gas_category_cache_info");
+        RBucket<List<GasCategory>> bucket = redissonClient.getBucket(SystemCacheKeyEnum.KEY_GAS_CATEGORY.getKey());
         List<GasCategory> gasCategories = bucket.get();
+        serialNumber = 4;
         logger.info("【##】开始生成日报 ,时间:"+startTime);
         OPCPackage opcPackage = null;
         //加载文档
         XWPFDocument doc = null;
         try {
-            FileInputStream originalFileInputStream = new FileInputStream("src/main/resources/template/dailyReportTemplate.docx");
-            FileOutputStream copyFileOutputStream = new FileOutputStream("src/main/resources/temp/dailyReportTemplate.docx");
-            // 创建副本文件
+            ClassLoader classLoader = Application.class.getClassLoader();
+            InputStream originalFileInputStream = classLoader.getResourceAsStream("template/dailyReportTemplate.docx");
+            String copyFileUrl = reportFilePathConfig.getDcPath()+"temp/dailyReportTemplate.docx";
+            File file = new File(copyFileUrl);
+            if (!file.exists()){
+                try {
+                    // 获取文件所在的文件夹路径
+                    String folderPath = file.getParent();
+                    // 创建文件夹(包括父目录)
+                    File folder = new File(folderPath);
+                    if (!folder.exists()){
+                        folder.mkdirs();
+                    }
+                    file.createNewFile();
+                } catch (IOException e) {
+                    e.printStackTrace();
+                }
+            }
+            FileOutputStream copyFileOutputStream = new FileOutputStream(copyFileUrl);
             IOUtils.copy(originalFileInputStream, copyFileOutputStream);
-            opcPackage = POIXMLDocument.openPackage("src/main/resources/temp/dailyReportTemplate.docx");
+            opcPackage = POIXMLDocument.openPackage(copyFileUrl);
+
+            //FileInputStream originalFileInputStream = new FileInputStream("src/main/resources/template/dailyReportTemplate.docx");
+            //FileOutputStream copyFileOutputStream = new FileOutputStream("src/main/resources/temp/dailyReportTemplate.docx");
+            //// 创建副本文件
+            //IOUtils.copy(originalFileInputStream, copyFileOutputStream);
+            //opcPackage = POIXMLDocument.openPackage("src/main/resources/temp/dailyReportTemplate.docx");
             doc = new XWPFDocument(opcPackage);
         } catch (IOException e) {
             e.printStackTrace();
         }
-
         Map<String, Object> map = dataMap(gasCategories);
-
         List<XWPFParagraph> paragraphList = doc.getParagraphs();
         for (XWPFParagraph par : paragraphList) {
             //获取段落的文本对象
@@ -167,13 +195,13 @@
         List<GasConcentration> gasConcentrations = gasConcentrationService.listDatabyTimeSlot(LocalDateTime.of(now.minusDays(1).toLocalDate(), LocalTime.MIN)
                 ,  LocalDateTime.of(now.minusDays(1).toLocalDate(), LocalTime.MAX));
         if (!CollectionUtils.isEmpty(gasConcentrations)) {
-            for (int i = 0; i < 30; i++) {
+            //for (int i = 0; i < 30; i++) {
+            for (int i = 0; i < gasCategories.size(); i++) {
                 String series = gasCategories.get(i).getMolecularFormula() + "浓度观测结果";
-                String title = "—" + gasCategories.get(i).getMolecularFormula();
+                String title = gasCategories.get(i).getMolecularFormula();
                 drawLineChart(gasConcentrations, fileurl, series, title, gasCategories.get(i).getMolecularFormula(), i);
             }
         }
-
         //List<GasFlux> gasFluxes = gasFluxService.listYesterday();
         List<GasFlux> gasFluxes = gasFluxService.listYesterdayTenAmToSixPm();
         List<Integer> areaNum = gasFluxes.stream().map(GasFlux::getAreaId).distinct().collect(Collectors.toList());
@@ -184,9 +212,13 @@
         Map<Integer, String> regionMap = allRegion.stream()
                 .collect(Collectors.toMap(Region::getId, Region::getName));
         if (!CollectionUtils.isEmpty(gasFluxes)) {
-            for (int i = 1; i <= areaNum.size(); i++) {
-                for (int j = 0; j < 20; j++) {
-                    drawBarChart(gasFluxes, fileurl, regionMap.get(i), "柱形图" + gasCategories.get(j).getMolecularFormula(), i, j + 1);
+            for (int j = 0; j < gasCategories.size(); j++) {
+                for (int i = 1; i <= areaNum.size(); i++) {
+                    //for (int j = 0; j < 20; j++) {
+                    int finalI = i;
+                    List<GasFlux> gasFluxesByArea = gasFluxes.stream().filter(gasFlux -> gasFlux.getAreaId() == finalI)
+                            .collect(Collectors.toList());
+                    drawBarChart(gasFluxesByArea, fileurl, regionMap.get(i), gasCategories.get(j).getMolecularFormula(), i, j + 1);
                 }
             }
         }
@@ -203,6 +235,10 @@
         logger.info("【##】日报生成成功!!! ,时间:"+endTime+",所耗时间: "+execTime+"s");
     }
 
+    /**
+     * @decription 画柱形图
+     * @author Mr.huang
+     */
     public void drawBarChart(List<GasFlux> gasFluxes,String fileurl,String series,String molecularFormula,Integer i,Integer j) {
         List<String> collect = gasFluxes.stream()
                 .map(gasFlux -> gasFlux.getTime().format(lineChartXDataDisplay))
@@ -242,6 +278,7 @@
         value2.add(n);
 
         List<XWPFChart> charts = doc.getCharts();
+        serialNumber++;
         for (int k = 0; k < charts.size(); k++) {
             XWPFChart xwpfChart = charts.get(k);
             XDDFTitle xddfTitletitle = xwpfChart.getTitle();
@@ -260,9 +297,9 @@
                     Element aT = (Element) root.getElementsByTagName("a:t").item(0);
                     // 获取 <a:t> 元素的文本内容 获取标题具体内容
                     String value = aT.getTextContent();
-                    String chartTitle = "图"+(31 + i * j)+" "+i+"号厂区排放"+molecularFormula+"通量反演结果";
-                    if ((i+"-"+j).equals(value)) {
-                        drawBarChartExec(charts.get(k), series2, x2, value2, chartTitle);
+                    String chartTitle = "图"+serialNumber+" "+i+"号厂区排放"+molecularFormula+"通量反演结果";
+                    if ((j+"-"+i).equals(value)) {
+                        drawBarChartExec(charts.get(k), series2, x2, value2, chartTitle,molecularFormula);
                     }
                 } catch (ParserConfigurationException e) {
                     e.printStackTrace();
@@ -370,12 +407,12 @@
 
 
     public void drawBarChartExec(XWPFChart chart, String[] series, String[] categories,
-                                  List<Number[]> values, String chartTitle) {
+                                  List<Number[]> values, String chartTitle,String molecularFormula) {
         final List<XDDFChartData> data = chart.getChartSeries();
 
         final XDDFBarChartData line = (XDDFBarChartData) data.get(0);//这里一般获取第一个,我们这里是折线图就是XDDFLineChartData
         line.getCategoryAxis().setTitle(chartTitle);
-        line.getValueAxes().get(0).setTitle("C2H2 Emission Rate (ug/m^2/s)");
+        line.getValueAxes().get(0).setTitle(molecularFormula+" Emission Rate (ug/m^2/s)");
         final int numOfPoints = categories.length;
 
         final String categoryDataRange = chart.formatRange(new CellRangeAddress(1, numOfPoints, 0, 0));
@@ -439,8 +476,8 @@
         Map<String, Object> map = new HashMap<>();
         map.put("${today}", today);
         map.put("${yesterday}", yesterday);
-        map.put("${generalradius}","10");
-        map.put("${generalradiu}","20");
+        map.put("${generalradius}","1km");
+        map.put("${generalradiu}","300m");
 
         List<GasWarnLog> gasWarnLogs = gasWarnLogService.listYesterday();
         if (!CollectionUtils.isEmpty(gasWarnLogs)){
@@ -459,21 +496,21 @@
         }
         String fiveGas = "";
         String sixGas = "";
-        for (int i = 0; i < 6; i++) {
+        for (int i = 0; i < 3; i++) {
             String name = gasCategories.get(i).getMolecularFormula()+"("+gasCategories.get(i).getName()+")";
             name = name + "、";
-            if (i < 5){
+            if (i < 3){
                 fiveGas = fiveGas + name;
-                if (i == 4){
+                if (i == 2){
                     int index = fiveGas.lastIndexOf("、");
-                    fiveGas =  fiveGas.substring(0, index)+"共五种";
+                    fiveGas =  fiveGas.substring(0, index)+"共三种";
                 }
             }
-            if (i < 6){
+            if (i < 3){
                 sixGas = sixGas + name;
-                if (i == 5){
+                if (i == 2){
                     int index = sixGas.lastIndexOf("、");
-                    sixGas =  sixGas.substring(0, index)+"共六种";
+                    sixGas =  sixGas.substring(0, index)+"共三种";
                 }
             }
         }

--
Gitblit v1.9.2