springBoot项目搭建包含RBAC模块 -- 角色管理模块开发（七）

码农小豪

已于 2022-07-25 14:33:48 修改

阅读量513

点赞数 2

分类专栏：框架文章标签： spring boot java

于 2022-07-05 15:33:05 首次发布

本文链接：https://blog.csdn.net/weixin_43470322/article/details/125620761

版权

框架专栏收录该内容

10 篇文章 1 订阅

订阅专栏

上一章讲了资源管理开发，这一章讲角色管理开发，角色管理不仅角色自己还包含和资源关联关系的开发。主要功能有新增角色、修改角色、删除角色、分页查询角色、获取用户对应角色（用户角色关联关系在用户管理模块开发）。

1. 角色管理开发

创建角色实体 id、roleName、roleDesc、createTime、updateTime

package com.swh.admin.pojo.db;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableName;
import com.swh.admin.pojo.req.ReqAddAdminRoleVo;
import com.swh.admin.pojo.req.ReqUpdateAdminRoleVo;
import com.swh.common.utils.DbIdUtil;
import com.swh.common.vo.BaseVo;
import lombok.Getter;
import lombok.NoArgsConstructor;
import lombok.Setter;

import java.util.Date;

/**
 * @ClassName: DbAdminRoleVo
 * @Description:
 * @Author: songWenHao
 * @Date: 2022/6/16 13:21
 */
@Setter
@Getter
@NoArgsConstructor
@TableName("admin_role")
public class DbAdminRoleVo extends BaseVo {

    private String id;
    private String roleName;
    private String roleDesc;
    @TableField(value = "create_time",fill = FieldFill.INSERT)
    private Date createTime;
    @TableField(value = "update_time",fill = FieldFill.INSERT_UPDATE)
    private Date updateTime;

    public DbAdminRoleVo(ReqAddAdminRoleVo req) {
        this.id = DbIdUtil.getInstance().nextUuid();
        this.roleName = req.getRoleName();
        this.roleDesc = req.getRoleDesc();
    }

    public DbAdminRoleVo(ReqUpdateAdminRoleVo req) {
        this.id = req.getId();
        this.roleName = req.getRoleName();
        this.roleDesc = req.getRoleDesc();
    }
}

创建角色管理controller层

AdminRoleController

package com.swh.admin.controller;

import com.swh.admin.pojo.req.ReqAddAdminRoleVo;
import com.swh.admin.pojo.req.ReqBatchAllAdminRoleVo;
import com.swh.admin.pojo.req.ReqBatchDelAdminRoleVo;
import com.swh.admin.pojo.req.ReqUpdateAdminRoleVo;
import com.swh.admin.pojo.resp.RespAdminRoleTreeVo;
import com.swh.admin.pojo.resp.RespAdminRoleVo;
import com.swh.admin.service.AdminRoleService;
import com.swh.common.annotation.ApiCode;
import com.swh.common.annotation.ApiCodes;
import com.swh.common.annotation.Menu;
import com.swh.common.menu.LevelThreeMenuEnum;
import com.swh.common.result.AjaxResult;
import com.swh.common.utils.Assert;
import com.swh.common.utils.JackJsonUtil;
import com.swh.common.vo.PageData;
import com.swh.common.vo.PageVo;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;

import java.util.List;

import static com.swh.common.constant.AdminUserConst.Url.*;
import static com.swh.common.result.ResultCode.*;

/**
 * @ClassName: AdminRoleController
 * @Description:
 * @Author: songWenHao
 * @Date: 2022/6/16 13:16
 */
@Slf4j
@Api(tags = "用户角色管理")
@RestController
public class AdminRoleController {

    @Autowired
    private AdminRoleService adminRoleService;

    @ApiCodes(value = {
            @ApiCode(menuName = "MENU_ADMIN_ROLE_LIST_PAGE", menuClass = LevelThreeMenuEnum.class)
    })
    @ApiOperation(value = "分页查询")
    @Menu(menuName = "MENU_ADMIN_ROLE_LIST_PAGE", menuClass = LevelThreeMenuEnum.class)
    @GetMapping(ADMIN_ROLE_LIST_PAGE)
    public AjaxResult<PageData<RespAdminRoleVo>> getAdminRoleListPage(@RequestBody PageVo pageVo) {
        LOGGER.debug("角色分页查询，请求参数:{}", JackJsonUtil.obj2StringPretty(pageVo));
        return AjaxResult.ok(adminRoleService.getAdminRoleListPage(pageVo));
    }

