RoleServiceImpl.java 4.7 KB

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