优化生成疾控周报

liutongyi-admin/src/main/java/com/citygis/web/service/impl/TabSystemMsgServiceImpl.java

@@ -1,19 +1,24 @@
 package com.citygis.web.service.impl;
 
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.util.MapUtils;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.citygis.common.annotation.DataSource;
 import com.citygis.common.enums.DataSourceType;
+import com.citygis.framework.datasource.DynamicDataSourceContextHolder;
 import com.citygis.web.domain.*;
 import com.citygis.web.mapper.TabSystemMsgMapper;
+import com.citygis.web.service.ITabFileInfoService;
 import com.citygis.web.service.ITabSystemMsgService;
-import com.citygis.web.utils.ExcelExportUtil;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.stereotype.Service;
 
 import javax.annotation.Resource;
 import java.time.DayOfWeek;
 import java.time.LocalDate;
+import java.util.Date;
 import java.util.List;
+import java.util.Map;
 
 /**
  * <p>
@@ -30,9 +35,15 @@ public class TabSystemMsgServiceImpl extends ServiceImpl<TabSystemMsgMapper, Tab
     @Value("${weekReport.filepath}")
     private String weekReportFilePath;
 
+    @Value("${weekReport.templateFilePath}")
+    private String templateFilePath;
+
     @Resource
     TabSystemMsgMapper tabSystemMsgMapper;
 
+    @Resource
+    ITabFileInfoService tabFileInfoService;
+
     @Override
     public List<TabSystemMsg> getStatistics(TabSystemMsg tabSystemMsg) {
         List<TabSystemMsg> list = tabSystemMsgMapper.getTabSystemMsgList(tabSystemMsg);
@@ -100,10 +111,12 @@ public class TabSystemMsgServiceImpl extends ServiceImpl<TabSystemMsgMapper, Tab
 
         List<TabAccessTableMsg> tabAccessTableMsg = tabSystemMsgMapper.getTabAccessTableMsgList();
 
+        Map<String, Object> map = MapUtils.newHashMap();
+
         list.forEach(tabSystemMsgReport -> {
             tabAccessTableMsg.forEach(tabAccessTableMsg1 -> {
                 if (tabAccessTableMsg1.getTableName().equals(tabSystemMsgReport.getTableEnglishName())) {
-                    tabSystemMsgReport.setCount(tabAccessTableMsg1.getCount());
+                    map.put(tabAccessTableMsg1.getTableName(), tabAccessTableMsg1.getCount());
                 }
             });
         });
@@ -118,9 +131,23 @@ public class TabSystemMsgServiceImpl extends ServiceImpl<TabSystemMsgMapper, Tab
         LocalDate sunday = today.with(DayOfWeek.SUNDAY);
 
         // 设置文件保存路径
-        String filePath = weekReportFilePath + monday + "-" + sunday + ".xlsx";
+        String filePath = weekReportFilePath + "周报" + monday + "-" + sunday + ".xlsx";
+
+        EasyExcel.write(filePath).withTemplate(templateFilePath).sheet().doFill(map);
+
+        saveTabFileInfo(filePath);
+    }
 
-        // 调用导出工具类生成 Excel 文件
-        ExcelExportUtil.exportExcel(list, filePath);
+    private void saveTabFileInfo(String filePath) {
+        //保存到主库，手动切换数据源
+        DynamicDataSourceContextHolder.setDataSourceType(DataSourceType.MASTER.name());
+        TabFileInfo tabFileInfo = new TabFileInfo();
+        tabFileInfo.setFileName(filePath.substring(filePath.lastIndexOf("\\") + 1));
+        tabFileInfo.setFilePath(filePath);
+        tabFileInfo.setCreateTime(new Date());
+        tabFileInfo.setCreateBy("system");
+        tabFileInfo.setSystemType("疾控周报");
+        tabFileInfoService.save(tabFileInfo);
+        DynamicDataSourceContextHolder.clearDataSourceType();
     }
 }

liutongyi-admin/src/main/resources/application.yml

@@ -150,4 +150,5 @@ jmx:
   jmxServiceUrl: service:jmx:rmi:///jndi/rmi://127.0.0.1:3010/jmxrmi
 
 weekReport:
-  filepath: D:/path_to_your_file/周报
+  templateFilePath: E:\Project\2024\2024 liutongyi\2024 liutongyi\liutongyi-admin\src\main\resources\templates\模版.xlsx
+  filepath: C:\Users\Administrator\Desktop\疾控周报\