From 826cfd6d1673337b1872632a84443537404917d4 Mon Sep 17 00:00:00 2001
From: value <z1292839451@163.com>
Date: 星期二, 21 五月 2024 09:21:11 +0800
Subject: [PATCH] 标准增家筛选

---
 performance-server/src/main/java/com/yuanchu/mom/controller/PerformanceShiftController.java |   55 +++++++++++++++++++++++++++++++++++++------------------
 1 files changed, 37 insertions(+), 18 deletions(-)

diff --git a/performance-server/src/main/java/com/yuanchu/mom/controller/PerformanceShiftController.java b/performance-server/src/main/java/com/yuanchu/mom/controller/PerformanceShiftController.java
index 45a48c6..0831c44 100644
--- a/performance-server/src/main/java/com/yuanchu/mom/controller/PerformanceShiftController.java
+++ b/performance-server/src/main/java/com/yuanchu/mom/controller/PerformanceShiftController.java
@@ -4,10 +4,12 @@
 import com.alibaba.excel.write.style.HorizontalCellStyleStrategy;
 import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.yuanchu.mom.annotation.ValueClassify;
 import com.yuanchu.mom.dto.PerformanceShiftAddDto;
 import com.yuanchu.mom.pojo.PerformanceShift;
 import com.yuanchu.mom.service.PerformanceShiftService;
-import com.yuanchu.mom.utils.StyleUtils;
+import com.yuanchu.mom.utils.StyleMonthUtils;
+import com.yuanchu.mom.utils.StyleYearUtils;
 import com.yuanchu.mom.vo.Result;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
@@ -37,47 +39,64 @@
     @Autowired
     private PerformanceShiftService performanceShiftService;
 
-    @ApiOperation(value = "缁╂晥绠＄悊-鐝-鎺掔彮")
+    @ValueClassify("鐝")
+    @ApiOperation(value = "鎺掔彮")
     @PostMapping("add")
     public Result<?> performanceShiftAdd(@RequestBody PerformanceShiftAddDto performanceShiftAddDto) {
         performanceShiftService.performanceShiftAdd(performanceShiftAddDto);
         return Result.success();
     }
 
-    @ApiOperation(value = "缁╂晥绠＄悊-鐝-鏈堜唤鍒嗛〉鏌ヨ")
+    @ValueClassify("鐝")
+    @ApiOperation(value = "鏈堜唤鍒嗛〉鏌ヨ")
     @PostMapping("page")
     public Result<?> performanceShiftPage(Integer size, Integer current, String time, String userName, String laboratory) {
         return Result.success(performanceShiftService.performanceShiftPage(new Page<>(current, size), time, userName, laboratory));
     }
 
-    @ApiOperation(value = "缁╂晥绠＄悊-鐝-骞翠唤鍒嗛〉鏌ヨ")
+    @ValueClassify("鐝")
+    @ApiOperation(value = "骞翠唤鍒嗛〉鏌ヨ")
     @PostMapping("pageYear")
     public Result<?> performanceShiftPageYear(Integer size, Integer current, String time, String userName, String laboratory) {
         return Result.success(performanceShiftService.performanceShiftPageYear(new Page<>(current, size), time, userName, laboratory));
     }
 
-    @ApiOperation(value = "缁╂晥绠＄悊-鐝-鐝鐘舵�佷慨鏀�")
+    @ValueClassify("鐝")
+    @ApiOperation(value = "鐝鐘舵�佷慨鏀�")
     @PutMapping("update")
     public Result<?> performanceShiftUpdate(@RequestBody PerformanceShift performanceShift) {
         performanceShiftService.performanceShiftUpdate(performanceShift);
         return Result.success();
     }
 
-    @ApiOperation(value = "缁╂晥绠＄悊-鐝-瀵煎嚭")
-    @GetMapping("update")
-    public void exportToExcel(@NotNull(message = "鏃堕棿涓嶈兘涓虹┖锛�") String time, String userName, String laboratory, HttpServletResponse response) throws Exception {
-        Map<Object, Object> data = performanceShiftService.exportToExcel(time, userName, laboratory);
-        // 璁剧疆鍗曞厓鏍兼牱寮�
-        HorizontalCellStyleStrategy horizontalCellStyleStrategy = new HorizontalCellStyleStrategy(StyleUtils.getHeadStyle(), StyleUtils.getContentStyle());
-        // 淇濆瓨鍒扮涓�涓猻heet涓�
+    @ValueClassify("鐝")
+    @ApiOperation(value = "瀵煎嚭")
+    @GetMapping("export")
+    public void exportToExcel(@NotNull(message = "鏃堕棿涓嶈兘涓虹┖锛�") String time, String userName, String laboratory, Boolean isMonth, HttpServletResponse response) throws Exception {
+        Map<Object, Object> data;
         response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
         response.setHeader("requestType","excel");
         response.setHeader("Access-Control-Expose-Headers", "requestType");
-        EasyExcel.write(response.getOutputStream())
-                .head((List<List<String>>) data.get("header"))
-                .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()) // 鑷�傚簲鍒楀
-                .registerWriteHandler(horizontalCellStyleStrategy)
-                .sheet("妯℃澘")
-                .doWrite((Collection<?>) data.get("data"));
+        if (!isMonth) {
+            data = performanceShiftService.exportToYearExcel(time, userName, laboratory);
+            // 璁剧疆鍗曞厓鏍兼牱寮�
+            HorizontalCellStyleStrategy horizontalCellStyleStrategy = new HorizontalCellStyleStrategy(StyleYearUtils.getHeadStyle(), StyleYearUtils.getContentStyle());
+            // 淇濆瓨鍒扮涓�涓猻heet涓�
+            EasyExcel.write(response.getOutputStream())
+                    .head((List<List<String>>) data.get("header"))
+                    .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()) // 鑷�傚簲鍒楀
+                    .registerWriteHandler(horizontalCellStyleStrategy)
+                    .sheet("骞村害")
+                    .doWrite((Collection<?>) data.get("data"));
+        } else {
+            data = performanceShiftService.exportToMonthExcel(time, userName, laboratory);
+            // 璁剧疆鍗曞厓鏍兼牱寮�
+            HorizontalCellStyleStrategy horizontalCellStyleStrategy = new HorizontalCellStyleStrategy(StyleMonthUtils.getHeadStyle(), StyleMonthUtils.getContentStyle());
+            EasyExcel.write(response.getOutputStream())
+                    .head((List<List<String>>) data.get("header"))
+                    .registerWriteHandler(horizontalCellStyleStrategy)
+                    .sheet("鏈堝害")
+                    .doWrite((Collection<?>) data.get("data"));
+        }
     }
 }

--
Gitblit v1.9.3