一、获取全部菜单

image.png

1、controller

  1. //获取全部菜单
  2. @ApiOperation(value = "查询所有菜单")
  3. @GetMapping
  4. public R indexAllPermission() {
  6.     List<Permission> list =  permissionService.queryAllMenuGuli();
  7.     return R.ok().data("children",list);
  8. }

2、serviceImpl

  1. //获取全部菜单
  2. @Override
  3. public List<Permission> queryAllMenuGuli() {
  5.     //1 查询菜单表所有数据
  6.     LambdaQueryWrapper<Permission> wrapper = new LambdaQueryWrapper<>();
  7.     wrapper.orderByDesc(Permission::getId);
  8.     List<Permission> permissionList = baseMapper.selectList(wrapper);
  10.     //2 把查询所有菜单list集合按照要求进行封装
  11.     List<Permission> resultList = bulidPermission(permissionList);
  12.     return resultList;
  13. }
  15. // 找到递归调用的起始节点：即顶层菜单
  16. //把返回所有菜单list集合进行封装的方法
  17. public static List<Permission> bulidPermission(List<Permission> permissionList) {
  19.     //创建list集合，用于数据最终封装
  20.     List<Permission> finalNode = new ArrayList<>();
  21.     //把所有菜单list集合遍历，得到顶层菜单 pid=0菜单，设置level是1
  22.     permissionList.forEach(e -> {
  23.         //得到顶层菜单 pid=0菜单
  24.         if("0".equals(e.getPid())){
  25.             //设置顶层菜单的level是1
  26.             e.setLevel(1);
  27.             //根据顶层菜单，向里面进行查询子菜单，封装到finalNode里面
  28.             finalNode.add(selectChildren(e,permissionList));
  29.         }
  30.     });
  31.     return finalNode;
  32. }
  34. /**
  35.  *
  36.  * @param permissionNode    父菜单
  37.  * @param permissionList    所有的菜单
  38.  * @return
  39.  */
  40. private static Permission selectChildren(Permission permissionNode, List<Permission> permissionList) {
  42.     //1 因为向一层菜单里面放二层菜单，二层里面还要放三层，把子菜单对象初始化
  43.     permissionNode.setChildren(new ArrayList<Permission>());
  45.     //2 遍历所有菜单list集合，进行判断比较，比较父菜单id和子菜单pid值是否相同
  46.     permissionList.forEach(permission -> {
  47.         //判断父菜单id和子菜单pid值是否相同
  48.         if(permission.getPid().equals(permissionNode.getId())){
  49.             //把父菜单的level值+1，赋值给子菜单
  50.             int level = permissionNode.getLevel() + 1;
  51.             permission.setLevel(level);
  52.             //如果children为空，进行初始化操作
  53.             if(permission.getChildren() == null){
  54.                 permission.setChildren(new ArrayList<Permission>());
  55.             }
  56.             //把查询出来的子菜单放到父菜单里面
  57.             permissionNode.getChildren().add(selectChildren(permission,permissionList));
  58. //                permissionNode.setChildren(new ArrayList<Permission>(){
  59. //                        {
  60. //                            add(selectChildren(permission,permissionList));
  61. //                        }
  62. //                });
  63.         }
  64.     });
  65.     return permissionNode;
  66. }

二、递归删除菜单

1、controller

  1. @ApiOperation(value = "递归删除菜单")
  2. @DeleteMapping("remove/{id}")
  3. public R remove(@PathVariable String id) {
  5.     permissionService.removeChildByIdGuli(id);
  6.     return R.ok();
  7. }

