add 数据权限

kimi

2020-03-24 b5636e416a2fe029f96aec6ddfd3489394925b15

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

@@ -1,6 +1,7 @@
package com.highdatas.mdm.service.act.impl;

import com.highdatas.mdm.entity.TUser;
import com.highdatas.mdm.mapper.TableInfoMapper;
import com.highdatas.mdm.pojo.CodeMsg;
import com.highdatas.mdm.pojo.Result;
import org.activiti.engine.IdentityService;
@@ -17,6 +18,7 @@

import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Service
public class IdentityServiceImpl extends BaseServiceImpl implements com.highdatas.mdm.service.act.IdentityService{
@@ -26,7 +28,8 @@
   RuntimeService runtimeService;
   @Autowired
   RepositoryService repositoryService;

   @Autowired
   TableInfoMapper tableInfoMapper;

    @Override
   public String startProcess(String businessId, String key) {
@@ -88,6 +91,10 @@

   @Override
   public Result addUser(String id) {
      User preUser = identityService.createUserQuery().userId(id).singleResult();
      if (preUser != null) {
         return  Result.error(new CodeMsg(6002, "已有相同id的用户存在"));
      }
      try {
         User user = identityService.newUser(id);
         identityService.saveUser(user);
@@ -101,6 +108,10 @@

   @Override
   public Result addRole(String id) {
      Group preGroup = identityService.createGroupQuery().groupId(id).singleResult();
      if (preGroup != null) {
         return  Result.error(new CodeMsg(6002, "已有相同id的角色存在"));
      }
      try {
         Group group = identityService.newGroup(id);
         identityService.saveGroup(group);
@@ -118,9 +129,10 @@
         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);
         }
         identityService.createMembership(userId,roleId);
         return Result.success(null);
@@ -174,6 +186,11 @@
         if (user == null) {
            return Result.error(new CodeMsg(3001, "当前角色不存在,请先角色用户"));
         }

            Map<String, String> memberShip = tableInfoMapper.selectActMemberShip(userId, roleId);
            if (memberShip == null) {
                return Result.success(null);
            }
         identityService.deleteMembership(userId,roleId);
         return Result.success(null);
      }