From 397bfc8424f2b72d6ff40818da09107fbbf28abe Mon Sep 17 00:00:00 2001 From: maven <2163098428@qq.com> Date: 星期五, 15 八月 2025 17:49:42 +0800 Subject: [PATCH] yys 修改导入导出 --- src/main/java/com/ruoyi/lavorissue/service/impl/LavorIssueServiceImpl.java | 184 +++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 181 insertions(+), 3 deletions(-) diff --git a/src/main/java/com/ruoyi/lavorissue/service/impl/LavorIssueServiceImpl.java b/src/main/java/com/ruoyi/lavorissue/service/impl/LavorIssueServiceImpl.java index b6a5e27..3d2c087 100644 --- a/src/main/java/com/ruoyi/lavorissue/service/impl/LavorIssueServiceImpl.java +++ b/src/main/java/com/ruoyi/lavorissue/service/impl/LavorIssueServiceImpl.java @@ -1,23 +1,37 @@ package com.ruoyi.lavorissue.service.impl; +import com.alibaba.excel.EasyExcel; +import com.alibaba.excel.ExcelWriter; +import com.alibaba.excel.write.metadata.WriteSheet; import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.ruoyi.approve.utils.StartAndEndDateDto; +import com.ruoyi.common.utils.SecurityUtils; +import com.ruoyi.common.utils.excel.ConfigurableMergeStrategy; +import com.ruoyi.common.utils.excel.CustomCellStyleHandler; import com.ruoyi.lavorissue.dto.StatisticsLaborIssue; import com.ruoyi.lavorissue.mapper.LavorIssueMapper; import com.ruoyi.lavorissue.pojo.LaborIssue; import com.ruoyi.lavorissue.service.LavorIssueService; +import com.ruoyi.project.system.domain.SysDept; +import com.ruoyi.project.system.domain.SysDictData; +import com.ruoyi.project.system.mapper.SysDeptMapper; +import com.ruoyi.project.system.mapper.SysDictDataMapper; import lombok.extern.slf4j.Slf4j; +import org.apache.poi.ss.util.CellRangeAddress; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import org.springframework.util.CollectionUtils; +import javax.servlet.http.HttpServletResponse; +import java.io.IOException; +import java.io.UnsupportedEncodingException; +import java.net.URLEncoder; import java.text.SimpleDateFormat; -import java.util.Calendar; -import java.util.Date; -import java.util.List; +import java.util.*; +import java.util.concurrent.atomic.AtomicInteger; import java.util.stream.Collectors; /** @@ -112,6 +126,170 @@ return statisticsLaborIssue; } + @Autowired + private SysDictDataMapper sysDictDataMapper; + + @Override + public void exportCopy(HttpServletResponse response) throws UnsupportedEncodingException { + List<SysDictData> sys_lavor_issue = sysDictDataMapper.selectDictDataByType("sys_lavor_issue"); + if(CollectionUtils.isEmpty(sys_lavor_issue)){ + throw new RuntimeException("瀛楀吀鏁版嵁涓虹┖"); + } + List<LaborIssue> laborIssues = lavorIssueMapper.list(new LaborIssue()); + if (CollectionUtils.isEmpty(laborIssues)) { + throw new RuntimeException("鏁版嵁涓虹┖"); + } + Map<String, List<LaborIssue>> laborIssueMap = laborIssues.stream() + .collect(Collectors.groupingBy(LaborIssue::getStaffNo)); + response.setContentType("application/vnd.ms-excel"); + response.setCharacterEncoding("UTF-8"); + // 杩欓噷URLEncoder.encode鍙互闃叉涓枃涔辩爜 褰撶劧鍜宔asyexcel娌℃湁鍏崇郴 + String fileName = URLEncoder.encode("澶栭儴瑁呯鍗�", "UTF-8"); + response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx"); + + try { + //鏂板缓ExcelWriter + ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream()) + .registerWriteHandler(new CustomCellStyleHandler()) + .registerWriteHandler(new ConfigurableMergeStrategy(exportExternalPackingListGetMerge(4,sys_lavor_issue.size(),laborIssueMap.size()))) + .build(); + + List<List<List<String>>> data = completeExternalPackingList(laborIssueMap, sys_lavor_issue); + + for (int i = 0; i < data.size(); i++) { + List<List<String>> list = data.get(i); + + //鑾峰彇sheet0瀵硅薄 + WriteSheet mainSheet = EasyExcel.writerSheet(i, "澶栭儴瑁呯鍗�" + i).build(); + //鍚憇heet0鍐欏叆鏁版嵁 浼犲叆绌簂ist杩欐牱鍙鍑鸿〃澶� + excelWriter.write(list, mainSheet); + } + + //鍏抽棴娴� + excelWriter.finish(); + } catch (IOException e) { + throw new RuntimeException("瀵煎嚭澶辫触"); + } + } + + @Autowired + private SysDeptMapper sysDeptMapper; + + +// 澶勭悊澶栭儴瑁呯鍗曟暟鎹� + private List<List<List<String>>> completeExternalPackingList(Map<String, List<LaborIssue>> listMap,List<SysDictData> sys_lavor_issue) { + List<List<List<String>>> data = new ArrayList<>(); + int num = sys_lavor_issue.size(); + + List<List<String>> item = new ArrayList<>(); + List<String> list = new ArrayList<>(); + list.add("閮ㄩ棬"); + list.add(""); + list.add("浼佷笟绠$悊绉�"); + list.add(""); + list.add("浼佷笟绠$悊绉�2025骞�1鏈�-2025骞�3鏈堝姵淇濆彂鏀捐鍒掕〃"); + for (int i = 1; i <= num; i++){ + list.add(""); + } + item.add(list); + + List<String> list1 = new ArrayList<>(); + list1.add("寮�濮嬪勾/鏈�"); + list1.add(""); + list1.add("缁撴潫骞�/鏈�"); + list1.add(""); + for (int i = 0; i <= num; i++){ + list1.add(""); + } + item.add(list1); + + List<String> list2 = new ArrayList<>(); + list2.add("2025"); + list2.add("1"); + list2.add("2025"); + list2.add("3"); + for (int i = 0; i <= num; i++){ + list.add(""); + } + item.add(list2); + + List<String> list3 = new ArrayList<>(); + list3.add("閮ㄩ棬鍚嶇О"); + list3.add(""); + list3.add("濮撳悕"); + list3.add("宸ュ彿"); + for (SysDictData sysDictData : sys_lavor_issue) { + list3.add(sysDictData.getDictLabel()); + } + list3.add("绛惧悕"); + item.add(list3); + + // 濉�� + SysDept sysDept = sysDeptMapper.selectDeptById(SecurityUtils.getLoginUser().getTenantId()); + List<String> sumList = new ArrayList<>(); + AtomicInteger i = new AtomicInteger(); + listMap.forEach((key, value) -> { + List<String> list4 = new ArrayList<>(); + list4.add(sysDept.getDeptName()); + list4.add(""); + list4.add(value.get(0).getStaffName()); + list4.add(value.get(0).getStaffNo()); + int j = 0; + for (SysDictData sysDictData : sys_lavor_issue) { + list4.add(value.stream().filter(laborIssue -> laborIssue.getDictId().equals(sysDictData.getDictValue())) + .mapToLong(LaborIssue::getNum) + .sum() + + ""); + if(i.get() == 0){ + sumList.add(value.stream().filter(laborIssue -> laborIssue.getDictId().equals(sysDictData.getDictValue())) + .mapToLong(LaborIssue::getNum) + .sum() + + ""); + }else { + sumList.set(j,(Long.parseLong(sumList.get(j)) + value.stream().filter(laborIssue -> laborIssue.getDictId().equals(sysDictData.getDictValue())) + .mapToLong(LaborIssue::getNum) + .sum()) + + ""); + } + j++; + } + i.getAndIncrement(); + item.add(list4); + }); + List<String> list5 = new ArrayList<>(); + list5.add("鍚堣"); + list5.add(""); + list5.add(""); + list5.add(""); + for (int h = 0; h < sumList.size(); h++){ + list5.add(sumList.get(h)); + } + item.add(list5); + data.add(item); + return data; + } + + /** + * 澶栭儴瑁呯鍗曞悎骞跺崟鍏冩牸 + * + * @return + */ + private List<CellRangeAddress> exportExternalPackingListGetMerge(Integer rowNum,Integer num,Integer listSize) { + List<CellRangeAddress> mergeRegions = new ArrayList<>(); + mergeRegions.add(new CellRangeAddress(0, 0, 0, 1)); // 鍚堝苟A1:Q2 + mergeRegions.add(new CellRangeAddress(0, 0, 2, 3)); // 鍚堝苟A4:Q4 + + mergeRegions.add(new CellRangeAddress(0, 2, rowNum, rowNum + num)); // 鍚堝苟A5:B6 + mergeRegions.add(new CellRangeAddress(1, 1, 0, 1)); // 鍚堝苟C5:E6 + mergeRegions.add(new CellRangeAddress(1, 1, 2, 3)); // 鍚堝苟F5:G6 + mergeRegions.add(new CellRangeAddress(3, 3, 0, 1)); // 鍚堝苟H5:I6 + for (int i = 1; i <= listSize; i++) { + mergeRegions.add(new CellRangeAddress(3 + i, 3 + i, 0, 1)); + } + mergeRegions.add(new CellRangeAddress(rowNum + listSize, rowNum + listSize, 0, 3)); + return mergeRegions; + } + public Date getLastDayOfMonth(Date date) { Calendar calendar = Calendar.getInstance(); calendar.setTime(date); // 璁剧疆浼犲叆鐨凞ate -- Gitblit v1.9.3