springBoot基于角色实现动态路由

分手后的思念是犯贱 2022-12-01 12:27 228阅读 0赞

# 1. spring boot 基于角色实现动态路由 #

## 1. 创建实体类 ##

@Data
    @Table(name = "tb_role")
    @NoArgsConstructor
    @AllArgsConstructor
    public class Role {
        @Id
        @KeySql(useGeneratedKeys = true)
        private Long roleId;
        @Column(name = "role_name")
        private String roleName;
        @Column(name = "role_desc")
        private String roleDes;
        private List<Permission> permissions;
        private List<Menu> menus;
    }
    
    @Data
    @Table(name = "tb_menu")
    public class Menu {
        @Id
        @Column(name = "menu_id")
        @KeySql(useGeneratedKeys = true)
        private Long menuId;
        @Column(name = "menu_name")
        private String menuName;
        private String url;//路由
        private String icon;//图标
        @Column(name = "order_num")
        private Long orderNum;//用于路由排序
        private Boolean type;// 0菜单，可以展开  1按钮不可展开
        private Boolean available;//0不可用 1可用
        @Column(name = "parent_id")
        private Long parentId; //上级菜单id 0一级菜单
        @Column(name = "modify_time")
        private Date modifyTime;//修改时间
        private Boolean authority;//是否需要权限
        private List<Menu> menus;//子菜单
    }

## 2. 创建mapper接口 ##

public interface MenuMapper extends Mapper<Menu> {
    
        /**
         * 根据roleId获取所有菜单信息
         *
         * @param roleId
         * @return
         */
        @Select("select m.* from tb_menu m left join tb_role_menu rm on rm.menu_id = m.menu_id where rm.role_id=#{roleId} and m.type=1")
        List<Menu> queryMenusByRoleId(Long roleId);
    
        /**
         * 根据parentId与是否需要权限查询
         *
         * @param parentId
         * @param authority
         * @return
         */
        @Select("select * from tb_menu where parent_id = #{parentId} and authority=#{authority}")
        List<Menu> queryMenusByParentId(@Param("parentId") Long parentId, @Param("authority") Boolean authority);
    
        /**
         * 根据是否需要权限与类型查询
         *
         * @param authority
         * @param type
         * @return
         */
        @Select("select * from tb_menu where authority=#{authority} and type=#{type}")
        List<Menu> queryMenusByAuthority(@Param("authority") Boolean authority, @Param("type") Boolean type);
    }

## 3. 创建service ##

/**
         * 根据用户信息获取对应的菜单信息
         *
         * @return
         */
        public List<Menu> queryMenuByRole() {
            //从线程域中获取用户信息
            UserInfo userInfo = UserInfoContext.getUserInfo();
            if (userInfo == null) {
                return null;
            }
            //获取无需权限的菜单
            List<Menu> noAuthorityMenus = menuMapper.queryMenusByAuthority(false, true);
            for (Menu noAuthorityMenu : noAuthorityMenus) {
                List<Menu> menus = menuMapper.queryMenusByParentId(noAuthorityMenu.getMenuId(), false);
                if (!CollectionUtils.isEmpty(menus)) {
                    noAuthorityMenu.setMenus(menus);
                }
            }
            //根据roleId获取对应的菜单
            List<Long> roleIds = userInfo.getRoles().stream().map(Role::getRoleId).collect(Collectors.toList());
            for (Long roleId : roleIds) {
                List<Menu> menus = menuMapper.queryMenusByRoleId(roleId);
                if (!CollectionUtils.isEmpty(menus)) {
                    for (Menu menu : menus) {
                        //根据parentId查询所有菜单下的子菜单
                        List<Menu> childMenu = menuMapper.queryMenusByParentId(menu.getMenuId(), true);
                        menu.setMenus(childMenu);
                    }
                    noAuthorityMenus.addAll(menus);
                    //根据orderNum字段进行排序操作
                    noAuthorityMenus.sort((menu1, menu2) -> {
                        Long orderNum1 = menu1.getOrderNum();
                        Long orderNum2 = menu2.getOrderNum();
                        return orderNum1.compareTo(orderNum2);
                    });
                }
            }
            return noAuthorityMenus;
        }

## 4. Controller层 ##

@GetMapping("/menus")
        public Result queryMenuByRole() {
            List<Menu> menus = userInfoService.queryMenuByRole();
            if (CollectionUtils.isEmpty(menus)) {
                return new Result(false, StatusCode.MENU_NOT_FOUND);
            }
            return new Result(true, StatusCode.OK, menus);
        }

### 想要源码，扫码关注下方公众号，留言即可立即获取相应的资源 ###

![Alt][]

[Alt]: /images/20221123/56d5d8a5eb0e465fb73c01fbe82e5a9b.png