詳解Spring Boot中使用AOP統(tǒng)一處理Web請求日志

更新時間：2017年05月13日 16:24:08 作者：zmken497300

本篇文章主要介紹了詳解Spring Boot中使用AOP統(tǒng)一處理Web請求日志，具有一定的參考價值，感興趣的小伙伴們可以參考一下

在spring boot中，簡單幾步，使用spring AOP實現(xiàn)一個攔截器：

1、引入依賴：

<dependency> 
  <groupId>org.springframework.boot</groupId> 
  <artifactId>spring-boot-starter-aop</artifactId> 
</dependency>

<dependency> 
   <groupId>org.springframework.boot</groupId> 
   <artifactId>spring-boot-starter-aop</artifactId> 
 </dependency>

2、創(chuàng)建攔截器類（在該類中，定義了攔截規(guī)則：攔截com.xjj.web.controller包下面的所有類中，有@RequestMapping注解的方法。）：

/** 
 * 攔截器：記錄用戶操作日志，檢查用戶是否登錄…… 
 * @author XuJijun 
 */ 
@Aspect 
@Component 
public class ControllerInterceptor { 
  private static final Logger logger = LoggerFactory.getLogger(ControllerInterceptor.class); 
   
  @Value(“${spring.profiles}”) 
  private String env; 
   
  /** 
   * 定義攔截規(guī)則：攔截com.xjj.web.controller包下面的所有類中，有@RequestMapping注解的方法。 
   */ 
  @Pointcut(“execution(* com.xjj.web.controller..*(..)) and @annotation(org.springframework.web.bind.annotation.RequestMapping)”) 
  public void controllerMethodPointcut(){} 
   
  /** 
   * 攔截器具體實現(xiàn) 
   * @param pjp 
   * @return JsonResult（被攔截方法的執(zhí)行結(jié)果，或需要登錄的錯誤提示。） 
   */ 
  @Around(“controllerMethodPointcut()”) //指定攔截器規(guī)則；也可以直接把“execution(* com.xjj………)”寫進這里 
  public Object Interceptor(ProceedingJoinPoint pjp){ 
    long beginTime = System.currentTimeMillis(); 
    MethodSignature signature = (MethodSignature) pjp.getSignature(); 
    Method method = signature.getMethod(); //獲取被攔截的方法 
    String methodName = method.getName(); //獲取被攔截的方法名 
     
    Set<Object> allParams = new LinkedHashSet<>(); //保存所有請求參數(shù)，用于輸出到日志中 
     
    logger.info(”請求開始，方法：{}”, methodName); 
     
    Object result = null; 
 
    Object[] args = pjp.getArgs(); 
    for(Object arg : args){ 
      //logger.debug(“arg: {}”, arg); 
      if (arg instanceof Map<?, ?>) { 
        //提取方法中的MAP參數(shù)，用于記錄進日志中 
        @SuppressWarnings(“unchecked”) 
        Map<String, Object> map = (Map<String, Object>) arg; 
 
        allParams.add(map); 
      }else if(arg instanceof HttpServletRequest){ 
        HttpServletRequest request = (HttpServletRequest) arg; 
        if(isLoginRequired(method)){ 
          if(!isLogin(request)){ 
            result = new JsonResult(ResultCode.NOT_LOGIN, “該操作需要登錄！去登錄嗎？\n\n（不知道登錄賬號？請聯(lián)系老許。）”, null); 
          } 
        } 
         
        //獲取query string 或 posted form data參數(shù) 
        Map<String, String[]> paramMap = request.getParameterMap(); 
        if(paramMap!=null && paramMap.size()>0){ 
          allParams.add(paramMap); 
        } 
      }else if(arg instanceof HttpServletResponse){ 
        //do nothing… 
      }else{ 
        //allParams.add(arg); 
      } 
    } 
     
    try { 
      if(result == null){ 
        // 一切正常的情況下，繼續(xù)執(zhí)行被攔截的方法 
        result = pjp.proceed(); 
      } 
    } catch (Throwable e) { 
      logger.info(”exception: ”, e); 
      result = new JsonResult(ResultCode.EXCEPTION, “發(fā)生異常：”+e.getMessage()); 
    } 
     
    if(result instanceof JsonResult){ 
      long costMs = System.currentTimeMillis() - beginTime; 
      logger.info(”{}請求結(jié)束，耗時：{}ms”, methodName, costMs); 
    } 
     
    return result; 
  } 
   
  /** 
   * 判斷一個方法是否需要登錄 
   * @param method 
   * @return 
   */ 
  private boolean isLoginRequired(Method method){ 
    if(!env.equals(“prod”)){ //只有生產(chǎn)環(huán)境才需要登錄 
      return false; 
    } 
     
    boolean result = true; 
    if(method.isAnnotationPresent(Permission.class)){ 
      result = method.getAnnotation(Permission.class).loginReqired(); 
    } 
     
    return result; 
  } 
   
  //判斷是否已經(jīng)登錄 
  private boolean isLogin(HttpServletRequest request) { 
    return true; 
    /*String token = XWebUtils.getCookieByName(request, WebConstants.CookieName.AdminToken); 
    if(“1”.equals(redisOperator.get(RedisConstants.Prefix.ADMIN_TOKEN+token))){ 
      return true; 
    }else { 
      return false; 
    }*/ 
  } 
}

