From 948194840fc18c00617cbf3cf72f9e81bb0ee5b3 Mon Sep 17 00:00:00 2001
From: zouyu <2723363702@qq.com>
Date: 星期二, 14 四月 2026 16:50:00 +0800
Subject: [PATCH] 人员考勤调整：导出区分年度和月度
---
inspect-server/src/main/java/com/ruoyi/inspect/service/impl/StaffAttendanceTrackingRecordServiceImpl.java | 64 ++++++++++++++++++++++----------
1 files changed, 44 insertions(+), 20 deletions(-)
diff --git a/inspect-server/src/main/java/com/ruoyi/inspect/service/impl/StaffAttendanceTrackingRecordServiceImpl.java b/inspect-server/src/main/java/com/ruoyi/inspect/service/impl/StaffAttendanceTrackingRecordServiceImpl.java
index b46865f..6dbf68b 100644
--- a/inspect-server/src/main/java/com/ruoyi/inspect/service/impl/StaffAttendanceTrackingRecordServiceImpl.java
+++ b/inspect-server/src/main/java/com/ruoyi/inspect/service/impl/StaffAttendanceTrackingRecordServiceImpl.java
@@ -3,7 +3,9 @@
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.util.ObjectUtil;
import com.alibaba.excel.EasyExcel;
+import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.support.ExcelTypeEnum;
+import com.alibaba.excel.write.metadata.WriteSheet;
import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
@@ -12,6 +14,7 @@
import com.google.common.util.concurrent.AtomicDouble;
import com.ruoyi.common.enums.ClockInState;
import com.ruoyi.common.enums.EnterOrExitType;
+import com.ruoyi.common.enums.StaffAttendanceReportType;
import com.ruoyi.common.enums.SyncStatus;
import com.ruoyi.common.utils.api.icc.IccApiUtil;
import com.ruoyi.common.utils.api.icc.model.GetResultPageRequest;
@@ -81,6 +84,8 @@
private DateTimeFormatter HHmm = DateTimeFormatter.ofPattern("HH:mm");
private DateTimeFormatter yyyMMddHHmmss = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
+
+ private DateTimeFormatter yyyMMStr = DateTimeFormatter.ofPattern("yyyy骞碝M鏈坉d鏃�");
/** 璐ㄩ噺閮╥d */
private final static String deptIds = "6";
@@ -420,9 +425,9 @@
for (int i = 0; i < userIdKeys.size(); i++) {
List<PerformanceShiftMapDto> shiftMapDtos = groupByUserId.get(userIdKeys.get(i));
StaffAttendanceExcelData attendanceExcelData = new StaffAttendanceExcelData();
- attendanceExcelData.setExcelIndex(i+1);
List<String> shiftList = new ArrayList<>();
attendanceExcelData.setPersonName(shiftMapDtos.get(0).getUserName());
+ attendanceExcelData.setExcelIndex(i+1);
AtomicInteger holidayCount = new AtomicInteger(0);//浼戞伅澶╂暟
AtomicInteger personalLeaveCount = new AtomicInteger(0);//浜嬪亣澶╂暟
AtomicInteger annualLeaveCount = new AtomicInteger(0);//骞村亣澶╂暟
@@ -469,11 +474,16 @@
}
}
}
- if(StringUtils.isNotBlank(shiftMapDto.getAnnotationText())){
- annotationTextList.add(new StaffAttendanceAnnotationTextExcelData(i,j,shiftMapDto.getAnnotationText()));
+ //鏈堝害缁熻鎵嶆彃鍏ユ壒娉ㄦ暟鎹�
+ if(StringUtils.isNoneBlank(staffAttendanceDTO.getAttendanceReportType()) && StringUtils.equals(staffAttendanceDTO.getAttendanceReportType(), StaffAttendanceReportType.MONTH.name())){
+ if(StringUtils.isNotBlank(shiftMapDto.getAnnotationText())){
+ annotationTextList.add(new StaffAttendanceAnnotationTextExcelData(i,j,shiftMapDto.getAnnotationText()));
+ }
}
}
- attendanceExcelData.setShiftList(shiftList);
+ if(StringUtils.isNoneBlank(staffAttendanceDTO.getAttendanceReportType()) && StringUtils.equals(staffAttendanceDTO.getAttendanceReportType(), StaffAttendanceReportType.MONTH.name())){
+ attendanceExcelData.setShiftList(shiftList);
+ }
attendanceExcelData.setAttendanceDayCount(attendanceDayCount.get());
attendanceExcelData.setAttendanceWorkHourCount(attendanceWorkHourCount.get());
//鐝鑰冨嫟澶╂暟
@@ -493,13 +503,19 @@
response.setHeader("Cache-Control", "no-cache");
response.setHeader("Content-Disposition", "attachment;filename=" + fileName);
response.setHeader("Access-Control-Expose-Headers", "Content-Disposition");
- InputStream resourceAsStream = buildAttendanceTemplate(attendanceDateList);
- EasyExcel.write(response.getOutputStream())
- .withTemplate(resourceAsStream)
- .registerWriteHandler(new CommentWriteHandler(excelData, annotationTextList))
- .relativeHeadRowIndex(4)
- .sheet()
- .doFill(excelData);
+ InputStream resourceAsStream = buildAttendanceTemplate(attendanceDateList,staffAttendanceDTO.getAttendanceReportType());
+ try (ExcelWriter excelWriter = EasyExcel.write(response.getOutputStream()).withTemplate(resourceAsStream).registerWriteHandler(new CommentWriteHandler(excelData, annotationTextList)).relativeHeadRowIndex(4).build()) {
+ WriteSheet writeSheet = EasyExcel.writerSheet().build();
+ excelWriter.fill(excelData, writeSheet);
+ if(StringUtils.equals(staffAttendanceDTO.getAttendanceReportType(),StaffAttendanceReportType.YEAR.name())){
+ String startDate = staffAttendanceDTO.getStartDate().format(yyyMMStr);
+ String endDate = staffAttendanceDTO.getEndDate().format(yyyMMStr);
+ Map<String, String> dateMap = new HashMap<>();
+ dateMap.put("startDate",startDate);
+ dateMap.put("endDate",endDate);
+ excelWriter.fill(dateMap, writeSheet);
+ }
+ }
} catch (Exception e) {
throw new RuntimeException(e);
}
@@ -518,20 +534,28 @@
for (LocalDate currentDate = startDate; !currentDate.isAfter(endDate); currentDate = currentDate.plusDays(1)) {
attendanceDateList.add(currentDate);
}
- if (attendanceDateList.size() > 31) {
+ if (attendanceDateList.size() > 31 && StringUtils.equals(staffAttendanceDTO.getAttendanceReportType(),StaffAttendanceReportType.MONTH.name())) {
throw new IllegalArgumentException("瀵煎嚭鏃堕棿鑼冨洿涓嶈兘瓒呰繃31澶�");
}
return attendanceDateList;
}
- private InputStream buildAttendanceTemplate(List<LocalDate> attendanceDateList) throws IOException {
- try (InputStream templateStream = this.getClass().getResourceAsStream("/static/staff_attendance_template.xlsx");
- Workbook workbook = WorkbookFactory.create(templateStream);
- ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
- Sheet sheet = workbook.getSheetAt(0);
- fillAttendanceHeader(sheet, attendanceDateList);
- workbook.write(outputStream);
- return new ByteArrayInputStream(outputStream.toByteArray());
+ private InputStream buildAttendanceTemplate(List<LocalDate> attendanceDateList,String attendanceReportType) throws IOException {
+ String templateName = "/static/staff_attendance_month_template.xlsx";
+ if(StringUtils.equals(attendanceReportType,StaffAttendanceReportType.YEAR.name())){
+ templateName = "/static/staff_attendance_year_template.xlsx";
+ }
+ try (InputStream templateStream = this.getClass().getResourceAsStream(templateName)) {
+ assert templateStream != null;
+ try (Workbook workbook = WorkbookFactory.create(templateStream);
+ ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
+ Sheet sheet = workbook.getSheetAt(0);
+ if(StringUtils.equals(attendanceReportType,StaffAttendanceReportType.MONTH.name())){
+ fillAttendanceHeader(sheet, attendanceDateList);
+ }
+ workbook.write(outputStream);
+ return new ByteArrayInputStream(outputStream.toByteArray());
+ }
} catch (Exception e) {
throw new IOException("鏋勫缓鑰冨嫟瀵煎嚭妯℃澘澶辫触", e);
}
--
Gitblit v1.9.3