快捷導(dǎo)航

Java之Spring AOP 實(shí)現(xiàn)用戶(hù)權(quán)限驗(yàn)證

更新時(shí)間：2017年02月03日 11:49:01 作者：Mafly

本篇文章主要介紹了Java之Spring AOP 實(shí)現(xiàn)用戶(hù)權(quán)限驗(yàn)證，用戶(hù)登錄、權(quán)限管理這些是必不可少的業(yè)務(wù)邏輯，具有一定的參考價(jià)值，有興趣的可以了解一下。

每個(gè)項(xiàng)目都會(huì)有權(quán)限管理系統(tǒng)

無(wú)論你是一個(gè)簡(jiǎn)單的企業(yè)站，還是一個(gè)復(fù)雜到爆的平臺(tái)級(jí)項(xiàng)目，都會(huì)涉及到用戶(hù)登錄、權(quán)限管理這些必不可少的業(yè)務(wù)邏輯。有人說(shuō)，企業(yè)站需要什么權(quán)限管理阿？那行吧，你那可能叫靜態(tài)頁(yè)面，就算這樣，但你肯定也會(huì)有后臺(tái)管理及登錄功能。

每個(gè)項(xiàng)目中都會(huì)有這些幾乎一樣的業(yè)務(wù)邏輯，我們能不能把他們做成通用的系統(tǒng)呢？

AOP 實(shí)現(xiàn)用戶(hù)權(quán)限驗(yàn)證

AOP 在實(shí)際項(xiàng)目中運(yùn)用的場(chǎng)景主要有權(quán)限管理（Authority Management）、事務(wù)管理（Transaction Management）、安全管理（Security）、日志管理（Logging）和調(diào)試管理（Debugging）等。

所以，權(quán)限驗(yàn)證正好我們可以使用 AOP 來(lái)直接實(shí)現(xiàn)。具體你項(xiàng)目中權(quán)限怎么管理，管理的粒度是什么級(jí)別這些完全取決于項(xiàng)目需要，這里完全不做任何的討論。

先說(shuō)思路：利用自定義注解及攔截器來(lái)在你需要的時(shí)候，進(jìn)行你需要的一些權(quán)限認(rèn)證。這里依然涉及到的有enum（枚舉）、annotation（自定義注解）及攔截器相關(guān)知識(shí)，廢話不多說(shuō)，直接開(kāi)寫(xiě)代碼。

開(kāi)始擼一下代碼

**一、建立AuthorityType.java枚舉類(lèi)

public enum AuthorityType {

  // 登錄和權(quán)限都驗(yàn)證 默認(rèn)
  Validate,

  // 不驗(yàn)證
  NoValidate,

  // 不驗(yàn)證權(quán)限
  NoAuthority;
}

這個(gè)枚舉類(lèi)的作用，依然是使自定義注解用起來(lái)爽到還想要。

二、新建Authority.java自定義注解類(lèi)

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 { 
  // 默認(rèn)驗(yàn)證
  AuthorityType value() default AuthorityType.Validate;

}

三、再建一個(gè)AuthorityAnnotationInterceptor.java類(lèi)

/**
 * 權(quán)限認(rèn)證攔截器
 *
 */
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;
        // 如果方法和類(lèi)聲明中同時(shí)存在這個(gè)注解，那么方法中的會(huì)覆蓋類(lèi)中的設(shè)定。
        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()) {
            // 標(biāo)記為不驗(yàn)證,放行
            return true;
          } else if (AuthorityType.NoAuthority == authority.value()) {
            // 不驗(yàn)證權(quán)限，驗(yàn)證是否登錄
            // TODO:
            return true;
          } else {
            // 驗(yàn)證登錄及權(quán)限
            // TODO:

            code = 1;
            msg = "驗(yàn)證成功！";
            return true;
          }
        }

        // //跳轉(zhuǎn)
        // String url = "";
        // response.getWriter().write("<script>top.location.href='"
        // + url + "'</script>");
        // return false;

        // 未通過(guò)驗(yàn)證，返回提示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;
  }  
}

這個(gè)類(lèi)的目的就是在打過(guò)Authority標(biāo)簽的方法及類(lèi)上，進(jìn)行權(quán)限認(rèn)證。我這里分了三種類(lèi)型：全部驗(yàn)證、只驗(yàn)證登錄、不驗(yàn)證用來(lái)滿(mǎn)足我們的業(yè)務(wù)需求。

這里的返回值可以是 JSON 串，也可以是跳轉(zhuǎn)到相應(yīng)的頁(yè)面，來(lái)實(shí)現(xiàn)你想要的效果。

四、配置攔截器

<mvc:interceptors>
  <!-- 權(quán)限認(rèn)證攔截器 -->
  <mvc:interceptor>
    <mvc:mapping path="/**"/>
    <bean class="cn.mayongfa.interceptor.AuthorityAnnotationInterceptor"></bean>
  </mvc:interceptor>
</mvc:interceptors>

在/WebContent/WEB-INF/springMVC-servlet.xml文件下的<mvc:interceptors>節(jié)點(diǎn)配置就行，這里可以配置具體要攔截的 Url。

到這里就完成了權(quán)限驗(yàn)證的工作了，如何使用呢？

使用就非常簡(jiǎn)單

因?yàn)槲覀兊臄r截器配置，然后我們?cè)谧远x注解的默認(rèn)是驗(yàn)證，所以，我們只需要在類(lèi)名及方法名上打標(biāo)簽就可以。