    @ApiCodes(value = {
            @ApiCode(menuName = "MENU_ADMIN_ROLE_ADD", menuClass = LevelThreeMenuEnum.class)
    })
    @ApiOperation(value = "新增角色")
    @Menu(menuName = "MENU_ADMIN_ROLE_ADD", menuClass = LevelThreeMenuEnum.class)
    @PostMapping(ADMIN_ROLE_ADD)
    public AjaxResult<String> addAdminRole(@RequestBody ReqAddAdminRoleVo req) {
        LOGGER.debug("角色新增，请求参数:{}", JackJsonUtil.obj2StringPretty(req));
        Assert.isNotBlank(req.getRoleName(), ROLE_NAME_NOT_NULL.getMsg());
        Assert.notEmpty(req.getResIds(), RES_IDS_NOT_NULL.getMsg());
        adminRoleService.addAdminRole(req);
        return AjaxResult.ok();
    }

    @ApiCodes(value = {
            @ApiCode(menuName = "MENU_ADMIN_ROLE_UPDATE", menuClass = LevelThreeMenuEnum.class)
    })
    @ApiOperation(value = "修改角色")
    @Menu(menuName = "MENU_ADMIN_ROLE_UPDATE", menuClass = LevelThreeMenuEnum.class)
    @PostMapping(ADMIN_ROLE_UPDATE)
    public AjaxResult<String> updateAdminRole(@RequestBody ReqUpdateAdminRoleVo req) {
        LOGGER.debug("修改角色，请求参数:{}", JackJsonUtil.obj2StringPretty(req));
        Assert.isNotBlank(req.getId(), ROLE_ID_NOT_NULL.getMsg());
        Assert.isNotBlank(req.getRoleName(), ROLE_NAME_NOT_NULL.getMsg());
        Assert.notEmpty(req.getResIds(), RES_IDS_NOT_NULL.getMsg());
        adminRoleService.updateAdminRole(req);
        return AjaxResult.ok();
    }

    @ApiCodes(value = {
            @ApiCode(menuName = "MENU_ADMIN_ROLE_BATCH_DEL", menuClass = LevelThreeMenuEnum.class)
    })
    @ApiOperation(value = "批量删除")
    @Menu(menuName = "MENU_ADMIN_ROLE_BATCH_DEL", menuClass = LevelThreeMenuEnum.class)
    @PostMapping(ADMIN_ROLE_BATCH_DEL)
    public AjaxResult<String> batchDelAdminRole(@RequestBody ReqBatchDelAdminRoleVo req) {
        LOGGER.debug("角色批量删除，请求参数:{}", JackJsonUtil.obj2StringPretty(req));
        Assert.notEmpty(req.getIds(), ROLE_ID_NOT_NULL.getMsg());
        adminRoleService.batchDelAdminRole(req);
        return AjaxResult.ok();
    }

    @ApiCodes(value = {
            @ApiCode(menuName = "MENU_ADMIN_ROLE_BATCH_ALL", menuClass = LevelThreeMenuEnum.class)
    })
    @ApiOperation(value = "获取所有的角色")
    @Menu(menuName = "MENU_ADMIN_ROLE_BATCH_ALL", menuClass = LevelThreeMenuEnum.class)
    @GetMapping(ADMIN_ROLE_BATCH_ALL)
    public AjaxResult<List<RespAdminRoleTreeVo>> batchGetAllAdminRole(@RequestBody ReqBatchAllAdminRoleVo req) {
        return AjaxResult.ok(adminRoleService.batchGetAllAdminRole(req));
    }


}

创建角色管理 service 和 serviceImpl

AdminRoleService

package com.swh.admin.service;

