merage

kimi

2020-05-27 c007f0ca1785db093d48f4846cda82fe8e955765

 src/main/java/com/highdatas/mdm/service/act/impl/IdentityServiceImpl.java

@@ -4,6 +4,8 @@
import com.highdatas.mdm.mapper.TableInfoMapper;
import com.highdatas.mdm.pojo.CodeMsg;
import com.highdatas.mdm.pojo.Result;
import com.highdatas.mdm.util.RedisClient;
import lombok.extern.slf4j.Slf4j;
import org.activiti.engine.IdentityService;
import org.activiti.engine.RepositoryService;
import org.activiti.engine.RuntimeService;
@@ -15,12 +17,16 @@
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

@Service
@Slf4j
public class IdentityServiceImpl extends BaseServiceImpl implements com.highdatas.mdm.service.act.IdentityService{
   @Autowired
   IdentityService identityService;
@@ -30,12 +36,30 @@
   RepositoryService repositoryService;
   @Autowired
   TableInfoMapper tableInfoMapper;
   @Autowired
   RedisClient redisClient;

   /**
    *
    * @description:  启动流程
    * @param businessId 业务id
    * @param key 流程key
    * @return: 实例id
    *
    */
    @Override
   public String startProcess(String businessId, String key) {
      return startProcess(businessId, key, null, null);
   }

