From 767122693cdc41e65bf84ae91d9166600ff0559e Mon Sep 17 00:00:00 2001 From: Crunchy <3114200645@qq.com> Date: 星期一, 13 五月 2024 17:11:08 +0800 Subject: [PATCH] 2024-5-10 绩效管理-班次-月度文件导出 --- performance-server/src/main/java/com/yuanchu/mom/controller/PerformanceShiftController.java | 39 ++++++++++++++++++++++++++------------- 1 files changed, 26 insertions(+), 13 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..965ce6a 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 @@ -7,7 +7,8 @@ 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; @@ -64,20 +65,32 @@ } @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涓� + @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