【爆肝推荐】手摸手带你做后台管理项目(第四章)整合redis添加shiro权限

前情提要

这一篇是关于动态权限和动态目录的,
shiro的授权器在碰到权限的校验时候才会去触发,这个时候就可以从数据库中获取到用户关联的角色，
角色绑定的权限,大概就如下图了

有兴趣可以了解一下RBAC，大概就是如下的一个关系

动态目录就更简单了，用户关联的角色，角色所拥有的目录，这个就是展示的目录了，修改数据库数据就可达到动态的目的。

正文开始

设计五个表，管理员表(也就是用户表，已存在) 、角色表、目录表、用户角色表、角色目录表，如果没懂这些关联，可以看一下图片，图片最下方标记了关系，希望能看懂

创建数据库sys_menu

创建sys_role

创建sys_role_menu

创建sys_user_role

记得创建对应的controller和service和dao以及xml、entity

太多了，就不一一创建展示了，记得service和dao集成mybatis plus提供的类

修改上篇没动的授权器UserRealm

  @Autowired
    private SysMenuService menuService;


  
    @Override
   	protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principalCollection) {
        System.out.println("这里是授权");
        UserEntity userEntity = (UserEntity) principalCollection.getPrimaryPrincipal();
        Integer id = userEntity.getId();
        //设置id为空则拥有所有权限,sql中设置了id为空则查询所有权限
        List menuList = menuService.findByUserId(id);
        //转存set是为了去重,保证权限唯一
        Set collect = menuList.stream().map(SysMenuEntity::getPerms).collect(Collectors.toSet());
        //所有权限
        Set perms = new HashSet<>();
        collect.stream().forEach(y -> {
            //防止空的造成异常
            if(!StringUtils.isEmpty(y)){
                //存放无论是否有多个或者单个,直接变成数组,更加清晰
                
                perms.addAll(Arrays.asList(y.split(",")));
            }

        });
        SimpleAuthorizationInfo simpleAuthorizationInfo = new SimpleAuthorizationInfo();
        //此处是放入权限中，不是role角色中
        simpleAuthorizationInfo.setStringPermissions(perms);
        return simpleAuthorizationInfo;
    }

SysMenuService

	//用户查询关联的目录
 List findByUserId(Integer userId);

SysMenuServiceImpl

 @Autowired
    private SysMenuDao menuDao;

    @Override
    public List findByUserId(Integer userId) {
        return menuDao.selectByUserId(userId);
    }

SysMenuDao

 List selectByUserId(@Param("userId") Integer userId);

SysMenuDao.xml

@GetMapping("type/{type}") public Result typeList(@PathVariable Integer type){ QueryWrapper wrapper = new QueryWrapper<>(); wrapper.eq("type",type); wrapper.eq("is_del","0"); List list = menuService.list(wrapper); return Result.success(list); }

角色名称

关联权限

import com.macro.Vo.PageEntity; import com.macro.entity.SysRoleEntity; import com.macro.service.SysRoleService; import com.macro.utils.Result; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.web.bind.annotation.*; @RestController @RequestMapping("role") public class SysRoleController { @Autowired private SysRoleService roleService; @GetMapping("list") public Result list(PageEntity param){ Result result = roleService.findList(param); return result; } @PostMapping("save") public Result save(@RequestBody SysRoleEntity role){ int num = roleService.insert(role); return num > 0?Result.success():Result.error("添加失败"); } @PostMapping("update") public Result update(@RequestBody SysRoleEntity role){ int num = roleService.updateEntity(role); return num > 0?Result.success():Result.error("更新失败"); } @GetMapping("info/{id}") public Result info(@PathVariable Integer id){ SysRoleEntity role = roleService.findById(id); return Result.success(role); } @PostMapping("del") public Result update(@RequestBody String[] ids){ roleService.delIds(ids); return Result.success(); } }

