SpringBoot使用Aspect切面攔截打印請求參數

 

 

引入依賴

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

 

也用到了fastjson打印參數 ， 如果引了就不需要(也可以根據自己的來打印)

  <!-- 添加fastjson 支持 -->
        <dependency>
            <groupId>com.alibaba</groupId>
            <artifactId>fastjson</artifactId>
            <version>1.2.15</version>
        </dependency>

 

 

 

LogAspect.java

import com.alibaba.fastjson.JSON;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;

/***/
@Slf4j
@Component
@Aspect  //表示它是一個切面
public class LogAspect {

    /**
     *
     *  execution：改成自己要打印的控制器路徑
     * @param proceedingJoinPoint
     * @return
     * @throws Throwable
     */
    @Around("execution(* com.example.*.controller.*.*(..)) ")
    public Object handleControllerMethod(ProceedingJoinPoint proceedingJoinPoint) throws Throwable {

        //原始的HTTP請求和響應的信息
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        HttpServletRequest request = attributes.getRequest();

        Signature signature = proceedingJoinPoint.getSignature();
        MethodSignature methodSignature = (MethodSignature)signature;
        //獲取當前執行的方法
        Method targetMethod = methodSignature.getMethod();

        //獲取參數
        Object[] objects = proceedingJoinPoint.getArgs();
        //獲取參數
        Object[] arguments  = new Object[objects.length];
        for (int i = 0; i < objects.length; i++) {
            if (objects[i] instanceof ServletRequest || objects[i] instanceof ServletResponse || objects[i] instanceof MultipartFile) {
              //過濾掉不能序列化的參數
                continue;
            }
            arguments[i] = objects[i];
        }


        //獲取返回對象
        Object object = proceedingJoinPoint.proceed();

        StringBuilder sb = new StringBuilder(1000);
        sb.append("-------------------------------------------------------------\n");
        sb.append("Controller: ").append(targetMethod.getDeclaringClass().getName()).append("\n");
        sb.append("Method    : ").append(targetMethod.getName()).append("\n");
        sb.append("Params    : ").append(JSON.toJSONString(arguments)).append("\n");
        sb.append("URI       : ").append(request.getRequestURI()).append("\n");
        sb.append("URL       : ").append(request.getRequestURL()).append("\n");
        sb.append("Return    : ").append(object).append("\n");
        sb.append("-------------------------------------------------------------\n");
        System.out.println(sb);

        return object;
    }
}