2、serviceImpl

  1. @Override
  2. public void removeChildByIdGuli(String id) {
  4.     //1 创建list集合，用于封装所有删除菜单id值
  5.     List<String> idList = new ArrayList<>();
  6.     //2 向idList集合设置删除菜单id
  7.     this.selectPermissionChildById(id,idList);
  8.     //把当前id封装到list里面
  9.     idList.add(id);
  10.     //执行删除操作
  11.     baseMapper.deleteBatchIds(idList);
  12. }
  15. //2 根据当前菜单id，查询菜单里面子菜单id，封装到list集合
  16. private void selectPermissionChildById(String id, List<String> idList) {
  18.     //查询菜单里面子菜单id
  19.     LambdaQueryWrapper<Permission> wrapper = new LambdaQueryWrapper<>();
  20.     wrapper.eq(Permission::getPid,id);
  21.     List<Permission> childIdList = baseMapper.selectList(wrapper);
  23.     if(!CollectionUtils.isEmpty(childIdList)){
  24.         //把childIdList里面菜单id值获取出来，封装idList里面，做递归查询
  25.         childIdList.stream().forEach(item -> {
  26.             //封装idList里面
  27.             idList.add(item.getId());
  28.             //递归查询
  29.             this.selectPermissionChildById(item.getId(), idList);
  30.         });
  31.     }
  32. }

三、给角色分配菜单

1、controller

  1. @ApiOperation(value = "给角色分配菜单")
  2. @PostMapping("/doAssign")
  3. public R doAssign(String roleId,String[] permissionId) {
  5.     permissionService.saveRolePermissionRealtionShipGuli(roleId,permissionId);
  6.     return R.ok();
  7. }

2、serviceImpl

  1. @Override
  2. public void saveRolePermissionRealtionShipGuli(String roleId, String[] permissionIds) {
  4.     //roleId角色id
  5.     //permissionId菜单id 数组形式
  6.     // 删除已分配的菜单
  7.     LambdaQueryWrapper<RolePermission> wrapper = new LambdaQueryWrapper<>();
  8.     wrapper.eq(RolePermission::getRoleId,roleId);
  9.     rolePermissionService.remove(wrapper);
  11.     // 创建list集合，用于封装添加数据
  12.     List<RolePermission> rolePermissionList = new ArrayList<>();
  14.     //遍历所有菜单数组
  15.     for(String perId : permissionIds) {
  17.         //RolePermission对象
  18.         RolePermission rolePermission = new RolePermission();
  19.         rolePermission.setRoleId(roleId);
  20.         rolePermission.setPermissionId(perId);
  21.         //封装到list集合
  22.         rolePermissionList.add(rolePermission);
  23.     }
  24.     //添加到角色菜单关系表
  25.     rolePermissionService.saveBatch(rolePermissionList);
  26. }

四、根据角色获取菜单

1、controller

  1. @ApiOperation(value = "根据角色获取菜单")
  2. @GetMapping("toAssign/{roleId}")
  3. public R toAssign(@PathVariable String roleId) {
  4.     List<Permission> list = permissionService.selectAllMenu(roleId);
  5.     return R.ok().data("children", list);
  6. }

