在springboot中使用AOP進行全局日志記錄

更新時間：2021年11月24日 09:14:52 作者：WXWhowever

這篇文章主要介紹就在springboot中使用AOP進行全局日志記錄，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

前言

此前項目上需要對用戶的操作進行日志記錄，以便后續(xù)追蹤問題，所以就學習了使用 spring AOP 來進行日志記錄。

1、 spring AOP 是什么？

spring 的兩大核心就是 IOC 和 AOP，AOP 是 spring 框架面向切面的編程思想，AOP是橫切性的技術，將涉及多業(yè)務流程的通用功能抽取并單獨封裝，形成獨立的切面，再將這些切面橫向切入到業(yè)務流程指定的位置中。

2、spring AOP 能做什么？

AOP 主要是從切面入手，將日志記錄、安全控制、事務處理、全局異常處理等全局功能從我們的業(yè)務邏輯代碼中劃分出來。

3、spring AOP 我能用 AOP 解決什么問題？

正好我這邊需要做一個全局的日志，如果每個方法都去寫日志記錄，這樣代碼量非常大，而且代碼繁瑣冗余，全局記錄的話，只需要在方法上增加一個注解就可以實現(xiàn)日志記錄。

好了，不多bb，直接上代碼實現(xiàn)

一、引入依賴，增加自定義注解

1、引入 maven 依賴

        <!--利用 AOP 做操作日志記錄-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-aop</artifactId>
        </dependency>

2、增加自定義注解 OperationLog

@Target(ElementType.METHOD) //注解放置的目標位置,METHOD是可注解在方法級別上
@Retention(RetentionPolicy.RUNTIME) //注解執(zhí)行階段
@Documented
public @interface OperationLog {
    String operDesc() default "";  // 操作說明
}

二、為自定義注解編寫切面實現(xiàn)

