tuqin
2 天以前 531b50cab6e42e6146a419bc71c5b494fff774b2
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
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.framework.mybatis.core.query.LambdaQueryWrapperX;
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.time.LocalDate;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
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());
        validateEffectiveDateRange(createReqVO.getEffectiveStartDate(), createReqVO.getEffectiveEndDate());
        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());
        validateEffectiveDateRange(updateReqVO.getEffectiveStartDate(), updateReqVO.getEffectiveEndDate());
        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(10);
        }
        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)) {
            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 validateEffectiveDateRange(LocalDate startDate, LocalDate endDate) {
        if (startDate != null && endDate != null && startDate.isAfter(endDate)) {
            throw exception(ATTENDANCE_RULE_EFFECTIVE_DATE_INVALID);
        }
    }
 
    @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, LocalDate date) {
        LocalDate targetDate = date != null ? date : LocalDate.now();
        // 1. 先查找部门专属规则(覆盖目标日期的,生效开始日期最晚的优先)
        if (deptId != null) {
            List<HrmAttendanceRuleDO> deptRules = selectEffectiveRules(deptId, targetDate);
            if (!deptRules.isEmpty()) {
                return deptRules.get(0);
            }
        }
        // 2. 查找全局规则(deptId 为 null)
        List<HrmAttendanceRuleDO> globalRules = selectEffectiveRules(null, targetDate);
        if (!globalRules.isEmpty()) {
            return globalRules.get(0);
        }
        return null;
    }
 
    /**
     * 查询在指定日期生效的规则
     *
     * @param deptId 部门ID,null 表示全局规则
     * @param date   目标日期
     */
    private List<HrmAttendanceRuleDO> selectEffectiveRules(Long deptId, LocalDate date) {
        LambdaQueryWrapperX<HrmAttendanceRuleDO> wrapper = new LambdaQueryWrapperX<>();
        if (deptId == null) {
            wrapper.isNull(HrmAttendanceRuleDO::getDeptId);
        } else {
            wrapper.eq(HrmAttendanceRuleDO::getDeptId, deptId);
        }
        wrapper.eq(HrmAttendanceRuleDO::getStatus, 0);
        // 生效开始日期为空(不限)或 <= 目标日期
        wrapper.and(w -> w.isNull(HrmAttendanceRuleDO::getEffectiveStartDate)
                .or().le(HrmAttendanceRuleDO::getEffectiveStartDate, date));
        // 生效结束日期为空(不限)或 >= 目标日期
        wrapper.and(w -> w.isNull(HrmAttendanceRuleDO::getEffectiveEndDate)
                .or().ge(HrmAttendanceRuleDO::getEffectiveEndDate, date));
        // 生效开始日期越晚优先级越高;同开始日期时取先创建的
        wrapper.orderByDesc(HrmAttendanceRuleDO::getEffectiveStartDate);
        wrapper.orderByAsc(HrmAttendanceRuleDO::getId);
        return attendanceRuleMapper.selectList(wrapper);
    }
 
    /**
     * 导入考勤规则(移植自新高电力)
     *
     * 说明:导入按「规则名称」判重;已存在时需 updateSupport=true 才覆盖。
     * 部门按名称匹配(deptName -> deptId),部门为空表示全局规则。
     */
    @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);
                    applyImportDefaults(rule);
                    attendanceRuleMapper.insert(rule);
                    respVO.getCreateNames().add(importRule.getName());
                    return;
                }
                // 3.2.2 已存在,判断是否允许更新
                if (!Boolean.TRUE.equals(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);
                applyImportDefaults(updateRule);
                attendanceRuleMapper.updateById(updateRule);
                respVO.getUpdateNames().add(importRule.getName());
            } catch (ServiceException ex) {
                respVO.getFailureNames().put(importRule.getName(), ex.getMessage());
            }
        });
        return respVO;
    }
 
    /**
     * 导入时的默认值补全(与页面新增保持一致的默认口径;不覆盖地点字段,避免导入清空已配置的地点)
     */
    private void applyImportDefaults(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);
        }
    }
}