2、serviceImpl

  1. //根据角色获取菜单
  2. @Override
  3. public List<Permission> selectAllMenu(String roleId) {
  5.     //获取所有菜单
  6.     // CAST(id AS SIGNED)，将id转换成int类型
  7.     // List<Permission> allPermissionList = baseMapper.selectList(new QueryWrapper<Permission>().orderByAsc("CAST(id AS SIGNED)"));
  8.     LambdaQueryWrapper<Permission> wrapper = new LambdaQueryWrapper<>();
  9.     wrapper.orderByDesc(Permission::getId);
  10.     List<Permission> allPermissionList = baseMapper.selectList(wrapper);
  12.     // 根据角色id获取角色权限
  13.     // List<RolePermission> rolePermissionList = rolePermissionService.list(new QueryWrapper<RolePermission>().eq("role_id",roleId));
  14.     LambdaQueryWrapper<RolePermission> wrapper1 = new LambdaQueryWrapper<>();
  15.     wrapper1.eq(RolePermission::getRoleId,roleId);
  16.     List<RolePermission> rolePermissionList = rolePermissionService.list(wrapper1);
  18.     // 获取角色对应的权限id
  19.     List<String> permissionIdList = rolePermissionList.stream().map(RolePermission::getPermissionId).collect(Collectors.toList());
  20.     // 设置权限列表的select字段
  21.     allPermissionList.forEach(p -> {
  22.         if(permissionIdList.contains(p.getId())){
  23.             p.setSelect(true);
  24.         }else {
  25.             p.setSelect(false);
  26.         }
  27.     });
  29.     //转换给角色id与角色权限对应Map对象
  30. //        List<String> permissionIdList = rolePermissionList.stream().map(e -> e.getPermissionId()).collect(Collectors.toList());
  31. //        allPermissionList.forEach(permission -> {
  32. //            if(permissionIdList.contains(permission.getId())) {
  33. //                permission.setSelect(true);
  34. //            } else {
  35. //                permission.setSelect(false);
  36. //            }
  37. //        });
  39. //        for (int i = 0; i < allPermissionList.size(); i++) {
  40. //            Permission permission = allPermissionList.get(i);
  41. //            for (int m = 0; m < rolePermissionList.size(); m++) {
  42. //                RolePermission rolePermission = rolePermissionList.get(m);
  43. //                if(rolePermission.getPermissionId().equals(permission.getId())) {
  44. //                    permission.setSelect(true);
  45. //                }
  46. //            }
  47. //        }
  49.     List<Permission> permissionList = bulid(allPermissionList);
  50.     return permissionList;
  51. }
  54. /**
  55.  * 使用递归方法建菜单
  56.  * @param treeNodes
  57.  * @return
  58.  */
  59. private static List<Permission> bulid(List<Permission> treeNodes) {
  61.     List<Permission> trees = new ArrayList<>();
  62.     for (Permission treeNode : treeNodes) {
  63.         if ("0".equals(treeNode.getPid())) {
  64.             treeNode.setLevel(1);
  65.             trees.add(findChildren(treeNode,treeNodes));
  66.         }
  67.     }
  68.     return trees;
  69. }
  72. /**
  73.  * 递归查找子节点
  74.  * @param treeNodes
  75.  * @return
  76.  */
  77. private static Permission findChildren(Permission treeNode,List<Permission> treeNodes) {
  79.     treeNode.setChildren(new ArrayList<Permission>());
  81.     for (Permission it : treeNodes) {
  82.         if(treeNode.getId().equals(it.getPid())) {
  83.             int level = treeNode.getLevel() + 1;
  84.             it.setLevel(level);
  85.             if (treeNode.getChildren() == null) {
  86.                 treeNode.setChildren(new ArrayList<>());
  87.             }
  88.             treeNode.getChildren().add(findChildren(it,treeNodes));
  89.         }
  90.     }
  91.     return treeNode;
  92. }

五、根据用户id获取用户菜单

1、PermissionService

  1. public interface PermissionService extends IService<Permission> {
  3.     //根据用户id获取用户菜单
  4.     List<String> selectPermissionValueByUserId(String id);
  5. }

2、PermissionServiceImpl

  1. //根据用户id获取用户菜单
  2. @Override
  3. public List<String> selectPermissionValueByUserId(String id) {
  5.     List<String> selectPermissionValueList = null;
  6.     if(this.isSysAdmin(id)) {
  7.         // 如果是系统管理员，获取所有权限
  8.         selectPermissionValueList = baseMapper.selectAllPermissionValue();
  9.     } else {
  10.         selectPermissionValueList = baseMapper.selectPermissionValueByUserId(id);
  11.     }
  12.     return selectPermissionValueList;
  13. }

3、PermissionMapper

  1. public interface PermissionMapper extends BaseMapper<Permission> {
  3.     // 系统管理员，获取所有权限
  4.     List<String> selectAllPermissionValue();
  6.     // 非系统管理员，根据用户id获取用户菜单
  7.     List<String> selectPermissionValueByUserId(String id);
  8. }