@Aspect //標識一個切面類
@Component
public class OperationLogAspect {
    private static final Logger logger = LoggerFactory.getLogger(OperationLogAspect.class);
    @Autowired
    private OperatelogsService OperatelogsService;
    @Autowired
    private SysUserService userService;
    /**
     * 設置操作日志切入點 記錄操作日志 在注解的位置切入代碼
     */
    @Pointcut("@annotation(com.wxw.annotation.OperationLog)")
    public void operLogPoinCut() {
    }
    /**
     * 正常返回通知，攔截用戶操作日志，連接點正常執(zhí)行完成后執(zhí)行， 如果連接點拋出異常，則不會執(zhí)行
     *
     * @param joinPoint 切入點
     * @param keys      返回結果
     */
    @AfterReturning(value = "operLogPoinCut()", returning = "keys")
    public void saveOperLog(JoinPoint joinPoint, Object keys) {
        // 獲取RequestAttributes
        RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
        // 從獲取RequestAttributes中獲取HttpServletRequest的信息
        HttpServletRequest request = (HttpServletRequest) requestAttributes
                .resolveReference(RequestAttributes.REFERENCE_REQUEST);
        Operatelogs operlog = new Operatelogs();
        try {
            // 從切面織入點處通過反射機制獲取織入點處的方法
            MethodSignature signature = (MethodSignature) joinPoint.getSignature();
            // 獲取切入點所在的方法
            Method method = signature.getMethod();
            // 獲取操作
            OperationLog opLog = method.getAnnotation(OperationLog.class);
            if (opLog != null) {
                String operDesc = opLog.operDesc();
                operlog.setfDescript(operDesc); // 操作描述
            }
            // 獲取請求的類名
            String className = joinPoint.getTarget().getClass().getName();
            // 獲取請求的方法名
            String methodName = method.getName();
            methodName = className + "." + methodName;
            operlog.setfRequestMethod(methodName); // 請求方法
            // 請求的參數(shù)
            Object[] arguments = joinPoint.getArgs();
            //判斷參數(shù)數(shù)組是否為空
            Stream<?> stream = ArrayUtils.isEmpty(arguments) ? Stream.empty() :  Arrays.asList(arguments).stream();
            //過濾 joinPoint.getArgs()返回的數(shù)組中攜帶有Request或者Response對象
            List<Object> logArgs = stream
                    .filter(arg -> (!(arg instanceof HttpServletRequest) && !(arg instanceof HttpServletResponse)))
                    .collect(Collectors.toList());
            // 先將參數(shù)所在的list 轉換成json 數(shù)組
            JSONArray jsonArray = JSONArray.parseArray(JSON.toJSONString(logArgs));
            //再轉 json 字符串
            String params = jsonArray.toJSONString();
            operlog.setfRequestData(params); // 請求參數(shù)
            operlog.setfResponseData(JSON.toJSONString(keys)); // 返回結果
            SysUser user = getUser();
            operlog.setfOptby(user != null ? user.getfUserid() : 0); // 請求用戶ID
            operlog.setfOptname(user != null ? user.getfUsercode() : "admin"); // 請求用戶名稱
            operlog.setfOpton(new Date()); // 創(chuàng)建時間
            operlog.setfIp(getIp()); // 請求IP
            operlog.setfRequestUrl(request.getRequestURI()); // 請求URI
            OperatelogsService.save(operlog);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
    /**
     * 異常返回通知，用于攔截異常日志信息 連接點拋出異常后執(zhí)行
     *
     * @param joinPoint 切入點
     * @param e         異常信息
     */
    @AfterThrowing(pointcut = "operLogPoinCut()", throwing = "e")
    public void saveExceptionLog(JoinPoint joinPoint, Throwable e) {
        // 獲取RequestAttributes
        RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
        // 從獲取RequestAttributes中獲取HttpServletRequest的信息
        HttpServletRequest request = (HttpServletRequest) requestAttributes
                .resolveReference(RequestAttributes.REFERENCE_REQUEST);
        Operatelogs operlog = new Operatelogs();
        try {
            // 從切面織入點處通過反射機制獲取織入點處的方法
            MethodSignature signature = (MethodSignature) joinPoint.getSignature();
            // 獲取切入點所在的方法
            Method method = signature.getMethod();
            // 獲取請求的類名
            String className = joinPoint.getTarget().getClass().getName();
            // 獲取請求的方法名
            String methodName = method.getName();
            methodName = className + "." + methodName;
            // 請求的參數(shù)
            Object[] arguments = joinPoint.getArgs();
             //判斷參數(shù)數(shù)組是否為空
            Stream<?> stream = ArrayUtils.isEmpty(arguments) ? Stream.empty() :  Arrays.asList(arguments).stream();
            //過濾 joinPoint.getArgs()返回的數(shù)組中攜帶有Request或者Response對象
            List<Object> logArgs = stream
                    .filter(arg -> (!(arg instanceof HttpServletRequest) && !(arg instanceof HttpServletResponse)))
                    .collect(Collectors.toList());
            // 先將參數(shù)所在的list 轉換成json 數(shù)組
            JSONArray jsonArray = JSONArray.parseArray(JSON.toJSONString(logArgs));
            //再轉 json 字符串
            String params = jsonArray.toJSONString();
            operlog.setfRequestData(params); // 請求參數(shù)
            operlog.setfRequestMethod(methodName); // 請求方法名
            operlog.setfExceptionName(e.getClass().getName()); // 異常名稱
            operlog.setfResponseData(stackTraceToString(e.getClass().getName(), e.getMessage(), e.getStackTrace())); // 異常信息
            SysUser user = getUser();
            operlog.setfOptby(user != null ? user.getfUserid() : 0); // 請求用戶ID
            operlog.setfOptname(user != null ? user.getfUsercode() : "admin"); // 請求用戶名稱
            operlog.setfOpton(new Date()); // 創(chuàng)建時間
            operlog.setfRequestUrl(request.getRequestURI()); // 請求URI
            operlog.setfIp(getIp()); // 請求IP
            operatelogsService.save(operlog);
        } catch (Exception e2) {
            e2.printStackTrace();
        }
    }
    /**
     * 獲取用戶信息,由于我這邊 token 是使用 jwt 生成，所以可以從中獲取用戶id，再根據(jù)id獲取用戶數(shù)據(jù)
     *
     * @return
     */
    public SysUser getUser() {
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        SysUser sysUser = null;
        String token = request.getHeader("token");// 從 http 請求頭中取出 token
        if(token != null){
            // 獲取 token 中的 user id
            String userId = JWT.decode(token).getAudience().get(0);
            sysUser = userService.getOne(new QueryWrapper<SysUser>().lambda().eq(SysUser::getfUserid,
                    userId));
        }
        return sysUser;
    }
    /**
     * 獲取ip
     *
     * @return
     */
    public String getIp() {
        //獲得本機的ip和名稱
        InetAddress addr = null;
        try {
            addr = InetAddress.getLocalHost();
        } catch (UnknownHostException e) {
            throw new CustomException(500,e.getMessage());
        }
        String ip = addr.getHostAddress();
//      String hostname = addr.getHostName();
        return ip;
    }
    /**
     * 轉換異常信息為字符串
     *
     * @param exceptionName    異常名稱
     * @param exceptionMessage 異常信息
     * @param elements         堆棧信息
     */
    public String stackTraceToString(String exceptionName, String exceptionMessage, StackTraceElement[] elements) {
        StringBuffer strbuff = new StringBuffer();
        for (StackTraceElement stet : elements) {
            strbuff.append(stet + "\n");
        }
        String message = exceptionName + ":" + exceptionMessage + "\n\t" + strbuff.toString();
        return message;
    }
}