package com.ycl.task;
|
|
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
|
import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper;
|
import com.ycl.common.enums.business.CodingRulerCodeTypeEnum;
|
import com.ycl.common.enums.business.CodingRulerIntervalTypeEnum;
|
import com.ycl.common.enums.business.CodingRulerSymbolTypeEnum;
|
import com.ycl.common.enums.business.ProjectProcessTypeEnum;
|
import com.ycl.domain.entity.*;
|
import com.ycl.domain.vo.CodingRulerVO;
|
import com.ycl.domain.vo.ProcessCodingVO;
|
import com.ycl.domain.vo.ProjectCodingStatusVO;
|
import com.ycl.domain.vo.ProjectProcessVO;
|
import com.ycl.factory.FlowServiceFactory;
|
import com.ycl.mapper.*;
|
import com.ycl.service.ProjectCodingStatusService;
|
import liquibase.pro.packaged.A;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
import org.springframework.util.CollectionUtils;
|
|
import java.util.ArrayList;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.Objects;
|
import java.util.function.Function;
|
import java.util.stream.Collectors;
|
|
@Slf4j
|
@Component("codingTask")
|
public class CodingTask extends FlowServiceFactory {
|
@Autowired
|
private ProjectCodingStatusMapper projectCodingStatusMapper;
|
|
@Autowired
|
private ProjectCodingStatusService projectCodingStatusService;
|
|
@Autowired
|
private ProjectProcessMapper projectProcessMapper;
|
@Autowired
|
private ProjectInfoMapper projectInfoMapper;
|
@Autowired
|
private CodingRulerMapper codingRulerMapper;
|
|
@Autowired
|
private ProcessCodingMapper processCodingMapper;
|
|
|
public void codingTask(){
|
log.info("开始计算项目并且赋码");
|
//获得启用的赋码规则
|
List<CodingRulerVO> codingRulerVOS = codingRulerMapper.getStartRuler();
|
//没启用直接结束
|
if (CollectionUtils.isEmpty(codingRulerVOS)){
|
log.info("未启用赋码规则");
|
return;
|
}
|
CodingRulerVO yellowRuler = null;
|
CodingRulerVO redRuler = null;
|
for (CodingRulerVO codingRulerVO : codingRulerVOS){
|
if (CodingRulerCodeTypeEnum.YELLOW.getValue().equals(codingRulerVO.getProjectCodeType())){
|
yellowRuler = codingRulerVO;
|
}else if (CodingRulerCodeTypeEnum.RED.getValue().equals(codingRulerVO.getProjectCodeType())){
|
redRuler = codingRulerVO;
|
}
|
}
|
|
//获得所有 超时任务数与流程实例id
|
List<ProcessCodingVO> processCodingVOS = processCodingMapper.countProjectOverTimeNum();
|
if (!CollectionUtils.isEmpty(processCodingVOS)){
|
//将流程实例id转换为list
|
List<String> processIds = processCodingVOS.stream().map(ProcessCodingVO::getProcessInsId).toList();
|
//获得所有 更具流程定义id，流程相关项目id， key为流程id
|
Map<String, ProjectProcess> projectMap = new LambdaQueryChainWrapper<>(projectProcessMapper)
|
.in(ProjectProcess::getProcessInsId, processIds)
|
.eq(ProjectProcess::getProjectType, ProjectProcessTypeEnum.PROJECT)
|
.list()
|
.stream()
|
.collect(Collectors.toMap(ProjectProcess::getProcessInsId, Function.identity()));
|
|
//修改为黄码的项目 id
|
List<String> yellowProjectCodingIds = new ArrayList<>();
|
//修改为红码的项目 id
|
List<String> redProjectCodingIds = new ArrayList<>();
|
|
|
//新增的记录集合
|
List<ProjectCodingStatus> addList = new ArrayList<>();
|
|
for (ProcessCodingVO processCoding : processCodingVOS) {
|
//判断 项目类型为 ProjectProcessTypeEnum.PROJECT 并且超时任务数大于0
|
if (processCoding.getTaskOvertimeNum() != 0 && projectMap.containsKey(processCoding.getProcessInsId())) {
|
ProjectCodingStatus projectCodingStatus = new ProjectCodingStatus();
|
projectCodingStatus.setProjectType(ProjectProcessTypeEnum.PROJECT.getValue());
|
projectCodingStatus.setProjectId(projectMap.get(processCoding.getProcessInsId()).getProjectId());
|
projectCodingStatus.setProcessInsId(processCoding.getProcessInsId());
|
projectCodingStatus.setTaskOvertimeNum(processCoding.getTaskOvertimeNum());
|
//添加到新增集合内
|
addList.add(projectCodingStatus);
|
//检验赋码规则
|
checkCodingType(projectCodingStatus.getTaskOvertimeNum(),
|
projectCodingStatus.getProjectId(),
|
yellowProjectCodingIds,redProjectCodingIds,
|
yellowRuler,redRuler);
|
}
|
}
|
|
if(!CollectionUtils.isEmpty(addList)){
|
projectCodingStatusMapper.delAll();
|
projectCodingStatusService.saveBatch(addList);
|
}
|
//修改项目赋码
|
updateProjectCoding(redProjectCodingIds,"red");
|
log.info("打印赋值为红码的项目id");
|
for (String s : redProjectCodingIds){
|
System.out.println(s);
|
}
|
log.info("打印赋值为红码的项目id完毕");
|
updateProjectCoding(yellowProjectCodingIds,"yellow");
|
log.info("打印赋值为黄码的项目id");
|
for (String s : yellowProjectCodingIds){
|
System.out.println(s);
|
}
|
log.info("打印赋值为黄码的项目id完毕");
|
}
|
|
log.info("结束计算项目并且赋码");
|
}
|
|
/**
|
* 赋码
|
*
|
* @param projectIds 流程实例ID列表
|
* @param coding 赋码值
|
*/
|
private void updateProjectCoding(List<String> projectIds, String coding) {
|
if (!CollectionUtils.isEmpty(projectIds)) {
|
//将id 类型转换为Long
|
List<Long> longList = projectIds.stream()
|
.map(Long::parseLong)
|
.collect(Collectors.toList());
|
new LambdaUpdateChainWrapper<>(projectInfoMapper)
|
.in(ProjectInfo::getId, longList)
|
.set(ProjectInfo::getCoding, coding)
|
.update();
|
}
|
}
|
|
|
/**
|
*
|
* @param overTimeNum 超时任务数
|
* @param projectId 项目id
|
* @param yellowProjectCodingIds 修改项目赋码为黄码的id集合
|
* @param redProjectCodingIds 修改项目赋码为红码的id集合
|
* @param yellowRuler 黄码规则
|
* @param redRuler 红码规则
|
*/
|
private void checkCodingType(Long overTimeNum,String projectId,
|
List<String> yellowProjectCodingIds, List<String> redProjectCodingIds,
|
CodingRulerVO yellowRuler,CodingRulerVO redRuler){
|
//先判断红码，规则如果满足，则不执行黄码规则
|
if (judgeState(overTimeNum, projectId, redProjectCodingIds, redRuler)) return;
|
|
//判断是否满足黄码规则
|
judgeState(overTimeNum, projectId, yellowProjectCodingIds, yellowRuler);
|
|
|
}
|
|
private boolean judgeState(Long overTimeNum, String projectId, List<String> projectCodingIds, CodingRulerVO ruler) {
|
if (ruler != null){
|
if (CodingRulerIntervalTypeEnum.Interval.getValue().equals(ruler.getIntervalType())){
|
//区间情况
|
boolean result = false;
|
//判断满足左区间
|
if (ruler.getLeftSymbol().equals("0")){
|
result = (overTimeNum > ruler.getLeftValue());
|
}else if (ruler.getLeftSymbol().equals("1")){
|
result = (overTimeNum >= ruler.getLeftValue());
|
}else if (ruler.getLeftSymbol().equals("2")){
|
result = (Objects.equals(overTimeNum, ruler.getLeftValue()));
|
}
|
//满足则判断满足右区间
|
if (result){
|
if (ruler.getRightSymbol().equals("2")){
|
result = (Objects.equals(overTimeNum, ruler.getRightValue()));
|
}else if (ruler.getRightSymbol().equals("3")){
|
result = (overTimeNum < ruler.getRightValue());
|
}else if (ruler.getRightSymbol().equals("4")){
|
result = (overTimeNum <= ruler.getRightValue());
|
}
|
}
|
if (result){
|
//符合该赋码条件
|
projectCodingIds.add(projectId);
|
return true;
|
}
|
}
|
else{
|
//单区间情况
|
boolean result = false;
|
if (ruler.getLeftSymbol().equals("0")){
|
result = (overTimeNum > ruler.getLeftValue());
|
}else if (ruler.getLeftSymbol().equals("1")){
|
result = (overTimeNum >= ruler.getLeftValue());
|
}else if (ruler.getLeftSymbol().equals("2")){
|
result = (Objects.equals(overTimeNum, ruler.getLeftValue()));
|
}else if (ruler.getLeftSymbol().equals("3")){
|
result = (overTimeNum < ruler.getLeftValue());
|
}else if (ruler.getLeftSymbol().equals("4")){
|
result = (overTimeNum <= ruler.getLeftValue());
|
}
|
if (result){
|
//符合该赋码条件
|
projectCodingIds.add(projectId);
|
return true;
|
}
|
|
}
|
}
|
return false;
|
}
|
|
}
|
