从零开始学 Java - Spring AOP 实现用户权限验证
ImaIHPC
8年前
<h2>每个项目都会有权限管理系统</h2> <p>无论你是一个简单的企业站，还是一个复杂到爆的平台级项目，都会涉及到用户登录、权限管理这些必不可少的业务逻辑。有人说，企业站需要什么权限管理阿？那行吧，你那可能叫静态页面，就算这样，但你肯定也会有后台管理及登录功能。</p> <p>每个项目中都会有这些几乎一样的业务逻辑，我们能不能把他们做成通用的系统呢？</p> <h2>AOP 实现用户权限验证</h2> <p>AOP 在实际项目中运用的场景主要有 <strong>权限管理（Authority Management）、事务管理（Transaction Management）、安全管理（Security）、日志管理（Logging）和调试管理（Debugging）</strong> 等。</p> <p>所以，权限验证正好我们可以使用 AOP 来直接实现。具体你项目中权限怎么管理，管理的粒度是什么级别这些完全取决于项目需要，这里完全不做任何的讨论。</p> <p>先说思路： <strong>利用自定义注解及拦截器来在你需要的时候，进行你需要的一些权限认证</strong>。 这里依然涉及到的有 enum（枚举） 、 annotation（自定义注解） 及拦截器相关知识，废话不多说，直接开写代码。</p> <h2><strong>开始撸一下代码</strong></h2> <p><strong>一、建立 AuthorityType.java 枚举类</strong></p> <pre> <code class="language-java">public enum AuthorityType { // 登录和权限都验证 默认 Validate, // 不验证 NoValidate, // 不验证权限 NoAuthority; }</code></pre> <p>这个枚举类的作用，依然是使自定义注解用起来爽到还想要。</p> <p><strong>二、新建 Authority.java 自定义注解类</strong></p> <pre> <code class="language-java">import java.lang.annotation.Documented; import java.lang.annotation.ElementType; import java.lang.annotation.Retention; import java.lang.annotation.RetentionPolicy; import java.lang.annotation.Target; @Retention(RetentionPolicy.RUNTIME) @Target(ElementType.METHOD) @Documented public @interface Authority { // 默认验证 AuthorityType value() default AuthorityType.Validate; }</code></pre> <p>三、<strong>再建一个 AuthorityAnnotationInterceptor.java 类</strong></p> <pre> <code class="language-java">/** * 权限认证拦截器 * */ public class AuthorityAnnotationInterceptor extends HandlerInterceptorAdapter { @Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { if (handler instanceof HandlerMethod) { HandlerMethod hm = (HandlerMethod) handler; Class<?> clazz = hm.getBeanType(); Method m = hm.getMethod(); try { if (clazz != null && m != null) { boolean isClzAnnotation = clazz.isAnnotationPresent(Authority.class); boolean isMethondAnnotation = m.isAnnotationPresent(Authority.class); Authority authority = null; // 如果方法和类声明中同时存在这个注解，那么方法中的会覆盖类中的设定。 if (isMethondAnnotation) { authority = m.getAnnotation(Authority.class); } else if (isClzAnnotation) { authority = clazz.getAnnotation(Authority.class); } int code = -1; String msg = ""; if (authority != null) { if (AuthorityType.NoValidate == authority.value()) { // 标记为不验证,放行 return true; } else if (AuthorityType.NoAuthority == authority.value()) { // 不验证权限，验证是否登录 // TODO: return true; } else { // 验证登录及权限 // TODO: code = 1; msg = "验证成功！"; return true; } } // //跳转 // String url = ""; // response.getWriter().write("<script>top.location.href='" // + url + "'</script>"); // return false; // 未通过验证，返回提示json Map<String, Object> responseMap = new HashMap<String, Object>(); responseMap.put("code", code); responseMap.put("msg", msg); responseMap.put("params", ""); responseMap.put("rows", ""); String json = new Gson().toJson(responseMap); response.setCharacterEncoding("UTF-8"); response.setContentType("application/json; charset=utf-8"); response.getWriter().write(json); return false; } } catch (Exception e) { } } return false; } }</code></pre> <p>这个类的目的就是在打过 Authority 标签的方法及类上，进行权限认证。我这里分了三种类型： <strong>全部验证、只验证登录、不验证</strong> 用来满足我们的业务需求。</p> <p>这里的返回值可以是 JSON 串，也可以是跳转到相应的页面，来实现你想要的效果。</p> <p><strong>四、配置拦截器</strong></p> <pre> <code class="language-java"><mvc:interceptors> <!-- 权限认证拦截器 --> <mvc:interceptor> <mvc:mapping path="/**"/> <bean class="cn.mayongfa.interceptor.AuthorityAnnotationInterceptor"></bean> </mvc:interceptor> </mvc:interceptors></code></pre> <p>在 /WebContent/WEB-INF/springMVC-servlet.xml 文件下的 <mvc:interceptors> 节点配置就行，这里可以配置具体要拦截的 Url 。</p> <p>到这里就完成了权限验证的工作了，如何使用呢？</p> <h2><strong>使用就非常简单</strong></h2> <p>因为我们的拦截器配置，然后我们在自定义注解的默认是验证，所以，我们只需要在类名及方法名上打标签就可以。</p> <p><img src="https://simg.open-open.com/show/bc1d46a21bcc7bca2afbb6b62220b1d0.png"></p> <p>当然，你完全是可以在拦截器中设置默认就验证所有请求的，接着设置不验证的请求。</p> <p> </p> <p> </p> <p>来自：http://blog.mayongfa.cn/88.html</p> <p> </p>