import com.baomidou.mybatisplus.extension.service.IService; import com.macro.Vo.PageEntity; import com.macro.entity.SysRoleEntity; import com.macro.utils.Result; public interface SysRoleService extends IService { int updateEntity(SysRoleEntity role); Result findList(PageEntity param); int insert(SysRoleEntity role); SysRoleEntity findById(Integer id); void delIds(String[] ids); }

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.github.pagehelper.PageHelper; import com.github.pagehelper.PageInfo; import com.macro.Vo.PageEntity; import com.macro.dao.SysRoleDao; import com.macro.entity.SysMenuEntity; import com.macro.entity.SysRoleEntity; import com.macro.entity.SysRoleMenuEntity; import com.macro.service.SysMenuService; import com.macro.service.SysRoleMenuService; import com.macro.service.SysRoleService; import com.macro.utils.Result; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import java.util.*; import java.util.stream.Collectors; @Service public class SysRoleServiceImpl extends ServiceImpl implements SysRoleService { @Autowired private SysRoleDao roleDao; @Autowired private SysRoleMenuService roleMenuService; @Autowired private SysMenuService menuService; @Override public int updateEntity(SysRoleEntity role) { int update = roleDao.updateById(role); QueryWrapper wrapper = new QueryWrapper<>(); wrapper.eq("role_id",role.getId()); //旧的所有关联菜单 List list = roleMenuService.list(wrapper); //新的关联菜单 List menuList = role.getMenuList(); if(update > 0 && menuList != null && menuList.size() > 0){ //抽取旧的目录id Set arr = list.stream().map(SysRoleMenuEntity::getMenuId).collect(Collectors.toSet()); //新数据,包含父级 menuList =getMenuList(menuList); //抽出所有的目录id,并且保证不重复 Set menuIds = menuList.stream().map(SysRoleMenuEntity::getMenuId).collect(Collectors.toSet()); //不存在则删除 list.stream().forEach(y -> { if(!menuIds.contains(y.getMenuId())){ roleMenuService.removeById(y.getId()); } }); //不存在则新增 menuList.stream().forEach(y -> { if(!arr.contains(y.getMenuId())){ y.setRoleId(role.getId()); roleMenuService.save(y); } }); } return update; } @Override public Result findList(PageEntity param) { PageHelper.startPage(param.getPage(), param.getLimit()); QueryWrapper wrapper = new QueryWrapper<>(); wrapper.eq("is_del","0"); List list = roleDao.selectList(wrapper); PageInfo pageInfo = new PageInfo<>(list); return Result.success(0,pageInfo.getTotal(),list); } @Override public int insert(SysRoleEntity role) { int num = roleDao.insert(role); //获取父级 if(num > 0){ List list = role.getMenuList(); if(list != null && list.size() > 0){ list =getMenuList(list); list.stream().forEach(y -> { y.setRoleId(role.getId()); roleMenuService.save(y); }); } } return num; } @Override public SysRoleEntity findById(Integer id) { SysRoleEntity entity = roleDao.selectById(id); if(entity != null){ List menuList = roleMenuService.findByRoleId(id); entity.setMenuList(menuList); } return entity; } @Override public void delIds(String[] ids) { //删除角色 roleDao.deleteBatchIds(Arrays.asList(ids)); //删除关联的信息 roleMenuService.remove(new QueryWrapper().in("role_id",ids)); } //新增和修改共用,获取目录和菜单id public List getMenuList(List list){ Set parentIdList = list.stream().map(SysRoleMenuEntity::getParentId).collect(Collectors.toSet()); if(parentIdList !=null && parentIdList.size() >0){ QueryWrapper wrapper = new QueryWrapper(); //排除为父级为0的数据 wrapper.ne("parent_id","0"); wrapper.in("id",parentIdList); //查出菜单和菜单，直接获取父节点的id List parentList = menuService.list(wrapper); Set collect = parentList.stream().map(SysMenuEntity::getParentId).collect(Collectors.toSet()); parentIdList.addAll(collect); parentIdList.stream().forEach(y -> { SysRoleMenuEntity entity = new SysRoleMenuEntity(); entity.setMenuId(y); list.add(entity); }); } return list; } }

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.macro.dao.SysRoleMenuDao; import com.macro.entity.SysRoleMenuEntity; import com.macro.service.SysRoleMenuService; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; import java.util.List; @Service public class SysRoleMenuServiceImpl extends ServiceImpl implements SysRoleMenuService { @Autowired private SysRoleMenuDao roleMenuDao; @Override public List findByRoleId(Integer roleId) { return roleMenuDao.findByRoleId(roleId); } }

import com.baomidou.mybatisplus.core.mapper.baseMapper; import com.macro.entity.SysRoleMenuEntity; import org.apache.ibatis.annotations.Param; import java.util.List; public interface SysRoleMenuDao extends baseMapper { List findByRoleId(@Param("roleId") Integer roleId); }

@GetMapping("menuList") public Result menuList(){ //用户展示树节点的 Map> map = new HashMap<>(); List menuList = menuService.menuList(); //用于比较是否选中的 QueryWrapper wrapper = new QueryWrapper<>(); wrapper.eq("is_del","0"); List list = menuService.list(wrapper); map.put("menuList",menuList); map.put("list",list); return Result.success(map); }

public List getHierarchyList(List menuList){ //获取所有的目录 //type = 2 为目录 List pathList = menuList.stream().filter(y -> y.getType().equals(2)).collect(Collectors.toList()); //获取所有菜单 //type = 1 为菜单 List childList = menuList.stream().filter(y -> y.getType().equals(1)).collect(Collectors.toList()); childList.stream().forEach( y -> { //直接从sys_menu表查询权限 List child = menuDao.selectList(new QueryWrapper().eq("parent_id",y.getId()).eq("is_del","0")); y.setChildList(child); }); pathList.stream().forEach( y-> { //目录id与菜单的父级id相同则菜单是目录的子级 List child = childList.stream().filter(x -> x.getParentId().equals(y.getId())).collect(Collectors.toList()); y.setChildList(child); }); return pathList; }

【爆肝推荐】手摸手带你做后台管理项目(第四章)整合redis添加shiro权限

Java相关栏目本月热门文章