   /**
    *
    * @description:  启动流程
    * @param businessId 业务id
    * @param key 流程key
    * @param variables 额外参数
    * @return: 实例id
    *
    */
    @Override
   public String startProcess(String businessId, String key, String workflowStall, HashMap<String, Object> variables) {
       TUser user = getOnlineUser();
@@ -60,19 +84,26 @@
           variables.put("workflowstall", workflowStallInt);
           
           identityService.setAuthenticatedUserId(user.getUserId());

         Date startDate = new Date();
            //get max version processDefinition
            List<Model> modelList = repositoryService.createModelQuery().modelKey(key).orderByModelVersion().desc().list();
         Date endDate = new Date();
         log.info("process load model:" + (endDate.getTime() - startDate.getTime()) +"ms");
            ProcessDefinition processDefinition = null;
            
            for (Model model : modelList) {
            if (processDefinition == null && model.getDeploymentId() != null) {
               String deploymentId = model.getDeploymentId();
               processDefinition = repositoryService.createProcessDefinitionQuery().deploymentId(deploymentId).singleResult();
               break;
            }
         }

         startDate = new Date();
         log.info("process load processDef:" + (startDate.getTime() - endDate.getTime()) +"ms");
            processInstance = runtimeService.startProcessInstanceById(processDefinition.getId(), businessId, variables);

         endDate = new Date();
         log.info("process start process :" + (endDate.getTime() - startDate.getTime()) +"ms");

            String processInstanceId = processInstance.getId();
            //TODO db
@@ -89,7 +120,15 @@
       }
   }

   /**
    *
    * @description:  添加工作流用户
    * @param id 用户id
    * @return: 添加结果
    *
    */
   @Override
   @Transactional(rollbackFor = {RuntimeException.class, Error.class})
   public Result addUser(String id) {
      User preUser = identityService.createUserQuery().userId(id).singleResult();
      if (preUser != null) {
@@ -106,7 +145,15 @@

   }

   /**
    *
    * @description:  添加工作流角色
    * @param id 角色id
    * @return: 添加结果
    *
    */
   @Override
   @Transactional(rollbackFor = {RuntimeException.class, Error.class})
   public Result addRole(String id) {
      Group preGroup = identityService.createGroupQuery().groupId(id).singleResult();
      if (preGroup != null) {
@@ -122,7 +169,16 @@
      }
   }

   /**
    *
    * @description:  添加工作流用户，角色关联
    * @param roleId 觉得id
    * @param userId 用户id
    * @return: 添加结果
    *
    */
   @Override
   @Transactional(rollbackFor = {RuntimeException.class, Error.class})
   public Result addUserRole(String roleId, String userId) {
       try {
         User user = identityService.createUserQuery().userId(userId).singleResult();
@@ -142,14 +198,31 @@
         return Result.error(new CodeMsg(3002, e.getMessage()));
      }
   }

   /**
    *
    * @description:  删除工作流用户
    * @param id 用户id
    * @return: 删除结果
    *
    */
   @Override
   @Transactional(rollbackFor = {RuntimeException.class, Error.class})
   public Result deleteUser(String id) {
      try {
         User user = identityService.createUserQuery().userId(id).singleResult();
         if (user == null) {
            return Result.error(new CodeMsg(3001, "当前用户不存在"));
         }
         List<Map<String, Object>> maps = tableInfoMapper.selectActMemberShipByUser(id);
         for (Map<String, Object> map : maps) {
            Object group_id_ = map.get("GROUP_ID_");
            Object user_id_ = map.get("USER_ID_");
            if (user_id_ == null || group_id_ == null) {
               continue;
            }
            identityService.deleteMembership(user_id_.toString(), group_id_.toString());
         }
         redisClient.delByCharacter(id);
         identityService.deleteUser(id);
         return Result.success(null);
      }catch (Exception e){
@@ -158,13 +231,30 @@
      }

   }

   /**
    *
    * @description:  删除工作流角色
    * @param id 角色id
    * @return: 删除结果
    *
    */
   @Override
   @Transactional(rollbackFor = {RuntimeException.class, Error.class})
   public Result deleteRole(String id) {
      try {
         Group group = identityService.createGroupQuery().groupId(id).singleResult();
         if (group == null) {
            return Result.error(new CodeMsg(3001, "当前角色不存在"));
         }
         List<Map<String, Object>> maps = tableInfoMapper.selectActMemberShipByRole(id);
         for (Map<String, Object> map : maps) {
            Object group_id_ = map.get("GROUP_ID_");
            Object user_id_ = map.get("USER_ID_");
            if (user_id_ == null || group_id_ == null) {
               continue;
            }
            identityService.deleteMembership(user_id_.toString(), group_id_.toString());
            redisClient.delByCharacter(user_id_.toString());
         }
         identityService.deleteGroup(id);
         return Result.success(null);
@@ -174,23 +264,24 @@
      }
   }


   /**
    *
    * @description:  删除工作流用户 角色关联
    * @param userId 用户id
    * @param roleId 角色id
    * @return: 删除结果
    *
    */
   @Override
   @Transactional(rollbackFor = {RuntimeException.class, Error.class})
   public Result deleteUserRole(String roleId, String userId) {
      try {
         User user = identityService.createUserQuery().userId(userId).singleResult();
         if (user == null) {
            return Result.error(new CodeMsg(3001, "当前用户不存在,请先创建用户"));
         }
         Group group = identityService.createGroupQuery().groupId(roleId).singleResult();
         if (user == null) {
            return Result.error(new CodeMsg(3001, "当前角色不存在,请先角色用户"));
         }

            Map<String, String> memberShip = tableInfoMapper.selectActMemberShip(userId, roleId);
            if (memberShip == null) {
                return Result.success(null);
            }
         redisClient.delByCharacter(userId);
         identityService.deleteMembership(userId,roleId);
         return Result.success(null);
      }
@@ -199,6 +290,21 @@
         return Result.error(new CodeMsg(3002, e.getMessage()));
      }
   }
   /**
    *
    * @description:  通过user获取关联的角色
    * @param userId 用户id
    * @return: 角色id列表
    *
    */
   @Override
   public List<String> getRoleByUser(String userId) {
      List<Group> list = identityService.createGroupQuery().groupMember(userId).list();
      if (list == null || list.isEmpty()) {
         return null;
      }
      return  list.stream().map(group -> group.getId()).collect(Collectors.toList());
   }


}