基于Spring AOP實現(xiàn)日志自動打印功能
更新時間：2025年01月06日 09:52:05 作者：程序員谷美
這篇文章主要介紹了基于Spring AOP實現(xiàn)日志自動打印功能,文中通過代碼示例講解的非常詳細(xì),對大家的學(xué)習(xí)或工作有一定的幫助,需要的朋友可以參考下
概述
日志記錄是現(xiàn)代應(yīng)用程序開發(fā)中的重要組成部分。然而，在實際開發(fā)中，我們經(jīng)常面臨一些痛點：
- 重復(fù)打印出入?yún)?/strong>：在開發(fā)中經(jīng)常需要在每個方法中手動編寫日志代碼，這樣的工作不僅重復(fù)，而且使代碼顯得冗余和雜亂。
- 統(tǒng)一的追蹤標(biāo)識： 在排查問題時，我們經(jīng)常通過接口請求信息去查日志，這個時候我們一般需要一個traceId將所有請求串起來，這幾乎是每個方法都必須實現(xiàn)的功能：打印traceId。
- 接口requestId的返回： 為了方便上下游查問題，一般在接口設(shè)計時，響應(yīng)體都包含一個requestId字段，這個 requestId 字段，一般就是我們打印的traceId。如果我們每次都手動設(shè)置，需要對各種返回情況(如204、500)等進(jìn)行判斷，顯得代碼雜亂和臃腫。
- 接口耗時打印：我們需要記錄每個功能耗時的情況，方便性能問題排查。
使用 Spring AOP和Logback來封裝日志打印組件
1. 引入相關(guān)依賴
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-aop</artifactId>
</dependency>
<dependency>
<groupId>ch.qos.logback</groupId>
<artifactId>logback-classic</artifactId>
</dependency>
2. 創(chuàng)建Aop切面
import com.fasterxml.jackson.databind.ObjectMapper;
import com.gis.components.dto.Response;
import com.gis.components.dto.StatusCode;
import com.gis.components.exception.BizException;
import com.gis.components.exception.SysException;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.slf4j.MDC;
import org.springframework.core.annotation.AnnotationUtils;
import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.UUID;
import java.util.function.Function;
@Slf4j
@Aspect
public class LogAspect {
private static ObjectMapper objectMapper = new ObjectMapper();
public static final String REQUEST_TRACE_ID = "REQUEST_TRACE_ID";
/**
* @LogTag 注解攔截器
* 在方法執(zhí)行前后記錄日志，并在出現(xiàn)異常時拋出RuntimeException。
* @param proceedingJoinPoint 切入點對象，提供了對目標(biāo)方法的控制權(quán)。
* @return 目標(biāo)方法的執(zhí)行結(jié)果。
*/
@Pointcut("execution(* com.yourpackage..*(..))")
public Object logExecution(ProceedingJoinPoint proceedingJoinPoint) {
return logExecutionInner(proceedingJoinPoint,e -> {
throw new RuntimeException(e);
});
}
/**
* 記錄方法執(zhí)行日志和處理異常。
* @param proceedingJoinPoint 切入點對象，用于獲取方法信息和執(zhí)行方法。
* @param exceptionHandler 異常處理函數(shù)，當(dāng)方法執(zhí)行過程中出現(xiàn)異常時交由異常處理函數(shù)處理
* @return 方法執(zhí)行的結(jié)果。
*/
private Object logExecutionInner(ProceedingJoinPoint proceedingJoinPoint, Function<Throwable,Object> exceptionHandler) {
long start = System.currentTimeMillis();
String traceId = null;
String currentTrace = MDC.get(REQUEST_TRACE_ID);
if(currentTrace == null || currentTrace.isEmpty()){
traceId = UUID.randomUUID().toString().replaceAll("-", "");
}else {
traceId = currentTrace;
}
MDC.put(REQUEST_TRACE_ID, traceId);
// 獲取方法聲明
MethodSignature signature = (MethodSignature) proceedingJoinPoint.getSignature();
Method method = signature.getMethod();
String s = proceedingJoinPoint.getSignature().toString();
log.info("START PROCESSING: {}",s);
// 方法入?yún)?
Object[] args = proceedingJoinPoint.getArgs();
try {
log.info("method[{}] request:{}",method.getName(), objectMapper.writeValueAsString(args));
// 執(zhí)行方法
Object result = proceedingJoinPoint.proceed();
Class returnType = signature.getReturnType();
if(returnType == Response.class || returnType.getGenericSuperclass() == Response.class){
((Response<?>) result).setRequestId(traceId);
}
log.info("method[{}] response:{}",method.getName(),objectMapper.writeValueAsString(result));
return result;
} catch (Throwable e){
log.info("method[{}] 執(zhí)行異常",method.getName(),e);
return exceptionHandler.apply(e);
}finally {
// 方法執(zhí)行時間
long executionTime = System.currentTimeMillis() - start;
log.info("END PROCESSING,cost:{} ms",executionTime);
MDC.remove(REQUEST_TRACE_ID);
}
}
}
通過上述代碼，你就能夠?qū)崿F(xiàn)切入點的方法的日志自動化打印、traceId自動打印、requestId自動填充的功能了。但還不夠通用，如果你想將上述功能封裝為一個包，提供給別人使用，該如何去做呢？請看下文。
使用自動裝配以及注解進(jìn)行擴(kuò)展
1. 取消LogAspect類上的@Component注解，通過配置類進(jìn)行創(chuàng)建bean對象。
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.EnableAspectJAutoProxy;
@Configuration
@EnableAspectJAutoProxy
public class LogAutoConfiguration {
@Bean
@ConditionalOnMissingBean(LogAspect.class)
public LogAspect catchLogAspect() {
return new LogAspect();
}
}
2. 創(chuàng)建注解
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
/**
* 日志注解，用于打印日志出入?yún)⒌裙δ?
*/
@Target({ElementType.METHOD,ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
public @interface LogTag {
/**
* 日志前綴
* @return
*/
String prefix() default "";
/**
* 是否將出入?yún)ο驤SON化打印
* @return true 表示轉(zhuǎn)化為JSON，false 表示打印原始入?yún)ⅰ?
*/
boolean printToJson() default true;
}
3.修改Aop切面代碼，改為攔截注解
import com.fasterxml.jackson.databind.ObjectMapper;
import com.gis.components.dto.Response;
import com.gis.components.dto.StatusCode;
import com.gis.components.exception.BizException;
import com.gis.components.exception.SysException;
import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.slf4j.MDC;
import org.springframework.core.annotation.AnnotationUtils;
import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.UUID;
import java.util.function.Function;
@Slf4j
@Aspect
@Component
public class LogAspect {
private static ObjectMapper objectMapper = new ObjectMapper();
public static final String REQUEST_TRACE_ID = "REQUEST_TRACE_ID";
@Pointcut("@annotation(LogTag) || @within(LogTag)")
public void logPointcut() {
}
/**
* @LogTag 注解攔截器
* 在方法執(zhí)行前后記錄日志，并在出現(xiàn)異常時拋出RuntimeException。
* @param proceedingJoinPoint 切入點對象，提供了對目標(biāo)方法的控制權(quán)。
* @return 目標(biāo)方法的執(zhí)行結(jié)果。
*/
@Around("logPointcut()")
public Object logExecution(ProceedingJoinPoint proceedingJoinPoint) {
return logExecutionInner(proceedingJoinPoint,e -> {
throw new RuntimeException(e);
});
}
/**
* 記錄方法執(zhí)行日志和處理異常。
* @param proceedingJoinPoint 切入點對象，用于獲取方法信息和執(zhí)行方法。
* @param exceptionHandler 異常處理函數(shù)，當(dāng)方法執(zhí)行過程中出現(xiàn)異常時交由異常處理函數(shù)處理
* @return 方法執(zhí)行的結(jié)果。
*/
private Object logExecutionInner(ProceedingJoinPoint proceedingJoinPoint, Function<Throwable,Object> exceptionHandler) {
long start = System.currentTimeMillis();
String traceId = null;
String currentTrace = MDC.get(REQUEST_TRACE_ID);
if(currentTrace == null || currentTrace.isEmpty()){
traceId = UUID.randomUUID().toString().replaceAll("-", "");
}else {
traceId = currentTrace;
}
MDC.put(REQUEST_TRACE_ID, traceId);
// 獲取方法聲明
MethodSignature signature = (MethodSignature) proceedingJoinPoint.getSignature();
Method method = signature.getMethod();
// 獲取注解
LogTag logTag = AnnotationUtils.findAnnotation(method, LogTag.class);
if (logTag == null) {
logTag = AnnotationUtils.findAnnotation(signature.getDeclaringType(), LogTag.class);
}
String LOG_PRE = logTag.prefix();
String s = proceedingJoinPoint.getSignature().toString();
log.info("{} START PROCESSING: {}", LOG_PRE,s);
// 方法入?yún)?
Object[] args = proceedingJoinPoint.getArgs();
try {
log.info("{} method[{}] request:{}",LOG_PRE,method.getName(), logTag.printToJson() ? objectMapper.writeValueAsString(args) : Arrays.toString(args));
// 執(zhí)行方法
Object result = proceedingJoinPoint.proceed();
Class returnType = signature.getReturnType();
if(returnType == Response.class || returnType.getGenericSuperclass() == Response.class){
((Response<?>) result).setRequestId(traceId);
}
log.info("{} method[{}] response:{}",LOG_PRE,method.getName(),logTag.printToJson() ? objectMapper.writeValueAsString(result) : result);
return result;
} catch (Throwable e){
log.info("{} method[{}] 執(zhí)行異常",LOG_PRE,method.getName(),e);
return exceptionHandler.apply(e);
}finally {
// 方法執(zhí)行時間
long executionTime = System.currentTimeMillis() - start;
log.info("{} END PROCESSING,cost:{} ms", LOG_PRE,executionTime);
MDC.remove(REQUEST_TRACE_ID);
}
}
}
4. 配置自動裝載
在resource目錄下，新建META-INF文件夾，文件夾中新建文件spring.factories，文件內(nèi)容為：
org.springframework.boot.autoconfigure.EnableAutoConfiguration =com.xxx.log.LogAutoConfiguration
效果展示
到此這篇關(guān)于基于Spring AOP實現(xiàn)日志自動打印功能的文章就介紹到這了,更多相關(guān)Spring AOP日志自動打印內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
Filter也稱之為過濾器，WEB開發(fā)人員通過Filter技術(shù)，對web服務(wù)器管理的所有web資源：從而實現(xiàn)一些特殊的功能。例如實現(xiàn)URL級別的權(quán)限訪問控制、過濾敏感詞匯、壓縮響應(yīng)信息等一些高級功能。下面我們來簡單學(xué)習(xí)一下吧2019-05-05
這篇文章主要介紹了淺談Java中hashCode的正確求值方法，分享了相關(guān)代碼示例，小編覺得還是挺不錯的，具有一定借鑒價值，需要的朋友可以參考下2018-02-02
這篇文章主要介紹了springboot整合 beatlsql的實例代碼，BeetSql是一個全功能DAO工具，同時具有hibernate 優(yōu)點 & Mybatis優(yōu)點功能，有興趣的可以了解一下2017-05-05
這篇文章主要介紹了Mybatis?如何傳入字符串參數(shù)，分割并遍歷，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-01-01
這篇文章主要介紹了Java中自定義異常詳解及實例代碼的相關(guān)資料,需要的朋友可以參考下2017-03-03
IntelliJ?IDEA教程之clean或者install?Maven項目的操作方法
這篇文章主要介紹了IntelliJ?IDEA教程之clean或者install?Maven項目的操作方法,本文分步驟給大家介紹兩種方式講解如何調(diào)試出窗口，需要的朋友可以參考下2023-04-04
這篇文章主要介紹了JAVA生成短8位UUID的實例講解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-01-01
MybatisPlus是一個高效的java持久層框架,它在Mybatis的基礎(chǔ)上增加了一些便捷的功能,提供了更加易用的API,可以大幅度提高開發(fā)效率,這篇文章主要給大家介紹了關(guān)于MyBatisPlus批量添加的優(yōu)化與報錯解決的相關(guān)資料,需要的朋友可以參考下2023-05-05
這篇文章主要介紹了Java FTPClient連接池的實現(xiàn)，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧2018-06-06
目錄穿越漏洞,也叫做目錄遍歷/路徑遍歷漏洞,本文主要介紹了java代碼審計之目錄遍歷的解決,文中通過案例介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2024-06-06