概述
在学习注解的时候,学了个懵懵懂懂。学了JavaWeb之后,在做Demo项目的过程中,借助注解和反射实现了对页面按钮的权限控制,对于注解才算咂摸出了点味儿来。
需求
以"角色列表"页面为例,该页面包含"新建","编辑","启用/禁用","删除"四个权限。根据用户所属角色的权限,来控制这些按钮是否显示。问题是,如何确定哪些页面下包含哪些按钮?
实现
定义注解
package com.ttpfx.bean;
import java.lang.annotation.*;
@Target(ElementType.METHOD) // 注解的作用对象:只能用于方法
@Retention(RetentionPolicy.RUNTIME) // 注解的级别:运行时有效,可以通过反射获取注解信息
@Repeatable(Buttons.class) // 可重复注解: 一个方法可以有多个Button注解,一个方法的多个Button注解将组成一个Buttons返回
public @interface Button {
String servlet(); // ServletName
String action(); // ActionName, 每个action是Servlet中的一个方法,代表一个请求地址:servletName?action=actionName
String name(); // 按钮名称
}
package com.ttpfx.bean;
import java.lang.annotation.*;
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Buttons {
Button[] value(); // Button注解的数组,通过Btuuons.value()获取页面的多个Button
}
在Servlet中使用注解
public class RoleServlet extends BaseServlet {
private RoleService roleService = new RoleServiceImpl();
// RoleServlet.index() 对应角色列表页面,访问地址为:/role?action=index
// 该页面包含"新建","编辑","启用/禁用","删除"四个权限
@Button(servlet = "role", action = "add", name = "add")
@Button(servlet = "role", action = "edit", name = "edit")
@Button(servlet = "role", action = "changeStatus", name = "changeStatus")
@Button(servlet = "role", action = "delete", name = "delete")
protected void index(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
// ....
}
}
在BaseServlet中校验页面按钮的权限,并将结果保存在requestScope中
// method:Servlet中的方法, 如上面RoleServlet.index()
private Map getButtonPermissions(Method method) {
// buttonMap是一个以Button.name为key, 以是否有权限的布尔值为value的键值对
// 该结果会被保存在requestScope中,供jsp页面使用
Map buttonMap = new HashMap<>();
Button[] buttonAnnotations = null;
// 页面有一个Button注解的时候可以直接获取
// 页面有多个Button注解的时候,只能获取到Buttons注解,再通过Buttons.value()方法得到多个Button注解
Buttons buttonsAnnotation = method.getAnnotation(Buttons.class);
Button buttonAnnotation = method.getAnnotation(Button.class);
if (buttonsAnnotation != null) {
buttonAnnotations = buttonsAnnotation.value();
} else if (buttonAnnotation != null){
buttonAnnotations = new Button[] {buttonAnnotation};
}
if (buttonAnnotations != null) {
for (Button button : buttonAnnotations) {
// 在这里实现对每个按钮的权限验证,将结果put至buttonMap
// 真正的验证过程已省略
buttonMap.put(button.name, true);
}
}
return buttonMap;
}
在页面的jsp文件中,控制按钮是否显示
新建角色