import com.swh.admin.pojo.req.ReqAddAdminRoleVo;
import com.swh.admin.pojo.req.ReqBatchAllAdminRoleVo;
import com.swh.admin.pojo.req.ReqBatchDelAdminRoleVo;
import com.swh.admin.pojo.req.ReqUpdateAdminRoleVo;
import com.swh.admin.pojo.resp.RespAdminRoleTreeVo;
import com.swh.admin.pojo.resp.RespAdminRoleVo;
import com.swh.common.vo.PageData;
import com.swh.common.vo.PageVo;

import java.util.List;

/**
 * @ClassName: AdminRoleService
 * @Description:
 * @Author: songWenHao
 * @Date: 2022/6/16 13:17
 */
public interface AdminRoleService {

    /**
     * 分页查询角色信息
     *
     * @param pageVo
     * @return
     */
    PageData<RespAdminRoleVo> getAdminRoleListPage(PageVo pageVo);

    /**
     * 新增角色
     *
     * @param req
     */
    void addAdminRole(ReqAddAdminRoleVo req);

    /**
     * 修改角色
     *
     * @param req
     */
    void updateAdminRole(ReqUpdateAdminRoleVo req);

    /**
     * 删除角色
     *
     * @param req
     */
    void batchDelAdminRole(ReqBatchDelAdminRoleVo req);

    /**
     * 获取所有角色
     *
     * @param req
     * @return
     */
    List<RespAdminRoleTreeVo> batchGetAllAdminRole(ReqBatchAllAdminRoleVo req);
}

AdminRoleServiceImpl

package com.swh.admin.service.impl;

import cn.hutool.core.collection.CollUtil;
import com.swh.admin.dao.AdminRoleDao;
import com.swh.admin.dao.RoleResMpDao;
import com.swh.admin.dao.UserRoleMpDao;
import com.swh.admin.pojo.db.DbAdminRoleVo;
import com.swh.admin.pojo.db.DbUserRoleMpVo;
import com.swh.admin.pojo.req.ReqAddAdminRoleVo;
import com.swh.admin.pojo.req.ReqBatchAllAdminRoleVo;
import com.swh.admin.pojo.req.ReqBatchDelAdminRoleVo;
import com.swh.admin.pojo.req.ReqUpdateAdminRoleVo;
import com.swh.admin.pojo.resp.RespAdminRoleTreeVo;
import com.swh.admin.pojo.resp.RespAdminRoleVo;
import com.swh.admin.service.AdminRoleService;
import com.swh.common.exception.RequestException;
import com.swh.common.vo.PageData;
import com.swh.common.vo.PageVo;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.stream.Collectors;

/**
 * @ClassName: AdminRoleServiceImpl
 * @Description:
 * @Author: songWenHao
 * @Date: 2022/6/16 13:17
 */
@Service
@Slf4j
public class AdminRoleServiceImpl implements AdminRoleService {

    @Autowired
    private AdminRoleDao adminRoleDao;
    @Autowired
    private RoleResMpDao roleResMpDao;
    @Autowired
    private UserRoleMpDao userRoleMpDao;

    @Override
    public PageData<RespAdminRoleVo> getAdminRoleListPage(PageVo pageVo) {
        PageData<DbAdminRoleVo> page = new PageData<>(adminRoleDao.findByPageCondition(pageVo));
        return new PageData<>(page.getPageIndex(), page.getPageSize(), page.getData().stream().map(RespAdminRoleVo::new).collect(Collectors.toList()), page.getTotal());
    }

    @Override
    public void addAdminRole(ReqAddAdminRoleVo req) {
        DbAdminRoleVo dbAdminRoleVo = new DbAdminRoleVo(req);
        // 添加角色资源关联关系
        roleResMpDao.addRoleResMp(dbAdminRoleVo.getId(), req.getResIds());
        // 添加角色信息
        adminRoleDao.addAdminRole(dbAdminRoleVo);
    }

    @Override
    public void updateAdminRole(ReqUpdateAdminRoleVo req) {
        // 修改角色资源关联关系
        roleResMpDao.updateRoleResMp(req.getId(), req.getResIds());
        // 修改角色信息
        adminRoleDao.updateAdminRole(new DbAdminRoleVo(req));
    }

