From 487bbcd45afcc288c27784962d3b40d5015bcd12 Mon Sep 17 00:00:00 2001
From: maven <2163098428@qq.com>
Date: 星期五, 15 八月 2025 15:33:07 +0800
Subject: [PATCH] yys  修改导入导出

---
 src/main/java/com/ruoyi/lavorissue/service/impl/LavorIssueServiceImpl.java |  178 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 176 insertions(+), 2 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 ff492b2..b85975f 100644
--- a/src/main/java/com/ruoyi/lavorissue/service/impl/LavorIssueServiceImpl.java
+++ b/src/main/java/com/ruoyi/lavorissue/service/impl/LavorIssueServiceImpl.java
@@ -1,14 +1,27 @@
 package com.ruoyi.lavorissue.service.impl;
 
+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.excel.ExcelUtils;
+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.SysDictData;
+import com.ruoyi.project.system.domain.SysDictType;
+import com.ruoyi.project.system.mapper.SysDictDataMapper;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.util.CollectionUtils;
+
+import javax.servlet.http.HttpServletResponse;
+import java.text.SimpleDateFormat;
+import java.util.*;
+import java.util.stream.Collectors;
 
 /**
  * @author :yys
@@ -23,7 +36,168 @@
 
     @Override
     public IPage<LaborIssue> listPage(Page page, LaborIssue laborIssue) {
-        IPage<LaborIssue> listPage = lavorIssueMapper.listPage(page, laborIssue);
-        return listPage;
+        StartAndEndDateDto startAndEndDateDto = getStartAndEndDateDto(laborIssue.getSeason(), laborIssue.getIssueDate());
+        laborIssue.setStartDate(startAndEndDateDto.getStartDate());
+        laborIssue.setEndDate(startAndEndDateDto.getEndDate());
+        IPage<LaborIssue> laborIssueIPage = lavorIssueMapper.listPage(page, laborIssue);
+        return laborIssueIPage;
+    }
+
+    public StartAndEndDateDto getStartAndEndDateDto(Integer season,Date payDate){
+        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
+        StartAndEndDateDto startAndEndDateDto = new StartAndEndDateDto();
+        if(season != null){
+            Calendar calendar = Calendar.getInstance();
+            int currentYear = calendar.get(Calendar.YEAR);
+            switch (season){
+                case 1:
+                    startAndEndDateDto.setStartDate(currentYear + "-01-01");
+                    startAndEndDateDto.setEndDate(currentYear + "-03-31");
+                    break;
+                case 2:
+                    startAndEndDateDto.setStartDate(currentYear + "-04-01");
+                    startAndEndDateDto.setEndDate(currentYear + "-06-30");
+                    break;
+                case 3:
+                    startAndEndDateDto.setStartDate(currentYear + "-07-01");
+                    startAndEndDateDto.setEndDate(currentYear + "-09-30");
+                    break;
+                case 4:
+                    startAndEndDateDto.setStartDate(currentYear + "-10-01");
+                    startAndEndDateDto.setEndDate(currentYear + "-12-31");
+                    break;
+            }
+        }
+        if(payDate != null){
+            Date lastDayOfMonth = getLastDayOfMonth(payDate);
+            startAndEndDateDto.setStartDate(sdf.format(payDate));
+            startAndEndDateDto.setEndDate(sdf.format(lastDayOfMonth));
+        }
+        return startAndEndDateDto;
+    }
+
+    @Override
+    public StatisticsLaborIssue statistics(StatisticsLaborIssue req) throws Exception{
+        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
+        StatisticsLaborIssue statisticsLaborIssue = new StatisticsLaborIssue();
+        StartAndEndDateDto startAndEndDateDto = getStartAndEndDateDto(req.getSeason(), req.getIssueDate());
+        LambdaQueryWrapper<LaborIssue> laborIssueLambdaQueryWrapper = new LambdaQueryWrapper<>();
+        if(req.getSeason() != null || req.getIssueDate() != null){
+            laborIssueLambdaQueryWrapper.ge(LaborIssue::getIssueDate, startAndEndDateDto.getStartDate())
+                    .le(LaborIssue::getIssueDate, startAndEndDateDto.getEndDate());
+        }
+        List<LaborIssue> laborIssues = lavorIssueMapper.selectList(laborIssueLambdaQueryWrapper);
+        if(!CollectionUtils.isEmpty(laborIssues)){
+            Long sum = laborIssues.stream()
+                    .filter(laborIssue -> laborIssue.getAdoptedDate() != null)
+                    .mapToLong(LaborIssue::getNum)
+                    .sum();
+            statisticsLaborIssue.setYlqNum(sum);
+            statisticsLaborIssue.setWlqNum(laborIssues.stream()
+                    .filter(laborIssue -> laborIssue.getAdoptedDate() == null)
+                    .mapToLong(LaborIssue::getNum)
+                    .sum());
+            Date currentDate = new Date();
+            Date parse = sdf.parse(sdf.format(currentDate));
+            statisticsLaborIssue.setCsylqNum(laborIssues.stream()
+                    .filter(laborIssue -> laborIssue.getIssueDate() != null
+                            && laborIssue.getAdoptedDate() != null
+                            && (laborIssue.getIssueDate().before(laborIssue.getAdoptedDate())))
+                    .mapToLong(LaborIssue::getNum)
+                    .sum());
+            statisticsLaborIssue.setCswlqNum(laborIssues.stream()
+                    .filter(laborIssue -> laborIssue.getIssueDate() != null
+                            && laborIssue.getIssueDate().before(parse)
+                            && laborIssue.getAdoptedDate() == null)
+                    .mapToLong(LaborIssue::getNum)
+                    .sum());
+        }
+        return statisticsLaborIssue;
+    }
+
+    @Autowired
+    private SysDictDataMapper sysDictDataMapper;
+
+    @Override
+    public void exportCopy(HttpServletResponse response) {
+        List<SysDictData> sys_lavor_issue = sysDictDataMapper.selectDictDataByType("sys_lavor_issue");
+        if(CollectionUtils.isEmpty(sys_lavor_issue)){
+            throw new RuntimeException("瀛楀吀鏁版嵁涓虹┖");
+        }
+        List<Object> head = new ArrayList<>();
+        head.add("閮ㄩ棬");
+        head.add(ExcelUtils.COLUMN_MERGE);
+        head.add("浼佷笟绠$悊绉�");
+        head.add(ExcelUtils.COLUMN_MERGE);
+        head.add("");
+        for (int i = 0; i < sys_lavor_issue.size(); i++) {
+            head.add(ExcelUtils.COLUMN_MERGE);
+        }
+
+        List<Object> head1 = new ArrayList<>();
+        head1.add("寮�濮嬪勾/鏈�");
+        head1.add(ExcelUtils.COLUMN_MERGE);
+        head1.add("缁撴潫骞�/鏈�");
+        head1.add(ExcelUtils.COLUMN_MERGE);
+        head1.add("");
+        for (int i = 0; i < sys_lavor_issue.size(); i++) {
+            head1.add(ExcelUtils.COLUMN_MERGE);
+        }
+
+        List<Object> head2 = new ArrayList<>();
+        head2.add("2025");
+        head2.add("1");
+        head2.add("2025");
+        head2.add("3");
+        head2.add("浼佷笟绠$悊绉�2025骞�1鏈�-2025骞�3鏈堝姵淇濆彂鏀捐鍒掕〃");
+        for (int i = 0; i < sys_lavor_issue.size(); i++) {
+            head2.add(ExcelUtils.COLUMN_MERGE);
+        }
+
+        List<Object> head3 = new ArrayList<>();
+        head3.add("閮ㄩ棬鍚嶇О");
+        head3.add(ExcelUtils.COLUMN_MERGE);
+        head3.add("濮撳悕");
+        head3.add("宸ュ彿");
+        for (SysDictData sysDictData : sys_lavor_issue) {
+            head3.add(sysDictData.getDictLabel());
+        }
+        head3.add("绛惧悕");
+
+        // 濉厖鏁版嵁
+        List<Object> head4 = new ArrayList<>();
+        List<LaborIssue> laborIssues = lavorIssueMapper.selectList(null);
+        if(CollectionUtils.isEmpty(laborIssues)){
+            throw new RuntimeException("鏁版嵁涓虹┖");
+        }
+
+        Map<String, List<LaborIssue>> collect = laborIssues.stream().collect(Collectors.groupingBy(LaborIssue::getStaffNo));
+        collect.forEach((k,v)->{
+            head4.add(v.get(0).getDeptName());
+            head4.add(ExcelUtils.COLUMN_MERGE);
+            head4.add(v.get(0).getStaffName());
+            head4.add(v.get(0).getStaffNo());
+            // 鍒ゆ柇瀛楀吀鏁版嵁
+            for (SysDictData sysDictData : sys_lavor_issue) {
+                long sum = v.stream().filter(laborIssue -> laborIssue.getDictId().equals(sysDictData.getDictValue())).mapToLong(LaborIssue::getNum).sum();
+                head4.add(sum);
+            }
+
+        });
+
+        List<List<Object>> sheetDataList = new ArrayList<>();
+        sheetDataList.add(head);
+        sheetDataList.add(head1);
+        sheetDataList.add(head2);
+        sheetDataList.add(head3);
+        sheetDataList.add(head4);
+        ExcelUtils.export(response, "鍔充繚鍙戞斁璁″垝琛�", sheetDataList);
+    }
+
+    public Date getLastDayOfMonth(Date date) {
+        Calendar calendar = Calendar.getInstance();
+        calendar.setTime(date); // 璁剧疆浼犲叆鐨凞ate
+        calendar.set(Calendar.DAY_OF_MONTH, calendar.getActualMaximum(Calendar.DAY_OF_MONTH)); // 璁剧疆涓哄綋鏈堟渶鍚庝竴澶�
+        return calendar.getTime(); // 杩斿洖Date瀵硅薄
     }
 }

--
Gitblit v1.9.3