servlet极简单权限拦截设计处理
8gw234
9年前
在java web的开发中 遇到的极其简单的权限控制,整个小项目就分为四五种用户的权限角色,因此不想考虑使用框架以及数据库建角色表,菜单表,操作表等来完成权限设计,而是采用反射,注解的简单方法完成。
核心代码
BaseServlet.java文件
import java.io.IOException; import java.lang.reflect.Method; import javax.servlet.ServletException; import javax.servlet.http.HttpServlet; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import com.zk.annotation.Permission; import com.zk.bean.User; /** * servlet基类 * * @Description TODO * @author zk * @version 1.0 * @date 2015-4-9 下午4:53:11 */ @SuppressWarnings("all") public abstract class BaseServlet extends HttpServlet { private static final long serialVersionUID = -6898295798172047477L; protected void service(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException { /** * 思路1 .根据method参数 处理 2.使用反射调用 * 2 加入简单的权限判断(当然也可以通过过滤器设计 拦截实现) * 由于本案例不需要复杂的权限设计,故没有采用分别建几张表 来存角色,菜单,操作等信息,只用在用户表添加一个字段 level 0 1,4,7来处理 * 0:此用户无效(软删除)相当于已经删除 * 1: 普通访客等级 无需登录 主要是前端的信息浏览 * 4 :会员等级 可以登录,进行相关操作 * 7 :顶级管理员操作 * 中间预留一些字段做不同的权限。 * 3权限判断的核心思路: * 1> 建一个 BaseServlet 父类 让之类Servlet继承 * 2> 在BaseServlet 中重写service 方法通过反射 以及在之类中的方法操作上加自定义注解【见下文自定义注解文件】 eg:@Permission(level=3) * 3> 在service 方法中通过 request.getParmenter("method") 获取在子类中对应的方法操作名称,并反射获取该方法名称 * 上面的注解信息 level * 4> 对注解信息 以及用户的登录状态 和用户的level状态进行判断 * 对于有些操作需要登录 或者权限等级较高的 就可以重定向处理拦截权限 * 0等级忽略 由于1 为 普通权限 故先判断是否为普通权限 就是普通网站访问者权限 * 当操作方法的leve为1 放行 * 不为1 说明需要更高的用户权限 * 那么首先判断是否登录 没有登录的 返回登录 * 登录成功的 获取登录用户的leve状态 和请求该方法的操作level权限等级对比 * 如果用户的leve小于该方法操作的 就拦截 返回,反之就可以进行方法操作 * * */ // 只处理了post的乱码问题,get自行处理 req.setCharacterEncoding("utf-8"); // 只是处理 了 字符流的问题,字节流自行处理 resp.setContentType("text/html;charset=utf-8"); String methodName = req.getParameter("method"); // 判断是否有方法 if (methodName == null || methodName.trim().isEmpty()) { throw new RuntimeException("亲!请传入method的参数"); } Class clazz = this.getClass(); Method method = null; try { method = clazz.getMethod(methodName, HttpServletRequest.class, HttpServletResponse.class); // 注意 此处利用注解 获取 每一个请求方法上的注解信息 Permission info = method.getAnnotation(Permission.class); // 如果获取到注解信息 if (info != null) { int level = info.level();// 访问此操作需要的权限级别 if (level != 1) { // level 级别为默认的 最低等级 故无需处理 跳过 // level !=1 说明需要登录 以及更高等级的用户权限 User user = (User) req.getSession() .getAttribute("backUser"); // 判断是否登录 if (req.getSession().getAttribute("backUser") == null) { String contextPath = getServletContext() .getContextPath(); resp.getWriter() .print("亲,您还没有登录,请<a href='" + contextPath + "/index.jsp' target='_parent'>登录</a>!"); return; } else { // 登录成功 判断用户所拥有等级与 此操作的等级 if (user.getLevel() < level) { String contextPath = getServletContext() .getContextPath(); resp.getWriter() .print("对不起,你暂且无权限操作,请<a href='" + contextPath + "/index.jsp' target='_parent'>登录</a>申请为会员!"); return; } } } } } catch (Exception e) { e.printStackTrace(); throw new RuntimeException("亲!传入method的参数错误"); } try { String result = (String) method.invoke(this, req, resp); if (result != null && !result.trim().isEmpty()) { req.getRequestDispatcher(result).forward(req, resp); } } catch (Exception e) { e.printStackTrace(); throw new RuntimeException(e); } } }
自定义注解文件Permission.java
import java.lang.annotation.ElementType; import java.lang.annotation.Retention; import java.lang.annotation.RetentionPolicy; import java.lang.annotation.Target; /** * 自定义权限注解 * @Description TODO * @author zk * @version 1.0 * @date 2015-4-12 下午9:34:46 */ @Retention(RetentionPolicy.RUNTIME) @Target(value={ElementType.METHOD,ElementType.TYPE}) public @interface Permission { int level (); }
子类Servlet的操作 使用示例:
import java.io.IOException; import java.util.ArrayList; import java.util.List; import java.util.Map; import javax.servlet.ServletException; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; import org.apache.commons.beanutils.BeanUtils; import com.zk.annotation.Permission; import com.zk.bean.Customer; import com.zk.bean.Message; import com.zk.bean.MessageType; import com.zk.bean.MessageTypeLev; import com.zk.bean.PageBean; import com.zk.bean.User; import com.zk.service.CustomerService; import com.zk.service.MessageService; import com.zk.service.MessageTypeService; /** * 和信息相关的控制器 继承自定义的BaseServlet * * @Description TODO * @author zk * @version 1.0 * @date 2015-4-11 上午10:21:56 */ public class MessageServlet extends BaseServlet { /** * 通过id删消息 * @param request * @param response * @return * @throws ServletException * @throws IOException */ @Permission(level = 3)// 注意 :注解的使用 不同的操作设置不同的值 public String deleteById(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException { String id = request.getParameter("id"); String pageCode = request.getParameter("pageCode"); String totalPage = request.getParameter("totalPage"); MessageService cs = new MessageService(); cs.deleteById(id); int pg = 1; if (Integer.parseInt(totalPage) % Integer.parseInt(pageCode) == 0) { if (Integer.parseInt(pageCode) > 1) { pg = Integer.parseInt(pageCode) - 1; } else { // 已结没有信息了 } } return "/message?method=listMsg&pageCode=" + pg; } }
附上user用户表设计的结构
-- ---------------------------- -- Table structure for t_customer -- ---------------------------- CREATE TABLE `t_customer` ( `id` varchar(40) NOT NULL, `username` varchar(20) DEFAULT NULL, `gender` varchar(10) DEFAULT NULL, `birthday` varchar(20) DEFAULT NULL, `cellphone` varchar(20) DEFAULT NULL, `email` varchar(40) DEFAULT NULL, `love` varchar(100) DEFAULT NULL, `type` varchar(40) DEFAULT NULL, PRIMARY KEY (`id`) ) ENGINE=InnoDB DEFAULT CHARSET=utf8;
总结:反射 注解 简单继承 统一处理。东西和思路比较粗糙,只是为了实现写简单功能,希望批判接受和建议,谢谢。