    @Override
    public void batchDelAdminRole(ReqBatchDelAdminRoleVo req) {
        List<DbUserRoleMpVo> userRoleMpVoList = userRoleMpDao.getMpInfoByRoleIds(req.getIds());
        if (CollUtil.isNotEmpty(userRoleMpVoList)) {
            List<DbAdminRoleVo> roleVoList = adminRoleDao.getRoleByIds(req.getIds());
            List<String> roleNames = roleVoList.parallelStream().map(DbAdminRoleVo::getRoleName).collect(Collectors.toList());
            throw new RequestException(CollUtil.join(roleNames, ",") + "已绑定用户无法删除");
        }
    }

    @Override
    public List<RespAdminRoleTreeVo> batchGetAllAdminRole(ReqBatchAllAdminRoleVo req) {
        List<DbAdminRoleVo> adminRoleVoList = adminRoleDao.getAllAdminRole();
        List<String> roleIds = userRoleMpDao.getRoleIdsByUserId(req.getUserId());
        return adminRoleVoList.stream().map(tree -> new RespAdminRoleTreeVo(tree, roleIds)).collect(Collectors.toList());
    }
}

创建角色管理dao层

AdminRoleDao

package com.swh.admin.dao;

import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.swh.admin.mapper.AdminRoleMapper;
import com.swh.admin.pojo.db.DbAdminRoleVo;
import com.swh.common.vo.BaseDao;
import com.swh.common.vo.PageVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Repository;

import java.util.ArrayList;
import java.util.List;

/**
 * @ClassName: AdminRoleDao
 * @Description:
 * @Author: songWenHao
 * @Date: 2022/6/16 14:37
 */
@Repository
public class AdminRoleDao extends BaseDao<DbAdminRoleVo> {

    @Autowired
    private AdminRoleMapper adminRoleMapper;

    /**
     * 分页查询
     *
     * @param pageVo
     * @return
     */
    public IPage<DbAdminRoleVo> findByPageCondition(PageVo pageVo) {
        IPage<DbAdminRoleVo> page = new Page<>(pageVo.getPageIndex(), pageVo.getPageSize());
        QueryWrapper<DbAdminRoleVo> wrapper = buildPageQueryWrapper(pageVo);
        return adminRoleMapper.selectPage(page, wrapper);
    }

    /**
     * 新增角色信息
     *
     * @param dbAdminRoleVo
     */
    public void addAdminRole(DbAdminRoleVo dbAdminRoleVo) {
        adminRoleMapper.insert(dbAdminRoleVo);
    }

    /**
     * 修改角色信息
     *
     * @param dbAdminRoleVo
     */
    public void updateAdminRole(DbAdminRoleVo dbAdminRoleVo) {
        adminRoleMapper.updateById(dbAdminRoleVo);
    }

    /**
     * 根据角色Ids获取角色集合
     *
     * @param roleIds
     * @return
     */
    public List<DbAdminRoleVo> getRoleByIds(List<String> roleIds) {
        if (CollUtil.isNotEmpty(roleIds)) {
            return adminRoleMapper.selectBatchIds(roleIds);
        }
        return new ArrayList<>();
    }

    /**
     * 获取所有角色
     *
     * @return
     */
    public List<DbAdminRoleVo> getAllAdminRole() {
        return adminRoleMapper.selectList(new QueryWrapper<>());
    }
}

创建角色管理mapper层

AdminRoleMapper

package com.swh.admin.mapper;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.swh.admin.pojo.db.DbAdminRoleVo;
import org.springframework.stereotype.Component;

/**
 * @ClassName: AdminRoleMapper
 * @Description:
 * @Author: songWenHao
 * @Date: 2022/6/16 14:37
 */
@Component
public interface AdminRoleMapper extends BaseMapper<DbAdminRoleVo> {
}

2.角色资源关联关系开发

创建角色资源关联实体 id、roleId、resourceId

package com.swh.admin.pojo.db;

import com.baomidou.mybatisplus.annotation.TableName;
import com.swh.common.vo.BaseVo;
import lombok.Getter;
import lombok.NoArgsConstructor;
import lombok.Setter;

