maven
4 天以前 e36e342b4a7bc01de70e47a7926a2e6372dd0970
src/main/java/com/ruoyi/lavorissue/service/impl/LavorIssueServiceImpl.java
@@ -9,6 +9,7 @@
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.StringUtils;
import com.ruoyi.common.utils.excel.ConfigurableMergeStrategy;
import com.ruoyi.common.utils.excel.CustomCellStyleHandler;
import com.ruoyi.lavorissue.dto.StatisticsLaborIssue;
@@ -51,17 +52,54 @@
        StartAndEndDateDto startAndEndDateDto = getStartAndEndDateDto(laborIssue.getSeason(), laborIssue.getIssueDate());
        laborIssue.setStartDate(startAndEndDateDto.getStartDate());
        laborIssue.setEndDate(startAndEndDateDto.getEndDate());
        IPage<LaborIssue> laborIssueIPage = lavorIssueMapper.listPage(page, laborIssue);
        return laborIssueIPage;
        return lavorIssueMapper.listPage(page, laborIssue);
    }
    @Override
    public List<Map<String, Object>> statisticsList(LaborIssue laborIssue){
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
        StartAndEndDateDto startAndEndDateDto = getStartAndEndDateDto(laborIssue.getSeason(), laborIssue.getIssueDate());
        laborIssue.setStartDate(startAndEndDateDto.getStartDate());
        laborIssue.setEndDate(startAndEndDateDto.getEndDate());
        List<LaborIssue> laborIssueIPage = lavorIssueMapper.list(laborIssue);
        // 使用可变的ArrayList替换默认的不可变列表
        List<Map<String, Object>> records = new ArrayList<>();
        Map<String, List<LaborIssue>> collect = laborIssueIPage.stream().collect(Collectors.groupingBy(LaborIssue::getStaffNo));
        collect.forEach((k, v) -> {
            HashMap<String, Object> hashMap = new HashMap<>();
            LaborIssue laborIssue1 = v.get(0);
            hashMap.put("id", v.stream().map(LaborIssue::getId));
            hashMap.put("deptName", laborIssue1.getDeptName());
            hashMap.put("orderNo", laborIssue1.getOrderNo());
            hashMap.put("staffId", laborIssue1.getStaffId());
            hashMap.put("staffName", laborIssue1.getStaffName());
            hashMap.put("staffNo", laborIssue1.getStaffNo());
            hashMap.put("dictType", laborIssue1.getDictType());
            hashMap.put("dictTypeName", laborIssue1.getDictTypeName());
            hashMap.put("factoryDate", laborIssue1.getFactoryDate() == null ? "" : sdf.format(laborIssue1.getFactoryDate()));
            hashMap.put("issueDate", laborIssue1.getIssueDate() == null ? "" : sdf.format(laborIssue1.getIssueDate()));
            hashMap.put("adoptedDate", laborIssue1.getAdoptedDate() == null ? "" : sdf.format(laborIssue1.getAdoptedDate()));
            hashMap.put("tenantId", laborIssue1.getTenantId());
            Map<String, List<LaborIssue>> collect1 = v.stream().collect(Collectors.groupingBy(LaborIssue::getDictId));
            collect1.forEach((k1, v1) -> {
                hashMap.put(k1, v1.stream()
                        .mapToLong(LaborIssue::getNum)
                        .sum());
            });
            records.add(hashMap);
        });
        return records;
    }
    public StartAndEndDateDto getStartAndEndDateDto(Integer season,Date payDate){
        SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
        StartAndEndDateDto startAndEndDateDto = new StartAndEndDateDto();
        Calendar calendar = Calendar.getInstance();
        int currentYear = calendar.get(Calendar.YEAR);
        startAndEndDateDto.setYear(currentYear);
        if(season != null){
            Calendar calendar = Calendar.getInstance();
            int currentYear = calendar.get(Calendar.YEAR);
            startAndEndDateDto.setYear(currentYear);
            switch (season){
                case 1:
                    startAndEndDateDto.setStartDate(currentYear + "-01-01");
@@ -91,8 +129,11 @@
        }
        if(payDate != null){
            Date lastDayOfMonth = getLastDayOfMonth(payDate);
            startAndEndDateDto.setStartDate(sdf.format(payDate));
            Date firstDayOfMonth = getFirstDayOfMonth(payDate);
            startAndEndDateDto.setStartDate(sdf.format(firstDayOfMonth));
            startAndEndDateDto.setEndDate(sdf.format(lastDayOfMonth));
            startAndEndDateDto.setStartMonth(payDate.getMonth() + 1);
            startAndEndDateDto.setEndMonth(payDate.getMonth() + 1);
        }
        return startAndEndDateDto;
    }
@@ -172,7 +213,7 @@
                List<List<String>> list = data.get(i);
                //获取sheet0对象
                WriteSheet mainSheet = EasyExcel.writerSheet(i, "外部装箱单" + i).build();
                WriteSheet mainSheet = EasyExcel.writerSheet(i, "劳保台账" + i).build();
                //向sheet0写入数据 传入空list这样只导出表头
                excelWriter.write(list, mainSheet);
            }
@@ -308,4 +349,17 @@
        calendar.set(Calendar.DAY_OF_MONTH, calendar.getActualMaximum(Calendar.DAY_OF_MONTH)); // 设置为当月最后一天
        return calendar.getTime(); // 返回Date对象
    }
    /**
     * 获取当月第一天
     * @param date
     * @return
     */
    public Date getFirstDayOfMonth(Date date) {
        Calendar calendar = Calendar.getInstance();
        calendar.setTime(date); // 设置传入的Date
        calendar.set(Calendar.DAY_OF_MONTH, calendar.getActualMinimum(Calendar.DAY_OF_MONTH)); // 设置为当月第一天
        return calendar.getTime(); // 返回Date对象
    }
}