RoleServiceImpl.java 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136
  1. package com.goafanti.permission.service.impl;
  2. import java.util.ArrayList;
  3. import java.util.HashMap;
  4. import java.util.List;
  5. import java.util.Map;
  6. import java.util.Set;
  7. import org.springframework.beans.factory.annotation.Autowired;
  8. import org.springframework.stereotype.Service;
  9. import com.goafanti.common.constant.AFTConstants;
  10. import com.goafanti.common.dao.RoleMapper;
  11. import com.goafanti.common.dao.RolePermissionMapper;
  12. import com.goafanti.common.dao.UserMapper;
  13. import com.goafanti.common.dao.UserRoleMapper;
  14. import com.goafanti.common.model.Permission;
  15. import com.goafanti.common.model.Role;
  16. import com.goafanti.common.model.RolePermission;
  17. import com.goafanti.common.model.UserRole;
  18. import com.goafanti.core.mybatis.BaseMybatisDao;
  19. import com.goafanti.core.shiro.token.TokenManager;
  20. import com.goafanti.permission.bo.RoleBo;
  21. import com.goafanti.permission.bo.RolePermissionBo;
  22. import com.goafanti.permission.service.RoleService;
  23. @Service
  24. public class RoleServiceImpl extends BaseMybatisDao<RoleMapper> implements RoleService {
  25. @Autowired
  26. RoleMapper roleMapper;
  27. @Autowired
  28. UserMapper userMapper;
  29. @Autowired
  30. UserRoleMapper userRoleMapper;
  31. @Autowired
  32. RolePermissionMapper rolePermissionMapper;
  33. @Override
  34. public Set<String> findRoleByUserId(String userId) {
  35. return roleMapper.findRoleByUserId(userId);
  36. }
  37. @Override
  38. public List<RoleBo> findRoles() {
  39. List<RolePermissionBo> res = null;
  40. if (!TokenManager.hasRole(AFTConstants.SUPERADMIN)) {
  41. res = roleMapper.findRolePermissionsExceptAreaManager();
  42. } else {
  43. res = roleMapper.findRolePermissions();
  44. }
  45. Map<String, RoleBo> roleMap = new HashMap<>();
  46. for (RolePermissionBo rp : res) {
  47. RoleBo rb = roleMap.get(rp.getRid());
  48. if (rb == null) {
  49. rb = new RoleBo();
  50. rb.setId(rp.getRid());
  51. rb.setRoleName(rp.getRname());
  52. rb.setRoleType(rp.getType());
  53. roleMap.put(rp.getRid(), rb);
  54. }
  55. Permission p = new Permission();
  56. p.setId(rp.getPid());
  57. p.setName(rp.getPname());
  58. p.setUrl(rp.getUrl());
  59. if (p.getId() != null) {
  60. rb.getPermissions().add(p);
  61. }
  62. }
  63. return new ArrayList<>(roleMap.values());
  64. }
  65. @Override
  66. public int insert(List<Role> records) {
  67. return roleMapper.insertBatch(records);
  68. }
  69. @Override
  70. public int deleteByPrimaryKeys(List<String> records) {
  71. rolePermissionMapper.deleteByRoleId(records);
  72. userRoleMapper.deleteByRoleId(records);
  73. TokenManager.clearUserAuthByUserId((String[]) userRoleMapper.findUserIdByRoleIds(records).toArray());
  74. return roleMapper.deleteByPrimaryKeys(records);
  75. }
  76. @Override
  77. public int bindPermission(String rid, String pid) {
  78. TokenManager.clearUserAuthByUserId((String[]) userRoleMapper.findUserIdByRoleId(rid).toArray());
  79. return rolePermissionMapper.insert(new RolePermission(rid, pid));
  80. }
  81. @Override
  82. public int bindUser(String rid, String uid) {
  83. TokenManager.clearUserAuthByUserId(new String[] { uid });
  84. return userRoleMapper.insert(new UserRole(uid, rid));
  85. }
  86. @Override
  87. public int unbindPermission(String roleId, String permissionId) {
  88. Map<String, Object> params = new HashMap<>();
  89. params.put("rid", roleId);
  90. params.put("pid", permissionId);
  91. TokenManager.clearUserAuthByUserId((String[]) userRoleMapper.findUserIdByRoleId(roleId).toArray());
  92. return rolePermissionMapper.deleteByPrimaryKey(params);
  93. }
  94. @Override
  95. public int unbindUser(String roleId, String userId) {
  96. Map<String, Object> params = new HashMap<>();
  97. params.put("rid", roleId);
  98. params.put("uid", userId);
  99. TokenManager.clearUserAuthByUserId(new String[] { userId });
  100. return 0;
  101. }
  102. @Override
  103. public int insert(Role record, List<String> permissions) {
  104. if (record.getId() != null) {
  105. List<RolePermission> rps = new ArrayList<>();
  106. rolePermissionMapper.deleteByRid(record.getId());
  107. for (String pid : permissions) {
  108. rps.add(new RolePermission(record.getId(), pid));
  109. }
  110. if (!rps.isEmpty()) {
  111. rolePermissionMapper.insertBatch(rps);
  112. }
  113. List<String> ids = userRoleMapper.findUserIdByRoleId(record.getId());
  114. TokenManager.clearUserAuthByUserId(ids.toArray(new String[ids.size()]));
  115. }
  116. record.getId();
  117. return roleMapper.insert(record);
  118. }
  119. }