/**
 * @ClassName: DbRoleResMpVo
 * @Description:
 * @Author: songWenHao
 * @Date: 2022/6/16 15:14
 */
@Setter
@Getter
@NoArgsConstructor
@TableName("role_res_mp")
public class DbRoleResMpVo extends BaseVo {

    private String id;
    private String roleId;
    private String resourceId;

    public DbRoleResMpVo(String roleId, String resId) {
        this.roleId = roleId;
        this.resourceId = resId;
    }
}

创建角色资源关联dao

RoleResMpDao

package com.swh.admin.dao;

import cn.hutool.core.collection.CollUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.swh.admin.mapper.RoleResMpMapper;
import com.swh.admin.pojo.db.DbRoleResMpVo;
import com.swh.common.vo.BaseDao;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Repository;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * @ClassName: RoleResMpDao
 * @Description:
 * @Author: songWenHao
 * @Date: 2022/6/16 15:16
 */
@Repository
public class RoleResMpDao extends BaseDao<DbRoleResMpVo> {

    @Autowired
    private RoleResMpMapper roleResMpMapper;

    /**
     * 新增角色资源关系
     *
     * @param roleId
     * @param resIds
     */
    public void addRoleResMp(String roleId, List<String> resIds) {
        if (CollUtil.isNotEmpty(resIds)) {
            resIds.parallelStream().forEach(resId -> roleResMpMapper.insert(new DbRoleResMpVo(roleId, resId)));
        }
    }

    /**
     * 修改角色资源关系
     *
     * @param roleId
     * @param resIds
     */
    public void updateRoleResMp(String roleId, List<String> resIds) {
        roleResMpMapper.delete(new QueryWrapper<DbRoleResMpVo>().eq("role_id", roleId));
        addRoleResMp(roleId, resIds);
    }

    /**
     * 根据角色Id获取资源信息
     *
     * @param roleId
     * @return
     */
    public List<String> getResIdsByRoleId(String roleId) {
        List<DbRoleResMpVo> roleResMpVoList = roleResMpMapper.selectList(new QueryWrapper<DbRoleResMpVo>().eq("role_id", roleId));
        if (CollUtil.isNotEmpty(roleResMpVoList)) {
            return roleResMpVoList.stream().map(DbRoleResMpVo::getResourceId).collect(Collectors.toList());
        }
        return new ArrayList<>();
    }

    /**
     * 根据roleIds获取resIds
     *
     * @param roleIds
     * @return
     */
    public List<String> getResIdsByRoleIds(List<String> roleIds) {
        if (CollUtil.isNotEmpty(roleIds)) {
            List<DbRoleResMpVo> roleResMpVoList = roleResMpMapper.selectList(new QueryWrapper<DbRoleResMpVo>().in("role_id", roleIds));
            if (CollUtil.isNotEmpty(roleResMpVoList)) {
                return roleResMpVoList.stream().map(DbRoleResMpVo::getResourceId).collect(Collectors.toList());
            }
        }
        return new ArrayList<>();
    }
}

创建角色资源关联mapper

RoleResMpMapper

package com.swh.admin.mapper;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.swh.admin.pojo.db.DbRoleResMpVo;
import org.springframework.stereotype.Component;

/**
 * @ClassName: RoleResMpMapper
 * @Description:
 * @Author: songWenHao
 * @Date: 2022/6/16 15:17
 */
@Component
public interface RoleResMpMapper extends BaseMapper<DbRoleResMpVo> {
}

码农小豪

关注

2
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
springBoot项目搭建包含RBAC模块 -- 角色管理模块开发（七）

上一章讲了资源管理开发，这一章讲角色管理开发，角色管理不仅角色自己还包含和资源关联关系的开发。主要功能有新增角色、修改角色、删除角色、分页查询角色、获取用户对应角色（用户角色关联关系在用户管理模块开发）。1. 角色管理开发创建角色实体 id、roleName、roleDesc、createTime、updateTime创建角色管理controller层AdminRoleController创建角色管理 service 和 serviceImplAdminRoleServiceAdmin
复制链接

扫一扫