4、PermissionMapper.xml

  1. <?xml version="1.0" encoding="UTF-8"?>
  2. <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
  3. <mapper namespace="com.atguigu.aclservice.mapper.PermissionMapper">
  4.     <resultMap id="permissionMap" type="com.atguigu.aclservice.entity.Permission">
  5.         <result property="id" column="id"/>
  6.         <result property="pid" column="pid"/>
  7.         <result property="name" column="name"/>
  8.         <result property="type" column="type"/>
  9.         <result property="permissionValue" column="permission_value"/>
  10.         <result property="path" column="path"/>
  11.         <result property="component" column="component"/>
  12.         <result property="icon" column="icon"/>
  13.         <result property="status" column="status"/>
  14.         <result property="isDeleted" column="is_deleted"/>
  15.         <result property="gmtCreate" column="gmt_create"/>
  16.         <result property="gmtModified" column="gmt_modified"/>
  17.     </resultMap>
  19.     <!-- 用于select查询公用抽取的列 -->
  20.     <sql id="columns">
  21.         p.id,p.pid,p.name,p.type,p.permission_value,path,p.component,p.icon,p.status,p.is_deleted,p.gmt_create,p.gmt_modified
  22.     </sql>
  24.     // 非系统管理员，根据用户id获取用户菜单
  25.     <select id="selectPermissionValueByUserId" resultType="String">
  26.         select
  27.         p.permission_value
  28.         from acl_user_role ur
  29.         inner join acl_role_permission rp on rp.role_id = ur.role_id
  30.         inner join acl_permission p on p.id = rp.permission_id
  31.         where ur.user_id = #{userId}
  32.         and p.type = 2
  33.         and ur.is_deleted = 0
  34.         and rp.is_deleted = 0
  35.         and p.is_deleted = 0
  36.     </select>
  38.     // 系统管理员，获取所有权限
  39.     <select id="selectAllPermissionValue" resultType="String">
  40.         select
  41.             permission_value
  42.         from
  43.              acl_permission
  44.         where
  45.               type = 2
  46.               and is_deleted = 0
  47.     </select>
  48. </mapper>

5、IndexServiceImpl

  1. /**
  2.  * 根据用户名获取用户登录信息
  3.  *
  4.  * @param username
  5.  * @return
  6.  */
  7. public Map<String, Object> getUserInfo(String username) {
  9.     Map<String, Object> result = new HashMap<>();
  10.     User user = userService.selectByUsername(username);
  11.     if (null == user) {
  12.         //throw new GuliException(ResultCodeEnum.FETCH_USERINFO_ERROR);
  13.     }
  15.     //根据用户id获取角色
  16.     List<Role> roleList = roleService.selectRoleByUserId(user.getId());
  17.     List<String> roleNameList = roleList.stream().map(item -> item.getRoleName()).collect(Collectors.toList());
  18.     if(roleNameList.size() == 0) {
  19.         //前端框架必须返回一个角色，否则报错，如果没有角色，返回一个空角色
  20.         roleNameList.add("");
  21.     }
  23.     //根据用户id获取操作权限值
  24.     List<String> permissionValueList = permissionService.selectPermissionValueByUserId(user.getId());
  25.     redisTemplate.opsForValue().set(username, permissionValueList);
  27.     result.put("name", user.getUsername());
  28.     result.put("avatar", "https://wpimg.wallstcn.com/f778738c-e4f8-4870-b634-56703b4acafe.gif");
  29.     result.put("roles", roleNameList);
  30.     result.put("permissionValueList", permissionValueList);
  31.     return result;
  32. }

6、IndexController

  1. /**
  2.  * 根据token获取用户信息
  3.  */
  4. @GetMapping("info")
  5. public R info(){
  7.     //获取当前登录用户用户名
  8.     String username = SecurityContextHolder.getContext().getAuthentication().getName();
  9.     Map<String, Object> userInfo = indexService.getUserInfo(username);
  10.     return R.ok().data(userInfo);
  11. }

六、根据用户id获取用户菜单权限

1、PermissionService

  1. public interface PermissionService extends IService<Permission> {
  3.     //根据用户id获取用户菜单权限
  4.     List<JSONObject> selectPermissionByUserId(String id);
  5. }

