package cn.iocoder.yudao.module.hrm.service.attendance;
|
|
import cn.hutool.core.collection.CollUtil;
|
import cn.hutool.core.util.StrUtil;
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
import cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX;
|
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmAttendanceSummaryPageReqVO;
|
import cn.iocoder.yudao.module.hrm.dal.dataobject.attendance.HrmAttendanceRecordDO;
|
import cn.iocoder.yudao.module.hrm.dal.dataobject.attendance.HrmAttendanceSummaryDO;
|
import cn.iocoder.yudao.module.hrm.dal.dataobject.employee.HrmEmployeeDO;
|
import cn.iocoder.yudao.module.hrm.dal.mysql.attendance.HrmAttendanceRecordMapper;
|
import cn.iocoder.yudao.module.hrm.dal.mysql.attendance.HrmAttendanceSummaryMapper;
|
import cn.iocoder.yudao.module.hrm.enums.HrmAttendanceClockTypeEnum;
|
import cn.iocoder.yudao.module.hrm.service.employee.HrmEmployeeService;
|
import jakarta.annotation.Resource;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
import org.springframework.validation.annotation.Validated;
|
|
import java.math.BigDecimal;
|
import java.time.LocalDate;
|
import java.time.YearMonth;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Objects;
|
|
/**
|
* 考勤统计汇总 Service 实现类
|
*
|
* @author 超级管理员
|
*/
|
@Service
|
@Validated
|
@Slf4j
|
public class HrmAttendanceSummaryServiceImpl implements HrmAttendanceSummaryService {
|
|
@Resource
|
private HrmAttendanceSummaryMapper attendanceSummaryMapper;
|
@Resource
|
private HrmAttendanceRecordMapper attendanceRecordMapper;
|
@Resource
|
private HrmEmployeeService employeeService;
|
@Resource
|
private HrmAttendanceHolidayService attendanceHolidayService;
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public void generateMonthlySummary(Long userId, String yearMonth) {
|
// 1. 解析年月，获取该月起始日期
|
YearMonth ym = YearMonth.parse(yearMonth);
|
LocalDate startDate = ym.atDay(1);
|
LocalDate endDate = ym.atEndOfMonth();
|
|
// 2. 获取员工信息
|
HrmEmployeeDO employee = employeeService.getEmployee(userId);
|
if (employee == null) {
|
log.warn("[generateMonthlySummary] 员工不存在: {}", userId);
|
return;
|
}
|
|
// 3. 查询该月所有考勤记录
|
List<HrmAttendanceRecordDO> records = attendanceRecordMapper.selectListByUserIdAndDateBetween(userId, startDate, endDate);
|
|
// 4. 计算应出勤天数（工作日数量）
|
int workDays = calculateWorkDays(startDate, endDate);
|
|
// 5. 统计各项数据
|
int lateCount = 0;
|
int earlyCount = 0;
|
int absentCount = 0;
|
BigDecimal overtimeHours = BigDecimal.ZERO;
|
int actualDays = 0;
|
|
for (HrmAttendanceRecordDO record : records) {
|
// 统计迟到
|
if (HrmAttendanceClockTypeEnum.LATE.getType().equals(record.getClockInType())) {
|
lateCount++;
|
}
|
// 统计早退
|
if (HrmAttendanceClockTypeEnum.EARLY.getType().equals(record.getClockOutType())) {
|
earlyCount++;
|
}
|
// 统计缺卡
|
if (HrmAttendanceClockTypeEnum.MISSING.getType().equals(record.getClockInType())) {
|
absentCount++;
|
}
|
if (HrmAttendanceClockTypeEnum.MISSING.getType().equals(record.getClockOutType())) {
|
absentCount++;
|
}
|
// 累计加班时长
|
if (record.getOvertimeHours() != null) {
|
overtimeHours = overtimeHours.add(record.getOvertimeHours());
|
}
|
// 统计实际出勤天数（有正常打卡记录）
|
if (record.getClockInTime() != null || record.getClockOutTime() != null) {
|
actualDays++;
|
}
|
}
|
|
// 6. 保存或更新汇总记录
|
HrmAttendanceSummaryDO summary = attendanceSummaryMapper.selectByUserIdAndYearMonth(userId, yearMonth);
|
if (summary == null) {
|
summary = HrmAttendanceSummaryDO.builder()
|
.userId(userId)
|
.deptId(employee.getDeptId())
|
.yearMonth(yearMonth)
|
.workDays(workDays)
|
.actualDays(actualDays)
|
.lateCount(lateCount)
|
.earlyCount(earlyCount)
|
.absentCount(absentCount)
|
.overtimeHours(overtimeHours)
|
.leaveHours(BigDecimal.ZERO)
|
.build();
|
attendanceSummaryMapper.insert(summary);
|
} else {
|
summary.setWorkDays(workDays);
|
summary.setActualDays(actualDays);
|
summary.setLateCount(lateCount);
|
summary.setEarlyCount(earlyCount);
|
summary.setAbsentCount(absentCount);
|
summary.setOvertimeHours(overtimeHours);
|
attendanceSummaryMapper.updateById(summary);
|
}
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public void generateDeptMonthlySummary(Long deptId, String yearMonth) {
|
// 获取部门所有员工（未指定员工状态时查询全部）
|
List<HrmEmployeeDO> employees = employeeService.getEmployeeListByStatus(null);
|
if (CollUtil.isEmpty(employees)) {
|
return;
|
}
|
// 筛选部门员工
|
List<HrmEmployeeDO> deptEmployees = employees.stream()
|
.filter(e -> deptId.equals(e.getDeptId()))
|
.toList();
|
for (HrmEmployeeDO employee : deptEmployees) {
|
generateMonthlySummary(employee.getId(), yearMonth);
|
}
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public void generateAllMonthlySummary(String yearMonth) {
|
// 获取所有员工（未指定员工状态时查询全部）
|
List<HrmEmployeeDO> employees = employeeService.getEmployeeListByStatus(null);
|
if (CollUtil.isEmpty(employees)) {
|
return;
|
}
|
for (HrmEmployeeDO employee : employees) {
|
generateMonthlySummary(employee.getId(), yearMonth);
|
}
|
}
|
|
/**
|
* 计算应出勤天数（工作日数量）
|
*/
|
private int calculateWorkDays(LocalDate startDate, LocalDate endDate) {
|
int workDays = 0;
|
LocalDate current = startDate;
|
while (!current.isAfter(endDate)) {
|
if (attendanceHolidayService.isWorkDay(current)) {
|
workDays++;
|
}
|
current = current.plusDays(1);
|
}
|
return workDays;
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public PageResult<HrmAttendanceSummaryDO> getAttendanceSummaryPage(HrmAttendanceSummaryPageReqVO pageReqVO) {
|
PageResult<HrmAttendanceSummaryDO> pageResult = attendanceSummaryMapper.selectPage(pageReqVO,
|
new LambdaQueryWrapperX<HrmAttendanceSummaryDO>()
|
.eqIfPresent(HrmAttendanceSummaryDO::getUserId, pageReqVO.getUserId())
|
.eqIfPresent(HrmAttendanceSummaryDO::getDeptId, pageReqVO.getDeptId())
|
.eqIfPresent(HrmAttendanceSummaryDO::getYearMonth, pageReqVO.getYearMonth())
|
.orderByDesc(HrmAttendanceSummaryDO::getYearMonth)
|
.orderByAsc(HrmAttendanceSummaryDO::getDeptId));
|
refreshWorkDays(pageResult.getList());
|
return pageResult;
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public HrmAttendanceSummaryDO getMonthlySummary(Long userId, String yearMonth) {
|
HrmAttendanceSummaryDO summary = attendanceSummaryMapper.selectByUserIdAndYearMonth(userId, yearMonth);
|
if (summary != null) {
|
refreshWorkDays(List.of(summary));
|
}
|
return summary;
|
}
|
|
@Override
|
@Transactional(rollbackFor = Exception.class)
|
public List<HrmAttendanceSummaryDO> getDeptMonthlySummaryList(Long deptId, String yearMonth) {
|
List<HrmAttendanceSummaryDO> list = attendanceSummaryMapper.selectListByDeptIdAndYearMonth(deptId, yearMonth);
|
refreshWorkDays(list);
|
return list;
|
}
|
|
/**
|
* 查询/导出时按“最新节假日配置”重算应出勤天数，并把变化的记录回写。
|
*
|
* 节假日配置可能在生成汇总之后被调整（如新增调休/法定假日），若只返回旧值，
|
* 页面上的应出勤天数会与节假日配置长期不一致。
|
*/
|
private void refreshWorkDays(List<HrmAttendanceSummaryDO> list) {
|
if (CollUtil.isEmpty(list)) {
|
return;
|
}
|
// 同一次请求内同一年月只计算一次
|
Map<String, Integer> workDaysCache = new HashMap<>();
|
for (HrmAttendanceSummaryDO summary : list) {
|
if (StrUtil.isBlank(summary.getYearMonth())) {
|
continue;
|
}
|
Integer expected;
|
try {
|
expected = workDaysCache.computeIfAbsent(summary.getYearMonth(), yearMonth -> {
|
YearMonth parsed = YearMonth.parse(yearMonth);
|
return calculateWorkDays(parsed.atDay(1), parsed.atEndOfMonth());
|
});
|
} catch (Exception e) {
|
log.warn("[refreshWorkDays][考勤汇总({}) 年月({}) 无法解析，跳过重算]",
|
summary.getId(), summary.getYearMonth(), e);
|
continue;
|
}
|
if (!Objects.equals(summary.getWorkDays(), expected)) {
|
log.info("[refreshWorkDays][考勤汇总({}) 年月({}) 应出勤天数 {} -> {}，按最新节假日配置回写]",
|
summary.getId(), summary.getYearMonth(), summary.getWorkDays(), expected);
|
summary.setWorkDays(expected);
|
try {
|
attendanceSummaryMapper.updateById(HrmAttendanceSummaryDO.builder()
|
.id(summary.getId()).workDays(expected).build());
|
} catch (Exception e) {
|
// 回写失败不影响本次查询结果，保留内存中重算后的值
|
log.warn("[refreshWorkDays][考勤汇总({}) 应出勤天数回写失败]", summary.getId(), e);
|
}
|
}
|
}
|
}
|
|
}
|
