From 21887ddd12bdba4e2b00456c94a3cb8e1568c000 Mon Sep 17 00:00:00 2001
From: ZengTao <2773468879@qq.com>
Date: 星期一, 12 五月 2025 18:21:52 +0800
Subject: [PATCH] Merge branch 'master' of http://10.153.19.25:10105/r/YiWuProject

---
 hangzhoumesParent/common/servicebase/src/main/java/com/mes/largenscreen/service/impl/LargenScreenServiceImpl.java |   44 ++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 44 insertions(+), 0 deletions(-)

diff --git a/hangzhoumesParent/common/servicebase/src/main/java/com/mes/largenscreen/service/impl/LargenScreenServiceImpl.java b/hangzhoumesParent/common/servicebase/src/main/java/com/mes/largenscreen/service/impl/LargenScreenServiceImpl.java
index 1e0505a..27572cc 100644
--- a/hangzhoumesParent/common/servicebase/src/main/java/com/mes/largenscreen/service/impl/LargenScreenServiceImpl.java
+++ b/hangzhoumesParent/common/servicebase/src/main/java/com/mes/largenscreen/service/impl/LargenScreenServiceImpl.java
@@ -1,15 +1,22 @@
 package com.mes.largenscreen.service.impl;
 
+import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.support.ExcelTypeEnum;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.mes.largenscreen.entity.DailyProductionData;
 import com.mes.largenscreen.entity.DailyProductionVO;
 import com.mes.largenscreen.entity.DateRequest;
 import com.mes.largenscreen.mapper.LargenScreenMapper;
 import com.mes.largenscreen.service.LargenScreenService;
 import com.mes.tools.PageUtil;
+import com.mes.utils.excel.ExcelMergeStrategy;
+import com.mes.utils.excel.ExcelUtil;
 import org.springframework.stereotype.Service;
 import org.springframework.util.StringUtils;
 
 import javax.annotation.Resource;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
 import java.text.SimpleDateFormat;
 import java.util.Calendar;
 import java.util.Date;
@@ -25,6 +32,8 @@
 
     @Resource
     private LargenScreenMapper largenScreenMapper;
+    @Resource
+    private HttpServletResponse response;
 
     @Override
     public Page<DailyProductionVO> queryDailyProduction(DateRequest query) {
@@ -50,5 +59,40 @@
         return list;
     }
 
+    @Override
+    public void exportDailyProduction(DateRequest query) {
+        try {
+            if (StringUtils.isEmpty(query.getBeginDate())) {
+                SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
+                Calendar calendar = Calendar.getInstance();
+                calendar.add(Calendar.DAY_OF_MONTH, -6);
+                query.setBeginDate(sdf.format(calendar.getTime()));
+                query.setEndDate(sdf.format(new Date()));
+            }
+            ExcelUtil.setExcelResponseProp(response, "渚涘簲鍟嗗垪琛�-鍚堝苟");
+            List<DailyProductionData> providerList = largenScreenMapper.exportDailyProduction(query.getBeginDate(), query.getEndDate());
+            EasyExcel.write(response.getOutputStream())
+                    .head(DailyProductionData.class)
+                    .registerWriteHandler(new ExcelMergeStrategy(DailyProductionData.class))
+                    .excelType(ExcelTypeEnum.XLSX).sheet().doWrite(providerList);
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    @Override
+    public List<DailyProductionData> queryProduction(DateRequest query) {
+        if (StringUtils.isEmpty(query.getBeginDate())) {
+            SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
+            Calendar calendar = Calendar.getInstance();
+            calendar.add(Calendar.DAY_OF_MONTH, -6);
+            query.setBeginDate(sdf.format(calendar.getTime()));
+            query.setEndDate(sdf.format(new Date()));
+        }
+        return largenScreenMapper.exportDailyProduction(query.getBeginDate(), query.getEndDate());
+
+
+    }
+
 
 }

--
Gitblit v1.8.0