权限管理中，角色授权与认证属于权限模块中的关键模块，角色授权即是将角色能够操作的菜单资源分配给指定角色的行为，角色认证即是当用户扮演指定角色登录系统后系统对于用户操作的资源进行权限校验的操作，意思这里说明白了，那么在代码中应该具体怎么实现呢？
前端页面展示控制
后端权限访问控制
完成角色记录基本 crud 功能之后，接下来实现角色授权功能，这里实现角色授权首先完成待授权资源显示功能。对于资源的显示，这里使用开源的 tree 插件 ztree。
前端 ztree 显示的资源数据格式参考这里。
ModuleMapper.xml
<selectid="queryAllModules"resultType="com.xxxx.crm.dto.TreeDto">select id, IFNULL(parent_id,0) as pId, module_name AS name from t_module where is_valid=1</select>
ModuleService.java
publicList<TreeDto>queryAllModules(){returnmoduleMapper.queryAllModules();}
ModuleController.java
@RequestMapping("queryAllModules")@ResponseBodypublicList<TreeDto>queryAllModules(){returnmoduleService.queryAllModules();}
role.js 添加授权点击事件
//头工具栏事件table.on('toolbar(roles)',function(obj){varcheckStatus=table.checkStatus(obj.config.id);switch(obj.event){case"add":openAddOrUpdateRoleDialog();break;case"grant":openAddGrantDailog(checkStatus.data);break;};});functionopenAddGrantDailog(datas){if(datas.length==0){layer.msg("请选择待授权角色记录!",{icon:5});return;}if(datas.length>1){layer.msg("暂不支持批量角色授权!",{icon:5});return;}varurl=ctx+"/role/toAddGrantPage?roleId="+datas[0].id;vartitle="角色管理-角色授权";layui.layer.open({title:title,type:2,area:["600px","280px"],maxmin:true,content:url});}
RoleController.java 添加视图转发方法
@RequestMapping("toAddGrantPage")publicStringtoAddGrantPage(Integer roleId,Model model){model.addAttribute("roleId",roleId);return"role/grant";}
准备显示资源数据模板
views/role 目录下添加 grant.ftl 模板文件
<html><head><linkrel="stylesheet"href="${ctx}/static/js/zTree_v3-3.5.32/css/zTreeStyle/zTreeStyle.css"type="text/css"><scripttype="text/javascript"src="${ctx}/static/lib/jquery-3.4.1/jquery-3.4.1.min.js"></script><scripttype="text/javascript"src="${ctx}/static/js/zTree_v3-3.5.32/js/jquery.ztree.core.js"></script><scripttype="text/javascript"src="${ctx}/static/js/zTree_v3-3.5.32/js/jquery.ztree.excheck.js"></script></head><body><divid="test1"class="ztree"></div><inputid="roleId"value="${roleId!}"type="hidden"><scripttype="text/javascript">varctx="${ctx}";</script><scripttype="text/javascript"src="${ctx}/static/js/role/grant.js"></script></body></html>
添加 grant.js
varzTreeObj;$(function(){loadModuleInfo();});functionloadModuleInfo(){$.ajax({type:"post",url:ctx+"/module/queryAllModules"dataType:"json",success:function(data){// zTree 的参数配置，深入使用请参考 API 文档（setting 配置详解）varsetting={data:{simpleData:{enable:true}},view:{showLine:false// showIcon: false},check:{enable:true,chkboxType:{"Y":"ps","N":"ps"}}};varzNodes=data;zTreeObj=$.fn.zTree.init($("#test1"),setting,zNodes);}})}
RoleService.java
publicvoidaddGrant(Integer[]mids,Integer roleId){/**
* 核心表-t_permission t_role(校验角色存在)
* 如果角色存在原始权限 删除角色原始权限
* 然后添加角色新的权限 批量添加权限记录到t_permission
*/Role temp=selectByPrimaryKey(roleId);AssertUtil.isTrue(null==roleId||null==temp,"待授权的角色不存在!");intcount=permissionMapper.countPermissionByRoleId(roleId);if(count>0){AssertUtil.isTrue(permissionMapper.deletePermissionsByRoleId(roleId)<count,"权限分配失败!");}if(null!=mids&&mids.length>0){List<Permission>permissions=newArrayList<Permission>();for(Integer mid:mids){Permission permission=newPermission();permission.setCreateDate(newDate());permission.setUpdateDate(newDate());permission.setModuleId(mid);permission.setRoleId(roleId);permission.setAclValue(moduleMapper.selectByPrimaryKey(mid).getOptValue());permissions.add(permission);}permissionMapper.insertBatch(permissions);}}
RoleController.java
@RequestMapping("addGrant")@ResponseBodypublicResultInfoaddGrant(Integer[]mids,Integer roleId){roleService.addGrant(mids,roleId);returnsuccess("权限添加成功");}
修改 grant.js 文件 添加 ztree 复选框点击回调 onCheck 事件。
varzTreeObj;$(function(){loadModuleInfo();});functionloadModuleInfo(){$.ajax({type:"post",url:ctx+"/module/queryAllModules",dataType:"json",success:function(data){// zTree 的参数配置，深入使用请参考 API 文档（setting 配置详解）varsetting={data:{simpleData:{enable:true}},view:{showLine:false// showIcon: false},check:{enable:true,chkboxType:{"Y":"ps","N":"ps"}},callback:{onCheck:zTreeOnCheck}};varzNodes=data;zTreeObj=$.fn.zTree.init($("#test1"),setting,zNodes);}})}functionzTreeOnCheck(event,treeId,treeNode){varnodes=zTreeObj.getCheckedNodes(true);varroleId=$("#roleId").val();varmids="mids=";for(vari=0;i<nodes.length;i++){if(i<nodes.length-1){mids=mids+nodes[i].id+"&mids=";}else{mids=mids+nodes[i].id;}}$.ajax({type:"post",url:ctx+"/role/addGrant",data:mids+"&roleId="+roleId,dataType:"json",success:function(data){console.log(data);}})}
这里要实现已添加的角色记录权限再次查看或授权时显示原始权限的功能，ztree 复选框是否选择属性配置参考这里。
ModuleService.java
publicList<TreeDto>queryAllModules02(Integer roleId){List<TreeDto>treeDtos=moduleMapper.queryAllModules();// 根据角色id 查询角色拥有的菜单id List<Integer>List<Integer>roleHasMids=permissionMapper.queryRoleHasAllModuleIdsByRoleId(roleId);if(null!=roleHasMids&&roleHasMids.size()>0){treeDtos.forEach(treeDto->{if(roleHasMids.contains(treeDto.getId())){// 说明当前角色 分配了该菜单treeDto.setChecked(true);}});}returntreeDtos;}
角色拥有权限 sql 查询
<selectid="queryRoleHasAllModuleIdsByRoleId"parameterType="int"resultType="java.lang.Integer">select module_id from t_permission where role_id=#{roleId}</select>
ModuleController.java
@RequestMapping("queryAllModules")@ResponseBodypublicList<TreeDto>queryAllModules(Integer roleId){returnmoduleService.queryAllModules02(roleId);}
这里修改 grant.js 查询资源时传入当前选择角色 id
functionloadModuleInfo(){$.ajax({type:"post",url:ctx+"/module/queryAllModules",data:{roleId:$("#roleId").val()},dataType:"json",success:function(data){// zTree 的参数配置，深入使用请参考 API 文档（setting 配置详解）varsetting={data:{simpleData:{enable:true}},view:{showLine:false// showIcon: false},check:{enable:true,chkboxType:{"Y":"ps","N":"ps"}},callback:{onCheck:zTreeOnCheck}};varzNodes=data;zTreeObj=$.fn.zTree.init($("#test1"),setting,zNodes);}})}
当完成角色权限添加功能后，下一步就是对角色操作的资源进行认证操作，这里对于认证包含两块：
菜单级别显示控制
后端方法访问控制
系统根据登录用户扮演的不同角色来对登录用户操作的菜单进行动态控制显示操作，这里显示的控制使用 freemarker 指令+内建函数实现，指令与内建函数操作参考这里。
IndexController.java
/**
* 后端管理主页面
* @return
*/@RequestMapping("main")publicStringmain(HttpServletRequest request){Integer userId=LoginUserUtil.releaseUserIdFromCookie(request);request.setAttribute("user",userService.selectByPrimaryKey(userId));List<String>permissions=permissionService.queryUserHasRolesHasPermissions(userId);request.getSession().setAttribute("permissions",permissions);return"main";}
PermissionService.java
@ServicepublicclassPermissionServiceextendsBaseService<Permission,Integer>{@AutowiredprivatePermissionMapper permissionMapper;publicList<String>queryUserHasRolesHasPermissions(Integer userId){returnpermissionMapper.queryUserHasRolesHasPermissions(userId);}}
PermissionMapper.java & PermissionMapper.xml
publicinterfacePermissionMapperextendsBaseMapper<Permission,Integer>{List<String>queryUserHasRolesHasPermissions(Integer userId);}
<selectid="queryUserHasRolesHasPermissions"parameterType="int"resultType="java.lang.String">select distinct p.acl_value from t_user_role ur left join t_permission p on ur.role_id = p.role_id where ur.user_id=#{userId}</select>
这里仅显示部分菜单控制。
<#ifpermissions?seq_contains("60")><liclass="layui-nav-item"><ahref="javascript:;"class="layui-menu-tips"><iclass="fa fa-gears"></i><spanclass="layui-left-nav">系统设置</span><spanclass="layui-nav-more"></span></a><dlclass="layui-nav-child"><#ifpermissions?seq_contains("6010")><dd><ahref="javascript:;"class="layui-menu-tips"data-type="tabAdd"data-tab-mpi="m-p-i-11"data-tab="user/index"target="_self"><iclass="fa fa-user"></i><spanclass="layui-left-nav">用户管理</span></a></dd></#if><#ifpermissions?seq_contains("6020")><ddclass=""><ahref="javascript:;"class="layui-menu-tips"data-type="tabAdd"data-tab-mpi="m-p-i-12"data-tab="role/index"target="_self"><iclass="fa fa-tachometer"></i><spanclass="layui-left-nav">角色管理</span></a></dd></#if><#ifpermissions?seq_contains("6030")><ddclass=""><ahref="javascript:;"class="layui-menu-tips"data-type="tabAdd"data-tab-mpi="m-p-i-13"data-tab="module/index"target="_self"><iclass="fa fa-tachometer"></i><spanclass="layui-left-nav">菜单管理</span></a></dd></#if></dl></li></#if>
实现了菜单级别显示控制，但最终客户端有可能会通过浏览器来输入资源地址从而越过 ui 界面来访问后端资源，所以接下来加入控制方法级别资源的访问控制操作，这里使用 aop+自定义注解实现
@Target({ElementType.METHOD})@Retention(RetentionPolicy.RUNTIME)@Documentedpublic@interfaceRequirePermission{Stringcode()default"";}
@RequestMapping("list")@ResponseBody@RequirePermission(code="101001")publicMap<String,Object>querySaleChancesByParams(Integer flag,HttpServletRequest request,SaleChanceQuery saleChanceQuery){if(null!=flag&&flag==1){// 查询分配给当前登录用户 营销记录saleChanceQuery.setAggsinMan(LoginUserUtil.releaseUserIdFromCookie(request));}returnsaleChanceService.queryByParamsForTable(saleChanceQuery);}
@Component@AspectpublicclassPermissionProxy{@AutowiredprivateHttpSession session;@Around(value="@annotation(com.xxx.sys.annotaions.RequirePermission)")publicObjectaround(ProceedingJoinPoint pjp)throwsThrowable{List<String>permissions=(List<String>)session.getAttribute("permissions");if(null==permissions||permissions.size()==0){thrownewNoPermissionException();}Object result=null;MethodSignature methodSignature=(MethodSignature)pjp.getSignature();RequirePermission requirePermission=methodSignature.getMethod().getDeclaredAnnotation(RequirePermission.class);if(!(permissions.contains(requirePermission.code()))){thrownewNoPermissionException();}result=pjp.proceed();returnresult;}}
从 JDK5 开始，Java 增加对元数据的支持，也就是注解，注解与注释是有一定区别的，可以把注解理解为代码里的特殊标记，这些标记可以在编译，类加载，运行时被读取，并执行相应的处理。通过注解开发人员可以在不改变原有代码和逻辑的情况下在源代码中嵌入补充信息。下面我们来看看如何自定义注解。
创建自定义注解类
packagecom.lebyte.annotations;importjava.lang.annotation.Documented;importjava.lang.annotation.ElementType;importjava.lang.annotation.Inherited;importjava.lang.annotation.Retention;importjava.lang.annotation.RetentionPolicy;importjava.lang.annotation.Target;/* @Target,@Retention,@Inherited,@Documented
* 这四个是对注解进行注解的元注解，负责自定义的注解的属性
*/@Target({ElementType.TYPE,ElementType.METHOD})//表示注解的作用对象，ElementType.TYPE表示类，ElementType.METHOD表示方法@Retention(RetentionPolicy.RUNTIME)//表示注解的保留机制，RetentionPolicy.RUNTIME表示运行时注解@Inherited//表示该注解可继承@Documented//表示该注解可生成文档public@interfaceDesign{Stringauthor();//注解成员，如果注解只有一个成员，则成员名必须为value()，成员类型只能为原始类型intdata()default0;//注解成员，默认值为0}
使用注解
packagecom.lebyte;importcom.lebyte.annotations.Design;@Design(author="lebyte",data=100)//使用自定义注解，有默认值的成员可以不用赋值，其余成员都要赋值publicclassPerson{@Design(author="lebyte",data=90)publicvoidlive(){}}
解析注解
packagecom.lebyte;importjava.lang.annotation.Annotation;importjava.lang.reflect.Method;importcom.lebyte.annotations.Design;publicclassMain{publicstaticvoidmain(String[]args)throwsClassNotFoundException{Classc=Class.forName("com.lebyte.Person");//使用类加载器加载类//1、找到类上的注解if(c.isAnnotationPresent(Design.class)){//判断类是否被指定注解注解Design d=(Design)c.getAnnotation(Design.class);//拿到类上的指定注解实例System.out.println(d.data());}//2、找到方法上的注解Method[]ms=c.getMethods();for(Method m:ms){if(m.isAnnotationPresent(Design.class)){//判断方法是否被指定注解注解Design d=m.getAnnotation(Design.class);//拿到类上的指定注解实例System.out.println(d.data());}}//3、另外一种方法for(Method m:ms){Annotation[]as=m.getAnnotations();//拿到类上的注解集合for(Annotation a:as){if(ainstanceofDesign){//判断指定注解Design d=(Design)a;System.out.println(d.data());}}}}}
Method m:ms){
if(m.isAnnotationPresent(Design.class)){ //判断方法是否被指定注解注解
Design d=m.getAnnotation(Design.class); //拿到类上的指定注解实例
System.out.println(d.data());
}
}
//3、另外一种方法
for(Method m:ms){
Annotation[] as=m.getAnnotations(); //拿到类上的注解集合
for(Annotation a:as){
if(a instanceof Design){ //判断指定注解
Design d=(Design) a;
System.out.println(d.data());
}
}
}
}
}