/** 
 * 攔截器：記錄用戶操作日志，檢查用戶是否登錄…… 
 * @author XuJijun 
 */ 
@Aspect 
@Component 
public class ControllerInterceptor { 
  private static final Logger logger = LoggerFactory.getLogger(ControllerInterceptor.class); 
 
  @Value("${spring.profiles}") 
  private String env; 
 
  /** 
   * 定義攔截規(guī)則：攔截com.xjj.web.controller包下面的所有類中，有@RequestMapping注解的方法。 
   */ 
  @Pointcut("execution(* com.xjj.web.controller..*(..)) and @annotation(org.springframework.web.bind.annotation.RequestMapping)") 
  public void controllerMethodPointcut(){} 
 
  /** 
   * 攔截器具體實現(xiàn) 
   * @param pjp 
   * @return JsonResult（被攔截方法的執(zhí)行結(jié)果，或需要登錄的錯誤提示。） 
   */ 
  @Around("controllerMethodPointcut()") //指定攔截器規(guī)則；也可以直接把“execution(* com.xjj.........)”寫進這里 
  public Object Interceptor(ProceedingJoinPoint pjp){ 
    long beginTime = System.currentTimeMillis(); 
    MethodSignature signature = (MethodSignature) pjp.getSignature(); 
    Method method = signature.getMethod(); //獲取被攔截的方法 
    String methodName = method.getName(); //獲取被攔截的方法名 
 
    Set<Object> allParams = new LinkedHashSet<>(); //保存所有請求參數(shù)，用于輸出到日志中 
 
    logger.info("請求開始，方法：{}", methodName); 
 
    Object result = null; 
 
    Object[] args = pjp.getArgs(); 
    for(Object arg : args){ 
      //logger.debug("arg: {}", arg); 
      if (arg instanceof Map<?, ?>) { 
        //提取方法中的MAP參數(shù)，用于記錄進日志中 
        @SuppressWarnings("unchecked") 
        Map<String, Object> map = (Map<String, Object>) arg; 
 
        allParams.add(map); 
      }else if(arg instanceof HttpServletRequest){ 
        HttpServletRequest request = (HttpServletRequest) arg; 
        if(isLoginRequired(method)){ 
          if(!isLogin(request)){ 
            result = new JsonResult(ResultCode.NOT_LOGIN, "該操作需要登錄！去登錄嗎？\n\n（不知道登錄賬號？請聯(lián)系老許。）", null); 
          } 
        } 
 
        //獲取query string 或 posted form data參數(shù) 
        Map<String, String[]> paramMap = request.getParameterMap(); 
        if(paramMap!=null && paramMap.size()>0){ 
          allParams.add(paramMap); 
        } 
      }else if(arg instanceof HttpServletResponse){ 
        //do nothing... 
      }else{ 
        //allParams.add(arg); 
      } 
    } 
 
    try { 
      if(result == null){ 
        // 一切正常的情況下，繼續(xù)執(zhí)行被攔截的方法 
        result = pjp.proceed(); 
      } 
    } catch (Throwable e) { 
      logger.info("exception: ", e); 
      result = new JsonResult(ResultCode.EXCEPTION, "發(fā)生異常："+e.getMessage()); 
    } 
 
    if(result instanceof JsonResult){ 
      long costMs = System.currentTimeMillis() - beginTime; 
      logger.info("{}請求結(jié)束，耗時：{}ms", methodName, costMs); 
    } 
 
    return result; 
  } 
 
  /** 
   * 判斷一個方法是否需要登錄 
   * @param method 
   * @return 
   */ 
  private boolean isLoginRequired(Method method){ 
    if(!env.equals("prod")){ //只有生產(chǎn)環(huán)境才需要登錄 
      return false; 
    } 
 
    boolean result = true; 
    if(method.isAnnotationPresent(Permission.class)){ 
      result = method.getAnnotation(Permission.class).loginReqired(); 
    } 
 
    return result; 
  } 
 
  //判斷是否已經(jīng)登錄 
  private boolean isLogin(HttpServletRequest request) { 
    return true; 
    /*String token = XWebUtils.getCookieByName(request, WebConstants.CookieName.AdminToken); 
    if("1".equals(redisOperator.get(RedisConstants.Prefix.ADMIN_TOKEN+token))){ 
      return true; 
    }else { 
      return false; 
    }*/ 
  } 
}

3、測試

瀏覽器中輸入：http://localhost:8082/api/admin/login

測試結(jié)果：

2016-07-26 11:58:12,057:INFO http-nio-8082-exec-1 (ControllerInterceptor.java:58) - 請求開始，方法：login 
2016-07-26 11:58:12,061:INFO http-nio-8082-exec-1 (ControllerInterceptor.java:103) - login請求結(jié)束，耗時：8ms

2016-07-26 11:58:12,057:INFO http-nio-8082-exec-1 (ControllerInterceptor.java:58) - 請求開始，方法：login 
2016-07-26 11:58:12,061:INFO http-nio-8082-exec-1 (ControllerInterceptor.java:103) - login請求結(jié)束，耗時：8ms

證明攔截器已經(jīng)生效。

源代碼參考：https://github.com/xujijun/my-spring-boot

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: