|
|
|
|
@ -0,0 +1,166 @@
|
|
|
|
|
package cn.iocoder.yudao.module.bpm.framework.flowable.core.behavior;
|
|
|
|
|
|
|
|
|
|
import cn.hutool.core.collection.CollUtil;
|
|
|
|
|
import cn.hutool.core.util.RandomUtil;
|
|
|
|
|
import cn.hutool.core.util.StrUtil;
|
|
|
|
|
import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
|
|
|
|
|
import cn.iocoder.yudao.module.bpm.dal.dataobject.definition.BpmTaskAssignRuleDO;
|
|
|
|
|
import cn.iocoder.yudao.module.bpm.dal.dataobject.definition.BpmUserGroupDO;
|
|
|
|
|
import cn.iocoder.yudao.module.bpm.enums.definition.BpmTaskAssignRuleTypeEnum;
|
|
|
|
|
import cn.iocoder.yudao.module.bpm.service.definition.BpmTaskAssignRuleService;
|
|
|
|
|
import cn.iocoder.yudao.module.bpm.service.definition.BpmUserGroupService;
|
|
|
|
|
import cn.iocoder.yudao.module.system.api.dept.DeptApi;
|
|
|
|
|
import cn.iocoder.yudao.module.system.api.dept.dto.DeptRespDTO;
|
|
|
|
|
import cn.iocoder.yudao.module.system.api.permission.PermissionApi;
|
|
|
|
|
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
|
|
|
|
|
import cn.iocoder.yudao.module.system.api.user.dto.AdminUserRespDTO;
|
|
|
|
|
import com.google.common.annotations.VisibleForTesting;
|
|
|
|
|
import lombok.Setter;
|
|
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
|
|
|
import org.flowable.bpmn.model.UserTask;
|
|
|
|
|
import org.flowable.common.engine.api.FlowableException;
|
|
|
|
|
import org.flowable.common.engine.impl.el.ExpressionManager;
|
|
|
|
|
import org.flowable.engine.delegate.DelegateExecution;
|
|
|
|
|
import org.flowable.engine.impl.bpmn.behavior.UserTaskActivityBehavior;
|
|
|
|
|
import org.flowable.engine.impl.cfg.ProcessEngineConfigurationImpl;
|
|
|
|
|
import org.flowable.engine.impl.util.TaskHelper;
|
|
|
|
|
import org.flowable.task.service.TaskService;
|
|
|
|
|
import org.flowable.task.service.impl.persistence.entity.TaskEntity;
|
|
|
|
|
|
|
|
|
|
import java.util.*;
|
|
|
|
|
|
|
|
|
|
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
|
|
|
|
|
import static cn.iocoder.yudao.framework.common.util.collection.CollectionUtils.convertSet;
|
|
|
|
|
import static cn.iocoder.yudao.framework.common.util.json.JsonUtils.toJsonString;
|
|
|
|
|
import static cn.iocoder.yudao.module.bpm.enums.ErrorCodeConstants.TASK_CREATE_FAIL_NO_CANDIDATE_USER;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 自定义的流程任务的 assignee 负责人的分配
|
|
|
|
|
* 第一步，获得对应的分配规则；
|
|
|
|
|
* 第二步，根据分配规则，计算出分配任务的候选人。如果找不到，则直接报业务异常，不继续执行后续的流程；
|
|
|
|
|
* 第三步，随机选择一个候选人，则选择作为 assignee 负责人。
|
|
|
|
|
*
|
|
|
|
|
* @author 芋道源码
|
|
|
|
|
*/
|
|
|
|
|
@Slf4j
|
|
|
|
|
public class BpmUserTaskActivityBehavior extends UserTaskActivityBehavior {
|
|
|
|
|
|
|
|
|
|
@Setter
|
|
|
|
|
private BpmTaskAssignRuleService bpmTaskRuleService;
|
|
|
|
|
@Setter
|
|
|
|
|
private BpmUserGroupService userGroupService;
|
|
|
|
|
@Setter
|
|
|
|
|
private DeptApi deptApi;
|
|
|
|
|
@Setter
|
|
|
|
|
private AdminUserApi adminUserApi;
|
|
|
|
|
@Setter
|
|
|
|
|
private PermissionApi permissionApi;
|
|
|
|
|
|
|
|
|
|
public BpmUserTaskActivityBehavior(UserTask userTask) {
|
|
|
|
|
super(userTask);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
protected void handleAssignments(TaskService taskService, String assignee, String owner, List<String> candidateUsers, List<String> candidateGroups, TaskEntity task, ExpressionManager expressionManager, DelegateExecution execution, ProcessEngineConfigurationImpl processEngineConfiguration) {
|
|
|
|
|
// 第一步，获得任务的规则
|
|
|
|
|
BpmTaskAssignRuleDO rule = getTaskRule(task);
|
|
|
|
|
// 第二步，获得任务的候选用户们
|
|
|
|
|
Set<Long> candidateUserIds = calculateTaskCandidateUsers(task, rule);
|
|
|
|
|
// 第三步，设置一个作为负责人
|
|
|
|
|
Long assigneeUserId = chooseTaskAssignee(candidateUserIds);
|
|
|
|
|
TaskHelper.changeTaskAssignee(task, String.valueOf(assigneeUserId));
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private BpmTaskAssignRuleDO getTaskRule(TaskEntity task) {
|
|
|
|
|
List<BpmTaskAssignRuleDO> taskRules = bpmTaskRuleService.getTaskAssignRuleListByProcessDefinitionId(task.getProcessDefinitionId(),
|
|
|
|
|
task.getTaskDefinitionKey());
|
|
|
|
|
if (CollUtil.isEmpty(taskRules)) {
|
|
|
|
|
throw new FlowableException(StrUtil.format("流程任务({}/{}/{}) 找不到符合的任务规则",
|
|
|
|
|
task.getId(), task.getProcessDefinitionId(), task.getTaskDefinitionKey()));
|
|
|
|
|
}
|
|
|
|
|
if (taskRules.size() > 1) {
|
|
|
|
|
throw new FlowableException(StrUtil.format("流程任务({}/{}/{}) 找到过多任务规则({})",
|
|
|
|
|
task.getId(), task.getProcessDefinitionId(), task.getTaskDefinitionKey(), taskRules.size()));
|
|
|
|
|
}
|
|
|
|
|
return taskRules.get(0);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@VisibleForTesting
|
|
|
|
|
Set<Long> calculateTaskCandidateUsers(TaskEntity task, BpmTaskAssignRuleDO rule) {
|
|
|
|
|
Set<Long> assigneeUserIds = null;
|
|
|
|
|
if (Objects.equals(BpmTaskAssignRuleTypeEnum.ROLE.getType(), rule.getType())) {
|
|
|
|
|
assigneeUserIds = calculateTaskCandidateUsersByRole(task, rule);
|
|
|
|
|
} else if (Objects.equals(BpmTaskAssignRuleTypeEnum.DEPT_MEMBER.getType(), rule.getType())) {
|
|
|
|
|
assigneeUserIds = calculateTaskCandidateUsersByDeptMember(task, rule);
|
|
|
|
|
} else if (Objects.equals(BpmTaskAssignRuleTypeEnum.DEPT_LEADER.getType(), rule.getType())) {
|
|
|
|
|
assigneeUserIds = calculateTaskCandidateUsersByDeptLeader(task, rule);
|
|
|
|
|
} else if (Objects.equals(BpmTaskAssignRuleTypeEnum.POST.getType(), rule.getType())) {
|
|
|
|
|
assigneeUserIds = calculateTaskCandidateUsersByPost(task, rule);
|
|
|
|
|
} else if (Objects.equals(BpmTaskAssignRuleTypeEnum.USER.getType(), rule.getType())) {
|
|
|
|
|
assigneeUserIds = calculateTaskCandidateUsersByUser(task, rule);
|
|
|
|
|
} else if (Objects.equals(BpmTaskAssignRuleTypeEnum.USER_GROUP.getType(), rule.getType())) {
|
|
|
|
|
assigneeUserIds = calculateTaskCandidateUsersByUserGroup(task, rule);
|
|
|
|
|
}
|
|
|
|
|
// else if (Objects.equals(BpmTaskAssignRuleTypeEnum.SCRIPT.getType(), rule.getType())) {
|
|
|
|
|
// assigneeUserIds = calculateTaskCandidateUsersByScript(task, rule);
|
|
|
|
|
// }
|
|
|
|
|
|
|
|
|
|
// 移除被禁用的用户
|
|
|
|
|
removeDisableUsers(assigneeUserIds);
|
|
|
|
|
// 如果候选人为空，抛出异常 TODO 芋艿：没候选人的策略选择。1 - 挂起；2 - 直接结束；3 - 强制一个兜底人
|
|
|
|
|
if (CollUtil.isEmpty(assigneeUserIds)) {
|
|
|
|
|
log.error("[calculateTaskCandidateUsers][流程任务({}/{}/{}) 任务规则({}) 找不到候选人]",
|
|
|
|
|
task.getId(), task.getProcessDefinitionId(), task.getTaskDefinitionKey(), toJsonString(rule));
|
|
|
|
|
throw exception(TASK_CREATE_FAIL_NO_CANDIDATE_USER);
|
|
|
|
|
}
|
|
|
|
|
return assigneeUserIds;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private Set<Long> calculateTaskCandidateUsersByRole(TaskEntity task, BpmTaskAssignRuleDO rule) {
|
|
|
|
|
return permissionApi.getUserRoleIdListByRoleIds(rule.getOptions());
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private Set<Long> calculateTaskCandidateUsersByDeptMember(TaskEntity task, BpmTaskAssignRuleDO rule) {
|
|
|
|
|
List<AdminUserRespDTO> users = adminUserApi.getUsersByDeptIds(rule.getOptions());
|
|
|
|
|
return convertSet(users, AdminUserRespDTO::getId);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private Set<Long> calculateTaskCandidateUsersByDeptLeader(TaskEntity task, BpmTaskAssignRuleDO rule) {
|
|
|
|
|
List<DeptRespDTO> depts = deptApi.getDepts(rule.getOptions());
|
|
|
|
|
return convertSet(depts, DeptRespDTO::getLeaderUserId);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private Set<Long> calculateTaskCandidateUsersByPost(TaskEntity task, BpmTaskAssignRuleDO rule) {
|
|
|
|
|
List<AdminUserRespDTO> users = adminUserApi.getUsersByPostIds(rule.getOptions());
|
|
|
|
|
return convertSet(users, AdminUserRespDTO::getId);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private Set<Long> calculateTaskCandidateUsersByUser(TaskEntity task, BpmTaskAssignRuleDO rule) {
|
|
|
|
|
return rule.getOptions();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private Set<Long> calculateTaskCandidateUsersByUserGroup(TaskEntity task, BpmTaskAssignRuleDO rule) {
|
|
|
|
|
List<BpmUserGroupDO> userGroups = userGroupService.getUserGroupList(rule.getOptions());
|
|
|
|
|
Set<Long> userIds = new HashSet<>();
|
|
|
|
|
userGroups.forEach(group -> userIds.addAll(group.getMemberUserIds()));
|
|
|
|
|
return userIds;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private Long chooseTaskAssignee(Set<Long> candidateUserIds) {
|
|
|
|
|
// TODO 芋艿：未来可以优化下，改成轮询的策略
|
|
|
|
|
int index = RandomUtil.randomInt(candidateUserIds.size());
|
|
|
|
|
return CollUtil.get(candidateUserIds, index);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@VisibleForTesting
|
|
|
|
|
void removeDisableUsers(Set<Long> assigneeUserIds) {
|
|
|
|
|
if (CollUtil.isEmpty(assigneeUserIds)) {
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
Map<Long, AdminUserRespDTO> userMap = adminUserApi.getUserMap(assigneeUserIds);
|
|
|
|
|
assigneeUserIds.removeIf(id -> {
|
|
|
|
|
AdminUserRespDTO user = userMap.get(id);
|
|
|
|
|
return user == null || !CommonStatusEnum.ENABLE.getStatus().equals(user.getStatus());
|
|
|
|
|
});
|
|
|
|
|
}
|
|
|
|
|
}
|
jason
4 years ago