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.exception.ServiceException;
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
|
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmAttendanceRuleImportExcelVO;
|
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmAttendanceRuleImportReqVO;
|
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmAttendanceRuleImportRespVO;
|
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmAttendanceRulePageReqVO;
|
import cn.iocoder.yudao.module.hrm.controller.admin.attendance.vo.HrmAttendanceRuleSaveReqVO;
|
import cn.iocoder.yudao.module.hrm.dal.dataobject.attendance.HrmAttendanceRuleDO;
|
import cn.iocoder.yudao.module.hrm.dal.mysql.attendance.HrmAttendanceRuleMapper;
|
import cn.iocoder.yudao.module.system.api.dept.DeptApi;
|
import cn.iocoder.yudao.module.system.api.dept.dto.DeptRespDTO;
|
import jakarta.annotation.Resource;
|
import org.springframework.stereotype.Service;
|
import org.springframework.validation.annotation.Validated;
|
|
import java.math.BigDecimal;
|
import java.util.ArrayList;
|
import java.util.HashSet;
|
import java.util.LinkedHashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Set;
|
import java.util.stream.Collectors;
|
|
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
|
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.filterList;
|
import static cn.iocoder.yudao.module.hrm.enums.ErrorCodeConstants.*;
|
|
/**
|
* 考勤规则配置 Service 实现类
|
*
|
* @author 超级管理员
|
*/
|
@Service
|
@Validated
|
public class HrmAttendanceRuleServiceImpl implements HrmAttendanceRuleService {
|
|
@Resource
|
private HrmAttendanceRuleMapper attendanceRuleMapper;
|
@Resource
|
private DeptApi deptApi;
|
|
@Override
|
public Long createAttendanceRule(HrmAttendanceRuleSaveReqVO createReqVO) {
|
validateLocation(createReqVO);
|
// 校验规则名称是否重复
|
validateAttendanceRuleNameUnique(null, createReqVO.getName());
|
// 校验部门是否已配置规则（如果指定了部门）
|
if (createReqVO.getDeptId() != null) {
|
validateAttendanceRuleDeptUnique(null, createReqVO.getDeptId());
|
}
|
HrmAttendanceRuleDO attendanceRule = BeanUtils.toBean(createReqVO, HrmAttendanceRuleDO.class);
|
// 设置默认值
|
if (attendanceRule.getLateAllowMinutes() == null) {
|
attendanceRule.setLateAllowMinutes(0);
|
}
|
if (attendanceRule.getEarlyAllowMinutes() == null) {
|
attendanceRule.setEarlyAllowMinutes(0);
|
}
|
if (attendanceRule.getOvertimeThreshold() == null) {
|
attendanceRule.setOvertimeThreshold(BigDecimal.ONE);
|
}
|
if (attendanceRule.getStatus() == null) {
|
attendanceRule.setStatus(0);
|
}
|
attendanceRuleMapper.insert(attendanceRule);
|
return attendanceRule.getId();
|
}
|
|
@Override
|
public void updateAttendanceRule(HrmAttendanceRuleSaveReqVO updateReqVO) {
|
HrmAttendanceRuleDO currentRule = attendanceRuleMapper.selectById(updateReqVO.getId());
|
if (currentRule == null) {
|
throw exception(ATTENDANCE_RULE_NOT_EXISTS);
|
}
|
// 兼容尚未传地点字段的旧管理端，避免编辑班次时意外关闭已配置的位置限制。
|
if (updateReqVO.getLocationRequired() == null) {
|
updateReqVO.setLocationRequired(currentRule.getLocationRequired());
|
if (updateReqVO.getLocationName() == null) updateReqVO.setLocationName(currentRule.getLocationName());
|
if (updateReqVO.getLocationAddress() == null) updateReqVO.setLocationAddress(currentRule.getLocationAddress());
|
if (updateReqVO.getLongitude() == null) updateReqVO.setLongitude(currentRule.getLongitude());
|
if (updateReqVO.getLatitude() == null) updateReqVO.setLatitude(currentRule.getLatitude());
|
if (updateReqVO.getRadiusMeters() == null) updateReqVO.setRadiusMeters(currentRule.getRadiusMeters());
|
if (updateReqVO.getCoordinateSystem() == null) updateReqVO.setCoordinateSystem(currentRule.getCoordinateSystem());
|
}
|
validateLocation(updateReqVO);
|
// 校验规则名称是否重复
|
validateAttendanceRuleNameUnique(updateReqVO.getId(), updateReqVO.getName());
|
// 校验部门是否已配置规则（如果指定了部门）
|
if (updateReqVO.getDeptId() != null) {
|
validateAttendanceRuleDeptUnique(updateReqVO.getId(), updateReqVO.getDeptId());
|
}
|
HrmAttendanceRuleDO updateObj = BeanUtils.toBean(updateReqVO, HrmAttendanceRuleDO.class);
|
attendanceRuleMapper.updateById(updateObj);
|
}
|
|
@Override
|
public void deleteAttendanceRule(Long id) {
|
validateAttendanceRuleExists(id);
|
attendanceRuleMapper.deleteById(id);
|
}
|
|
private void validateAttendanceRuleExists(Long id) {
|
if (attendanceRuleMapper.selectById(id) == null) {
|
throw exception(ATTENDANCE_RULE_NOT_EXISTS);
|
}
|
}
|
|
private void validateLocation(HrmAttendanceRuleSaveReqVO reqVO) {
|
if (reqVO.getLocationRequired() == null) {
|
reqVO.setLocationRequired(false);
|
}
|
if (reqVO.getRadiusMeters() == null) {
|
reqVO.setRadiusMeters(200);
|
}
|
if (StrUtil.isBlank(reqVO.getCoordinateSystem())) {
|
reqVO.setCoordinateSystem("WGS84");
|
} else {
|
reqVO.setCoordinateSystem(reqVO.getCoordinateSystem().trim().toUpperCase(java.util.Locale.ROOT));
|
}
|
if (!"WGS84".equals(reqVO.getCoordinateSystem())) {
|
throw exception(ATTENDANCE_LOCATION_SYSTEM_INVALID);
|
}
|
if (reqVO.getLongitude() != null && (reqVO.getLongitude().compareTo(BigDecimal.valueOf(-180)) < 0
|
|| reqVO.getLongitude().compareTo(BigDecimal.valueOf(180)) > 0)) {
|
throw exception(ATTENDANCE_LOCATION_COORDINATE_INVALID);
|
}
|
if (reqVO.getLatitude() != null && (reqVO.getLatitude().compareTo(BigDecimal.valueOf(-90)) < 0
|
|| reqVO.getLatitude().compareTo(BigDecimal.valueOf(90)) > 0)) {
|
throw exception(ATTENDANCE_LOCATION_COORDINATE_INVALID);
|
}
|
if (reqVO.getRadiusMeters() < 10 || reqVO.getRadiusMeters() > 5000) {
|
throw exception(ATTENDANCE_LOCATION_RULE_INVALID);
|
}
|
if (Boolean.TRUE.equals(reqVO.getLocationRequired())
|
&& (StrUtil.isBlank(reqVO.getLocationName()) || reqVO.getLongitude() == null
|
|| reqVO.getLatitude() == null || StrUtil.isBlank(reqVO.getCoordinateSystem()))) {
|
throw exception(ATTENDANCE_LOCATION_RULE_INVALID);
|
}
|
}
|
|
private void validateAttendanceRuleNameUnique(Long id, String name) {
|
HrmAttendanceRuleDO rule = attendanceRuleMapper.selectOne(HrmAttendanceRuleDO::getName, name);
|
if (rule == null) {
|
return;
|
}
|
if (id == null) {
|
throw exception(ATTENDANCE_RULE_NAME_EXISTS);
|
}
|
if (!rule.getId().equals(id)) {
|
throw exception(ATTENDANCE_RULE_NAME_EXISTS);
|
}
|
}
|
|
private void validateAttendanceRuleDeptUnique(Long id, Long deptId) {
|
List<HrmAttendanceRuleDO> rules = attendanceRuleMapper.selectListByDeptId(deptId);
|
if (rules.isEmpty()) {
|
return;
|
}
|
for (HrmAttendanceRuleDO rule : rules) {
|
if (id == null || !rule.getId().equals(id)) {
|
throw exception(ATTENDANCE_RULE_DEPT_EXISTS);
|
}
|
}
|
}
|
|
@Override
|
public HrmAttendanceRuleDO getAttendanceRule(Long id) {
|
return attendanceRuleMapper.selectById(id);
|
}
|
|
@Override
|
public PageResult<HrmAttendanceRuleDO> getAttendanceRulePage(HrmAttendanceRulePageReqVO pageReqVO) {
|
return attendanceRuleMapper.selectPage(pageReqVO);
|
}
|
|
@Override
|
public List<HrmAttendanceRuleDO> getAttendanceRuleList() {
|
return attendanceRuleMapper.selectListByStatus(0);
|
}
|
|
@Override
|
public HrmAttendanceRuleDO getAttendanceRuleByDeptId(Long deptId) {
|
// 1. 从员工所在部门开始逐级向上查找：配置在上级部门的规则对其子部门生效
|
Set<Long> visitedDeptIds = new HashSet<>();
|
Long currentDeptId = deptId;
|
while (currentDeptId != null && currentDeptId != 0 && visitedDeptIds.add(currentDeptId)) {
|
List<HrmAttendanceRuleDO> deptRules = attendanceRuleMapper.selectEnabledListByDeptId(currentDeptId);
|
if (!deptRules.isEmpty()) {
|
// 同一部门（如「总公司」）可能配置了多条启用规则，以最新创建的为当前生效规则
|
return deptRules.get(0);
|
}
|
DeptRespDTO dept = deptApi.getDept(currentDeptId);
|
if (dept == null) {
|
break;
|
}
|
currentDeptId = dept.getParentId();
|
}
|
// 2. 查找全局规则（deptId 为 null）
|
List<HrmAttendanceRuleDO> globalRules = attendanceRuleMapper.selectList(new cn.iocoder.yudao.framework.mybatis.core.query.LambdaQueryWrapperX<HrmAttendanceRuleDO>()
|
.isNull(HrmAttendanceRuleDO::getDeptId)
|
.eq(HrmAttendanceRuleDO::getStatus, 0)
|
.orderByDesc(HrmAttendanceRuleDO::getId));
|
if (!globalRules.isEmpty()) {
|
return globalRules.get(0);
|
}
|
return null;
|
}
|
|
@Override
|
public HrmAttendanceRuleImportRespVO importAttendanceRuleList(
|
List<HrmAttendanceRuleImportExcelVO> importList, HrmAttendanceRuleImportReqVO importReqVO) {
|
// 1. 校验非空
|
importList = filterList(importList, item -> StrUtil.isNotBlank(item.getName()));
|
if (CollUtil.isEmpty(importList)) {
|
throw exception(ATTENDANCE_RULE_IMPORT_LIST_IS_EMPTY);
|
}
|
// 2. 预加载部门名称 -> 部门ID
|
Map<String, Long> deptMap = deptApi.getDeptList().stream()
|
.filter(dept -> StrUtil.isNotBlank(dept.getName()))
|
.collect(Collectors.toMap(DeptRespDTO::getName, DeptRespDTO::getId, (a, b) -> a));
|
// 3. 逐条处理
|
HrmAttendanceRuleImportRespVO respVO = HrmAttendanceRuleImportRespVO.builder()
|
.createNames(new ArrayList<>())
|
.updateNames(new ArrayList<>())
|
.failureNames(new LinkedHashMap<>())
|
.build();
|
importList.forEach(importRule -> {
|
try {
|
// 3.1 解析部门
|
Long deptId = null;
|
if (StrUtil.isNotBlank(importRule.getDeptName())) {
|
deptId = deptMap.get(importRule.getDeptName());
|
if (deptId == null) {
|
throw exception(ATTENDANCE_RULE_DEPT_NOT_EXISTS, importRule.getDeptName());
|
}
|
}
|
// 3.2 判断是否存在（按名称唯一）
|
HrmAttendanceRuleDO existRule = attendanceRuleMapper.selectOne(HrmAttendanceRuleDO::getName, importRule.getName());
|
if (existRule == null) {
|
// 3.2.1 新增
|
HrmAttendanceRuleDO rule = BeanUtils.toBean(importRule, HrmAttendanceRuleDO.class).setDeptId(deptId);
|
applyDefaults(rule);
|
attendanceRuleMapper.insert(rule);
|
respVO.getCreateNames().add(importRule.getName());
|
return;
|
}
|
// 3.2.2 已存在，判断是否允许更新
|
if (!importReqVO.getUpdateSupport()) {
|
respVO.getFailureNames().put(importRule.getName(),
|
StrUtil.format(ATTENDANCE_RULE_EXISTS.getMsg(), importRule.getName()));
|
return;
|
}
|
HrmAttendanceRuleDO updateRule = BeanUtils.toBean(importRule, HrmAttendanceRuleDO.class)
|
.setId(existRule.getId()).setDeptId(deptId);
|
applyDefaults(updateRule);
|
attendanceRuleMapper.updateById(updateRule);
|
respVO.getUpdateNames().add(importRule.getName());
|
} catch (ServiceException ex) {
|
respVO.getFailureNames().put(importRule.getName(), ex.getMessage());
|
}
|
});
|
return respVO;
|
}
|
|
private void applyDefaults(HrmAttendanceRuleDO rule) {
|
if (rule.getLateAllowMinutes() == null) {
|
rule.setLateAllowMinutes(0);
|
}
|
if (rule.getEarlyAllowMinutes() == null) {
|
rule.setEarlyAllowMinutes(0);
|
}
|
if (rule.getOvertimeThreshold() == null) {
|
rule.setOvertimeThreshold(BigDecimal.ONE);
|
}
|
if (rule.getStatus() == null) {
|
rule.setStatus(0);
|
}
|
}
|
|
}
|