2、PermissionServiceImpl

  1. /**
  2.  * 根据用户id获取用户菜单权限
  3.  * @param userId
  4.  * @return
  5.  */
  6. @Override
  7. public List<JSONObject> selectPermissionByUserId(String userId) {
  9.     List<Permission> selectPermissionList = null;
  10.     if(this.isSysAdmin(userId)) {
  11.         // 如果是超级管理员，获取所有菜单
  12.         selectPermissionList = baseMapper.selectList(null);
  13.     } else {
  14.         selectPermissionList = baseMapper.selectPermissionByUserId(userId);
  15.     }
  17.     // 根据权限数据构建菜单数据
  18.     List<Permission> permissionList = PermissionHelper.bulid(selectPermissionList);
  20.     // 根据权限数据构建登录用户左侧菜单数据
  21.     List<JSONObject> result = MemuHelper.bulid(permissionList);
  22.     return result;
  23. }

1）PermissionHelper

  1. /**
  2.  * 根据权限数据构建菜单数据
  3.  */
  4. public class PermissionHelper {
  6.     /**
  7.      * 使用递归方法建菜单
  8.      * @param treeNodes
  9.      * @return
  10.      */
  11.     public static List<Permission> bulid(List<Permission> treeNodes) {
  12.         List<Permission> trees = new ArrayList<>();
  13.         for (Permission treeNode : treeNodes) {
  14.             if ("0".equals(treeNode.getPid())) {
  15.                 treeNode.setLevel(1);
  16.                 trees.add(findChildren(treeNode,treeNodes));
  17.             }
  18.         }
  19.         return trees;
  20.     }
  22.     /**
  23.      * 递归查找子节点
  24.      * @param treeNodes
  25.      * @return
  26.      */
  27.     public static Permission findChildren(Permission treeNode,List<Permission> treeNodes) {
  28.         treeNode.setChildren(new ArrayList<Permission>());
  30.         for (Permission it : treeNodes) {
  31.             if(treeNode.getId().equals(it.getPid())) {
  32.                 int level = treeNode.getLevel() + 1;
  33.                 it.setLevel(level);
  34.                 if (treeNode.getChildren() == null) {
  35.                     treeNode.setChildren(new ArrayList<>());
  36.                 }
  37.                 treeNode.getChildren().add(findChildren(it,treeNodes));
  38.             }
  39.         }
  40.         return treeNode;
  41.     }
  42. }

2）MemuHelper

  1. /**
  2.  * 根据权限数据构建登录用户左侧菜单数据
  3.  */
  4. public class MemuHelper {
  6.     /**
  7.      * 构建菜单
  8.      * @param treeNodes
  9.      * @return
  10.      */
  11.     public static List<JSONObject> bulid(List<Permission> treeNodes) {
  13.         List<JSONObject> meuns = new ArrayList<>();
  14.         if(treeNodes.size() == 1) {
  15.             Permission topNode = treeNodes.get(0);
  16.             //左侧一级菜单
  17.             List<Permission> oneMeunList = topNode.getChildren();
  18.             for(Permission one :oneMeunList) {
  19.                 JSONObject oneMeun = new JSONObject();
  20.                 oneMeun.put("path", one.getPath());
  21.                 oneMeun.put("component", one.getComponent());
  22.                 oneMeun.put("redirect", "noredirect");
  23.                 oneMeun.put("name", "name_"+one.getId());
  24.                 oneMeun.put("hidden", false);
  26.                 JSONObject oneMeta = new JSONObject();
  27.                 oneMeta.put("title", one.getName());
  28.                 oneMeta.put("icon", one.getIcon());
  29.                 oneMeun.put("meta", oneMeta);
  31.                 List<JSONObject> children = new ArrayList<>();
  32.                 List<Permission> twoMeunList = one.getChildren();
  33.                 for(Permission two :twoMeunList) {
  34.                     JSONObject twoMeun = new JSONObject();
  35.                     twoMeun.put("path", two.getPath());
  36.                     twoMeun.put("component", two.getComponent());
  37.                     twoMeun.put("name", "name_"+two.getId());
  38.                     twoMeun.put("hidden", false);
  40.                     JSONObject twoMeta = new JSONObject();
  41.                     twoMeta.put("title", two.getName());
  42.                     twoMeun.put("meta", twoMeta);
  44.                     children.add(twoMeun);
  46.                     List<Permission> threeMeunList = two.getChildren();
  47.                     for(Permission three :threeMeunList) {
  48.                         if(StringUtils.isEmpty(three.getPath())) continue;
  50.                         JSONObject threeMeun = new JSONObject();
  51.                         threeMeun.put("path", three.getPath());
  52.                         threeMeun.put("component", three.getComponent());
  53.                         threeMeun.put("name", "name_"+three.getId());
  54.                         threeMeun.put("hidden", true);
  56.                         JSONObject threeMeta = new JSONObject();
  57.                         threeMeta.put("title", three.getName());
  58.                         threeMeun.put("meta", threeMeta);
  60.                         children.add(threeMeun);
  61.                     }
  62.                 }
  63.                 oneMeun.put("children", children);
  64.                 meuns.add(oneMeun);
  65.             }
  66.         }
  67.         return meuns;
  68.     }
  69. }

