| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150 |
- package com.goafanti.permission.service.impl;
- import com.goafanti.common.constant.AFTConstants;
- import com.goafanti.common.dao.RoleMapper;
- import com.goafanti.common.dao.RolePermissionMapper;
- import com.goafanti.common.dao.UserMapper;
- import com.goafanti.common.dao.UserRoleMapper;
- import com.goafanti.common.model.Permission;
- import com.goafanti.common.model.Role;
- import com.goafanti.common.model.RolePermission;
- import com.goafanti.common.model.UserRole;
- import com.goafanti.core.mybatis.BaseMybatisDao;
- import com.goafanti.core.shiro.token.TokenManager;
- import com.goafanti.permission.bo.RoleBo;
- import com.goafanti.permission.bo.RolePermissionBo;
- import com.goafanti.permission.service.RoleService;
- import net.sourceforge.pinyin4j.PinyinHelper;
- import net.sourceforge.pinyin4j.format.HanyuPinyinOutputFormat;
- import net.sourceforge.pinyin4j.format.exception.BadHanyuPinyinOutputFormatCombination;
- import org.springframework.beans.factory.annotation.Autowired;
- import org.springframework.stereotype.Service;
- import java.util.*;
- @Service
- public class RoleServiceImpl extends BaseMybatisDao<RoleMapper> implements RoleService {
- @Autowired
- RoleMapper roleMapper;
- @Autowired
- UserMapper userMapper;
- @Autowired
- UserRoleMapper userRoleMapper;
- @Autowired
- RolePermissionMapper rolePermissionMapper;
- @Override
- public Set<String> findRoleByUserId(String userId) {
- return roleMapper.findRoleByUserId(userId);
- }
- @Override
- public List<RoleBo> findRoles() {
- List<RolePermissionBo> res = null;
- if (!TokenManager.hasRole(AFTConstants.SUPERADMIN)) {
- res = roleMapper.findRolePermissionsExceptAreaManager();
- } else {
- res = roleMapper.findRolePermissions();
- }
- Map<String, RoleBo> roleMap = new HashMap<>();
- for (RolePermissionBo rp : res) {
- RoleBo rb = roleMap.get(rp.getRid());
- if (rb == null) {
- rb = new RoleBo();
- rb.setId(rp.getRid());
- rb.setRoleName(rp.getRname());
- rb.setRoleType(rp.getType());
- roleMap.put(rp.getRid(), rb);
- }
- Permission p = new Permission();
- p.setId(rp.getPid());
- p.setName(rp.getPname());
- p.setUrl(rp.getUrl());
- if (p.getId() != null) {
- rb.getPermissions().add(p);
- }
- }
- ArrayList<RoleBo> roleBos = new ArrayList<>(roleMap.values());
- return listSrot(roleBos);
- }
- private List<RoleBo> listSrot(ArrayList<RoleBo> roleBos) {
- HanyuPinyinOutputFormat format=new HanyuPinyinOutputFormat();
- Collections.sort(roleBos, (o1, o2) -> {
- String str1=null;
- String str2=null;
- try {
- str1= PinyinHelper.toHanYuPinyinString(o1.getRoleName(),format," ",true);
- str2= PinyinHelper.toHanYuPinyinString(o2.getRoleName(),format," ",true);
- } catch (BadHanyuPinyinOutputFormatCombination e) {
- throw new RuntimeException(e);
- }
- return str1.compareTo(str2);
- });
- return roleBos;
- }
- @Override
- public int insert(List<Role> records) {
- return roleMapper.insertBatch(records);
- }
- @Override
- public int deleteByPrimaryKeys(List<String> records) {
- rolePermissionMapper.deleteByRoleId(records);
- userRoleMapper.deleteByRoleId(records);
- TokenManager.clearUserAuthByUserId((String[]) userRoleMapper.findUserIdByRoleIds(records).toArray());
- return roleMapper.deleteByPrimaryKeys(records);
- }
- @Override
- public int bindPermission(String rid, String pid) {
- TokenManager.clearUserAuthByUserId((String[]) userRoleMapper.findUserIdByRoleId(rid).toArray());
- return rolePermissionMapper.insert(new RolePermission(rid, pid));
- }
- @Override
- public int bindUser(String rid, String uid) {
- TokenManager.clearUserAuthByUserId(new String[] { uid });
- return userRoleMapper.insert(new UserRole(uid, rid));
- }
- @Override
- public int unbindPermission(String roleId, String permissionId) {
- Map<String, Object> params = new HashMap<>();
- params.put("rid", roleId);
- params.put("pid", permissionId);
- TokenManager.clearUserAuthByUserId((String[]) userRoleMapper.findUserIdByRoleId(roleId).toArray());
- return rolePermissionMapper.deleteByPrimaryKey(params);
- }
- @Override
- public int unbindUser(String roleId, String userId) {
- Map<String, Object> params = new HashMap<>();
- params.put("rid", roleId);
- params.put("uid", userId);
- TokenManager.clearUserAuthByUserId(new String[] { userId });
- return 0;
- }
- @Override
- public int insert(Role record, List<String> permissions) {
- if (record.getId() != null) {
- List<RolePermission> rps = new ArrayList<>();
- rolePermissionMapper.deleteByRid(record.getId());
- for (String pid : permissions) {
- rps.add(new RolePermission(record.getId(), pid));
- }
- if (!rps.isEmpty()) {
- rolePermissionMapper.insertBatch(rps);
- }
- List<String> ids = userRoleMapper.findUserIdByRoleId(record.getId());
- TokenManager.clearUserAuthByUserId(ids.toArray(new String[ids.size()]));
- }
- record.getId();
- return roleMapper.insert(record);
- }
- }
|