KpiGroupUserTask.java
5.17 KB
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
package cn.fw.morax.server.task;
import cn.fw.common.cache.locker.DistributedLocker;
import cn.fw.morax.common.constant.TimeTaskConstant;
import cn.fw.morax.common.utils.PublicUtil;
import cn.fw.morax.domain.db.kpi.KpiGroup;
import cn.fw.morax.domain.db.kpi.KpiGroupUser;
import cn.fw.morax.domain.enums.KpiIgnoreCauseEnum;
import cn.fw.morax.service.biz.kpi.KpiGroupBizService;
import cn.fw.morax.service.biz.kpi.KpiGroupUserBizService;
import cn.fw.morax.service.data.kpi.KpiGroupUserService;
import cn.hutool.core.date.StopWatch;
import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import lombok.Getter;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.redisson.api.RLock;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;
import java.time.LocalDate;
import java.util.*;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.Lock;
import java.util.stream.Collectors;
/**
* @author : jiangchao
* @className : KpiGroupStatusTask
* @description : 绩效组人员定时器
* @date : 2022-04-07 15:29
*/
@Component
@Slf4j
@RequiredArgsConstructor
@ConditionalOnProperty(prefix = "task", name = "switch", havingValue = "on")
public class KpiGroupUserTask {
private final KpiGroupUserBizService kpiGroupUserBizService;
private final KpiGroupUserService kpiGroupUserService;
private final KpiGroupBizService kpiGroupBizService;
private final DistributedLocker distributedLocker;
@Value("${spring.cache.custom.global-prefix}:kpi:group:user")
@Getter
private String kpiGroupUserDistKey;
/**
* 每天凌晨30分执行（人事角色、岗位调整在5点执行）
* 1. 将待生效数据改为生效中
* 2. 将之前的配置设置为失效
* 3. 处理重复数据，合并、拆分门店的绩效组配置需要把其他的重复配置失效
*/
@Scheduled(cron = TimeTaskConstant.KPI_GROUP_USER)
@Transactional(rollbackFor = Exception.class)
public void processKpiUser() {
Lock lock = distributedLocker.lock(getKpiGroupUserDistKey());
if (((RLock) lock).isLocked()) {
try {
log.info("定时任务【绩效组人员更新】开始执行");
StopWatch stopWatch = new StopWatch();
stopWatch.start("绩效组人员更新,查询人事系统人员状态");
//查询员工
LocalDate yesterday = LocalDate.now().minusDays(1);
List<KpiGroupUser> kpiGroupUsersForSave = this.queryKpiStaff(yesterday);
stopWatch.stop();
if (PublicUtil.isEmpty(kpiGroupUsersForSave)) {
log.info(stopWatch.prettyPrint());
return;
}
stopWatch.start("绩效组人员保存，保存条数："+ kpiGroupUsersForSave.size());
//逻辑删除今天已保存人员，保存人员
Set<Long> kpiGroupIds = kpiGroupUsersForSave.stream().map(KpiGroupUser::getKpiGroupId).collect(Collectors.toSet());
kpiGroupUserService.update(Wrappers.<KpiGroupUser>lambdaUpdate()
.in(KpiGroupUser::getKpiGroupId, kpiGroupIds)
.eq(KpiGroupUser::getDataDate, yesterday)
.eq(KpiGroupUser::getYn, Boolean.TRUE)
.set(KpiGroupUser::getYn, Boolean.FALSE)
);
kpiGroupUserService.saveBatch(kpiGroupUsersForSave);
stopWatch.stop();
log.info(stopWatch.prettyPrint(TimeUnit.SECONDS));
} catch (Exception e){
log.error(e.getMessage(), e);
} finally {
lock.unlock();
}
}
}
/**
* 查询绩效员工
* @return
*/
private List<KpiGroupUser> queryKpiStaff(LocalDate yesterday) {
Map<Long, List<KpiGroup>> postKpis = kpiGroupBizService.postEffectKpis();
if (! postKpis.isEmpty()) {
log.info("需要生成绩效组人员的绩效组：{}", postKpis);
}
Long postId = null;
List<KpiGroup> kpiGroups = null;
List<KpiGroupUser> kpiGroupUsers = null;
List<KpiGroupUser> kpiGroupUsersForSave = Lists.newArrayListWithCapacity(300);
KpiIgnoreCauseEnum filterEnum = KpiIgnoreCauseEnum.PROBATION;
for (Map.Entry<Long, List<KpiGroup>> entry : postKpis.entrySet()) {
postId = entry.getKey();
kpiGroups = entry.getValue();
kpiGroupUsers = kpiGroupUserBizService.fetchBuildKpiUser(postId, kpiGroups, yesterday).stream()
// .filter(user -> (PublicUtil.isEmpty(user.getIgnoreCause()) || (! filterEnum.equals(user.getIgnoreCause()))))
.collect(Collectors.toList());
kpiGroupUsersForSave.addAll(kpiGroupUsers);
}
return kpiGroupUsersForSave;
}
}