3、PermissionMapper

  1. public interface PermissionMapper extends BaseMapper<Permission> {
  3.     // 非超级管理员，根据用户id获取用户菜单权限
  4.     List<Permission> selectPermissionByUserId(String userId);
  5. }

4、PermissionMapper.xml

  1. <?xml version="1.0" encoding="UTF-8"?>
  2. <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
  3. <mapper namespace="com.atguigu.aclservice.mapper.PermissionMapper">
  4.     <resultMap id="permissionMap" type="com.atguigu.aclservice.entity.Permission">
  5.         <result property="id" column="id"/>
  6.         <result property="pid" column="pid"/>
  7.         <result property="name" column="name"/>
  8.         <result property="type" column="type"/>
  9.         <result property="permissionValue" column="permission_value"/>
  10.         <result property="path" column="path"/>
  11.         <result property="component" column="component"/>
  12.         <result property="icon" column="icon"/>
  13.         <result property="status" column="status"/>
  14.         <result property="isDeleted" column="is_deleted"/>
  15.         <result property="gmtCreate" column="gmt_create"/>
  16.         <result property="gmtModified" column="gmt_modified"/>
  17.     </resultMap>
  19.     <!-- 用于select查询公用抽取的列 -->
  20.     <sql id="columns">
  21.         p.id,p.pid,p.name,p.type,p.permission_value,path,p.component,p.icon,p.status,p.is_deleted,p.gmt_create,p.gmt_modified
  22.     </sql>
  24.     // 非超级管理员，根据用户id获取用户菜单权限
  25.     <select id="selectPermissionByUserId" resultMap="permissionMap">
  26.         select
  27.         <include refid="columns" />
  28.         from acl_user_role ur
  29.         inner join acl_role_permission rp on rp.role_id = ur.role_id
  30.         inner join acl_permission p on p.id = rp.permission_id
  31.         where ur.user_id = #{userId}
  32.         and ur.is_deleted = 0
  33.         and rp.is_deleted = 0
  34.         and p.is_deleted = 0
  35.     </select>
  36. </mapper>

5、IndexServiceImpl

  1. /**
  2.  * 根据用户名获取动态菜单
  3.  * @param username
  4.  * @return
  5.  */
  6. public List<JSONObject> getMenu(String username) {
  7.     User user = userService.selectByUsername(username);
  9.     //根据用户id获取用户菜单权限
  10.     List<JSONObject> permissionList = permissionService.selectPermissionByUserId(user.getId());
  11.     return permissionList;
  12. }

6、IndexController

  1. /**
  2.  * 获取菜单
  3.  * @return
  4.  */
  5. @GetMapping("menu")
  6. public R getMenu(){
  8.     //获取当前登录用户用户名
  9.     String username = SecurityContextHolder.getContext().getAuthentication().getName();
  10.     List<JSONObject> permissionList = indexService.getMenu(username);
  11.     return R.ok().data("permissionList", permissionList);
  12. }