From 0ec581bd195ea9c06c38ef4f91ac7497a87b8ca3 Mon Sep 17 00:00:00 2001
From: chenhj <1263187585@qq.com>
Date: 星期三, 22 四月 2026 17:49:54 +0800
Subject: [PATCH] Merge branch 'dev_New_pro' of http://114.132.189.42:9002/r/product-inventory-management-after into dev_New_pro

---
 src/main/java/com/ruoyi/staff/controller/PersonalShiftController.java |   49 +++++++++++++++++++++++++++++++------------------
 1 files changed, 31 insertions(+), 18 deletions(-)

diff --git a/src/main/java/com/ruoyi/staff/controller/PersonalShiftController.java b/src/main/java/com/ruoyi/staff/controller/PersonalShiftController.java
index 85ff013..676c672 100644
--- a/src/main/java/com/ruoyi/staff/controller/PersonalShiftController.java
+++ b/src/main/java/com/ruoyi/staff/controller/PersonalShiftController.java
@@ -14,11 +14,13 @@
 import com.ruoyi.staff.utils.StyleYearUtils;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
+import lombok.AllArgsConstructor;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.*;
 
-import javax.servlet.http.HttpServletResponse;
-import javax.validation.constraints.NotNull;
+import jakarta.servlet.ServletOutputStream;
+import jakarta.servlet.http.HttpServletResponse;
+import jakarta.validation.constraints.NotNull;
 import java.util.Collection;
 import java.util.List;
 import java.util.Map;
@@ -34,9 +36,9 @@
 @RestController
 @RequestMapping("/personalShift")
 @Api(tags = "浜哄憳鎺掔彮")
+@AllArgsConstructor
 public class PersonalShiftController {
 
-    @Autowired
     private PersonalShiftService personalShiftService;
 
     @ApiOperation("浜哄憳鎺掔彮")
@@ -67,31 +69,42 @@
     @ApiOperation(value = "瀵煎嚭")
     @GetMapping("export")
     public void exportToExcel(@NotNull(message = "鏃堕棿涓嶈兘涓虹┖锛�") String time, String userName, Integer sysDeptId, Boolean isMonth, HttpServletResponse response) throws Exception {
-        Map<Object, Object> data;
+        ServletOutputStream out = response.getOutputStream();
         response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
-        response.setHeader("requestType","excel");
-        response.setHeader("Access-Control-Expose-Headers", "requestType");
+        response.setCharacterEncoding("utf-8");
+        response.setHeader("Content-Disposition", "attachment;filename*=UTF-8''" + time + ".xlsx");
+
         if (!isMonth) {
-            data = personalShiftService.exportToYearExcel(time, userName, sysDeptId);
-            // 璁剧疆鍗曞厓鏍兼牱寮�
-            HorizontalCellStyleStrategy horizontalCellStyleStrategy = new HorizontalCellStyleStrategy(StyleYearUtils.getHeadStyle(), StyleYearUtils.getContentStyle());
-            // 淇濆瓨鍒扮涓�涓猻heet涓�
-            EasyExcel.write(response.getOutputStream())
+            Map<Object, Object> data = personalShiftService.exportToYearExcel(time, userName, sysDeptId);
+
+            HorizontalCellStyleStrategy style = new HorizontalCellStyleStrategy(
+                    StyleYearUtils.getHeadStyle(),
+                    StyleYearUtils.getContentStyle()
+            );
+
+            EasyExcel.write(out)
                     .head((List<List<String>>) data.get("header"))
-                    .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()) // 鑷�傚簲鍒楀
-                    .registerWriteHandler(horizontalCellStyleStrategy)
+                    .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy())
+                    .registerWriteHandler(style)
                     .sheet("骞村害")
                     .doWrite((Collection<?>) data.get("data"));
+
         } else {
-            data = personalShiftService.exportToMonthExcel(time, userName, sysDeptId);
-            // 璁剧疆鍗曞厓鏍兼牱寮�
-            HorizontalCellStyleStrategy horizontalCellStyleStrategy = new HorizontalCellStyleStrategy(StyleMonthUtils.getHeadStyle(), StyleMonthUtils.getContentStyle());
-            EasyExcel.write(response.getOutputStream())
+            Map<Object, Object> data = personalShiftService.exportToMonthExcel(time, userName, sysDeptId);
+
+            HorizontalCellStyleStrategy style = new HorizontalCellStyleStrategy(
+                    StyleMonthUtils.getHeadStyle(),
+                    StyleMonthUtils.getContentStyle()
+            );
+
+            EasyExcel.write(out)
                     .head((List<List<String>>) data.get("header"))
-                    .registerWriteHandler(horizontalCellStyleStrategy)
+                    .registerWriteHandler(style)
                     .sheet("鏈堝害")
                     .doWrite((Collection<?>) data.get("data"));
         }
+
+        out.flush();
     }
 
 

--
Gitblit v1.9.3