spring boot通過自定義注解和AOP攔截指定的請求

一 准備工作

1.1 添加依賴

通過spring boot創建好工程后，添加如下依賴，不然工程中無法使用切面的注解，就無法對制定的方法進行攔截

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

 

1.2 工程目錄結構

其中 

MonitorRequest 自定義注解

RequestAspect 切面類

BookController 測試接口請求

 

二 自定義注解

package com.tinno.word.advice;

import java.lang.annotation.*;
import org.springframework.web.bind.annotation.Mapping;

/**
 * Created by xingle on 2019/5/15
 */
@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Mapping
@Documented
public @interface MonitorRequest {


}

注解的作用目標：

　　@Target(ElementType.TYPE)                      // 接口、類、枚舉、注解

　　@Target(ElementType.FIELD)                     // 字段、枚舉的常量

　　@Target(ElementType.METHOD)                 // 方法

　　@Target(ElementType.PARAMETER)            // 方法參數

　　@Target(ElementType.CONSTRUCTOR)       // 構造函數

　　@Target(ElementType.LOCAL_VARIABLE)   // 局部變量

　　@Target(ElementType.ANNOTATION_TYPE) // 注解

　　@Target(ElementType.PACKAGE)               // 包

 

三 切面類

 

package com.tinno.word.aspect;

import com.tinno.word.utils.CodeUtils;
import javax.servlet.http.HttpServletRequest;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

/**
 * Created by xingle on 2019/5/15
 * 此類為一個切面類，主要作用就是對接口的請求進行攔截
 * 攔截的方式，只需要在指定接口方法上面加上@MonitorRequest注解即可
 */

@Aspect
@Component
public class RequestAspect {

  private Logger logger = LoggerFactory.getLogger(this.getClass());

  /**
   * 環繞通知：
   * 環繞通知非常強大，可以決定目標方法是否執行，什么時候執行，執行時是否需要替換方法參數，執行完畢是否需要替換返回值。
   * 環繞通知第一個參數必須是org.aspectj.lang.ProceedingJoinPoint類型
   */
  @Around(value = "@annotation(com.tinno.word.advice.MonitorRequest)")
  public Object doBefore(ProceedingJoinPoint joinPoint) {

    //獲取到請求的屬性
    ServletRequestAttributes attributes =
        (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
    //獲取到請求對象
    HttpServletRequest request = attributes.getRequest();

    //獲取請求的方法，是Get還是Post請求
    logger.info("method=" + request.getMethod());

    Object[] args = joinPoint.getArgs();
    String de_input = "";
    try {
      logger.debug("請求參數 解密前:" + args[0].toString());
      //這里具體解密方法不再列出
      de_input = CodeUtils.decrypt(args[0].toString());
      logger.debug("請求參數 解密后:" + de_input);
    } catch (Exception e) {
      e.printStackTrace();
    }
    args[0] = de_input;

    Object returnValue = null;
    try {
      //執行方法，以新的參數（如果不帶args就是用原先的參數）
      returnValue = joinPoint.proceed(args);
      logger.debug("returnValue 加密前:" +returnValue.toString());
    } catch (Throwable throwable) {
      throwable.printStackTrace();
    }
    //執行完畢也可以替換目標方法返回值，這里加密替換，也可以不加密原樣返回
    returnValue = CodeUtils.encrypt(returnValue.toString());
    logger.debug("returnValue 加密后:" +returnValue.toString());
    return returnValue;

  }


}

 

四 Controller類

 

  @MonitorRequest
  @RequestMapping(value = "/getBookVersionLs", method = {RequestMethod.GET,RequestMethod.POST},produces={"text/html;charset=UTF-8;","application/json;"})
  public String getBookVersionLs(@RequestBody String json){
    logger.info("Controller-getBookVersionLs 版本選擇書本列表,請求參數："+json);
    JSONObject jsonObject = JSONObject.parseObject(json);
    String userId = jsonObject.getString("udid");
    String device_id = jsonObject.getString("device_id");
    Map<String, Object> data = JsonUtil.jsonText2JsonMap(jsonObject.getString("data"));

    String rand = data.get("rand").toString();
    String version =  data.get("version").toString();

    if (isEmpty(userId) || isEmpty(device_id)){
      ResultBody result = ResultBody.toFailure(ErrorInfoEnum.PARAMS_NO_COMPLETE.getStatus(),ErrorInfoEnum.PARAMS_NO_COMPLETE.getDest());
      return new Gson().toJson(result);
    }
    List<AllBookVersion> allList = bookService.getAllBookVersionLs();
    AllBookVersion first = allList.get(0);
    if (isEmpty(rand) && isEmpty(version)){
      rand = first.getRand();
      version = first.getVersionLs().get(0);
    }
    List<BookVersion> bookList = bookService.getBookVersionLs(device_id,rand,version);

    JSONObject obj = new JSONObject();
    obj.put("bookList",bookList);
    obj.put("allList",allList);
    ResultBody resultBody = ResultBody.toSussess(obj,"");
    return new Gson().toJson(resultBody);
  }

 

四 結果

請求參數：

D1D329BF7D4C02CB97C29CB03DF243D79A280D3DD1C6B0EC89B353394409DE0C950A48E2809E6539DE3A641CBA6A89957296D9E7E1DD4906505C5ADFAF89C0AA0B585B1338A57F1CAD8ABE8538E74B75C7947C0B593E3C6DB66DB2CDE07305B671E4405FFA5701C44590D3DE4973174E7D8FB983E82768A0DE086534494CAA49

 

返回結果：

加密后的結果