Spring MVC日志增加TraceId

在很多大型软件系统中,通常会在分布式环境下运行。在这种情况下,由于一个请求可能会跨越多个服务,因此在诊断问题时,将请求的上下文信息添加到每个日志条目中是非常有用的。

一个常见的做法是为每个请求生成一个唯一的ID,即traceId,并在所有相关的日志消息中包含它。这样,通过查看带有特定traceId的日志消息,我们可以轻松地找到与特定请求相关的所有消息。

以下是如何使用Log4j和Spring MVC拦截器实现这一目标的步骤:

  1. 使用MDC和拦截器设置traceId
    我们可以使用Log4j的Mapped Diagnostic Context (MDC)功能来将traceId添加到日志消息中。首先,我们需要创建一个Spring MVC拦截器,在处理请求之前在MDC中设置traceId。
import org.slf4j.MDC;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.util.UUID;

public class TraceInterceptor extends HandlerInterceptorAdapter {

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        MDC.put("traceId", UUID.randomUUID().toString().replace("-", ""));
        return true;
    }
}
  1. 在Spring MVC中启用拦截器
    然后,我们需要在Spring MVC配置中启用我们的拦截器。如果你正在使用XML配置,你可以这样做:
<mvc:interceptors>
    <mvc:interceptor>
        <mvc:mapping path="/**"/>
        <bean class="com.yourpackage.TraceInterceptor"/>
    </mvc:interceptor>
</mvc:interceptors>

如果你正在使用Spring Boot和Java配置,你可以这样做:

import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;

@Configuration
public class WebConfig implements WebMvcConfigurer {

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new TraceInterceptor());
    }
}
  1. 在日志消息中包含traceId
    最后,我们需要更新我们的日志配置,以便在每个日志消息中包含traceId。使用Log4j时,你可以在PatternLayout中使用%mdc来引用MDC的内容。以下是一个例子:
<?xml version="1.0" encoding="UTF-8"?>
<!--日志级别以及优先级排序: OFF > FATAL > ERROR > WARN > INFO > DEBUG > TRACE > ALL -->
<!--Configuration后面的status,这个用于设置log4j2自身内部的信息输出,可以不设置,当设置成trace时,你会看到log4j2内部各种详细输出-->
<!--monitorInterval:Log4j能够自动检测修改配置 文件和重新配置本身,设置间隔秒数-->
<configuration status="WARN" monitorInterval="30">
    <!--先定义所有的appender-->
    <appenders>
        <!--这个输出控制台的配置-->
        <console name="Console" target="SYSTEM_OUT">
            <!--输出日志的格式-->
            <PatternLayout pattern="[%d{yyyy-MM-dd HH:mm:ss:SSS}] [%p] [traceId:%X{traceId}] - %l - %m%n"/>
        </console>

        <!-- 这个会打印出所有的info及以下级别的信息,每次大小超过size,则这size大小的日志会自动存入按年份-月份建立的文件夹下面并进行压缩,作为存档-->
        <RollingFile name="RollingFileInfo" fileName="${sys:user.home}/logs/ceweb_info.log"
                     filePattern="${sys:user.home}/logs/$${date:yyyy-MM}/ceweb_info-%d{yyyy-MM-dd}-%i.log">
            <!--控制台只输出level及以上级别的信息(onMatch),其他的直接拒绝(onMismatch)-->
            <ThresholdFilter level="debug" onMatch="ACCEPT" onMismatch="DENY" />
            <PatternLayout pattern="[%d{yyyy-MM-dd HH:mm:ss:SSS}] [%p] [traceId:%X{traceId}] - %l - %m%n"/>
            <Policies>
                <TimeBasedTriggeringPolicy/>
                <SizeBasedTriggeringPolicy size="100 MB"/>
            </Policies>
        </RollingFile>
        <RollingFile name="RollingFileWarn" fileName="${sys:user.home}/logs/ceweb_warn.log"
                     filePattern="${sys:user.home}/logs/$${date:yyyy-MM}/ceweb_warn-%d{yyyy-MM-dd}-%i.log">
            <ThresholdFilter level="warn" onMatch="ACCEPT" onMismatch="DENY"/>
            <PatternLayout pattern="[%d{yyyy-MM-dd HH:mm:ss:SSS}] [%p] [traceId:%X{traceId}] - %l - %m%n"/>
            <Policies>
                <TimeBasedTriggeringPolicy/>
                <SizeBasedTriggeringPolicy size="100 MB"/>
            </Policies>
            <!-- DefaultRolloverStrategy属性如不设置,则默认为最多同一文件夹下7个文件,这里设置了20 -->
            <DefaultRolloverStrategy max="20"/>
        </RollingFile>
        <RollingFile name="RollingFileError" fileName="${sys:user.home}/logs/ceweb_error.log"
                     filePattern="${sys:user.home}/logs/$${date:yyyy-MM}/ceweb_error-%d{yyyy-MM-dd}-%i.log">
            <ThresholdFilter level="error" onMatch="ACCEPT" onMismatch="DENY"/>
            <PatternLayout pattern="[%d{yyyy-MM-dd HH:mm:ss:SSS}] [%p] [traceId:%X{traceId}] - %l - %m%n"/>
            <Policies>
                <TimeBasedTriggeringPolicy/>
                <SizeBasedTriggeringPolicy size="100 MB"/>
            </Policies>
        </RollingFile>
    </appenders>
    <!--然后定义logger,只有定义了logger并引入的appender,appender才会生效-->
    <loggers>
        <!--过滤掉spring和mybatis的一些无用的DEBUG信息-->
        <logger name="org.springframework" level="INFO"></logger>
        <logger name="org.mybatis" level="INFO"></logger>
        <!--<logger name="com.alibaba.druid.pool.DruidDataSource" level="INFO"></logger>-->
        <root level="INFO">
            <appender-ref ref="Console"/>
            <appender-ref ref="RollingFileWarn"/>
            <appender-ref ref="RollingFileInfo"/>
            <appender-ref ref="RollingFileError"/>
        </root>
    </loggers>
</configuration>

springboot配置文件设置

logging.level.root=INFO
logging.level.org.springframework.web=DEBUG
logging.level.org.hibernate=ERROR
#logging.path=D:\\demo1
logging.file.path=d:\\springboot3\\log\\demo2.log

log4j.properties

log4j.rootLogger=info,error,CONSOLE,DEBUG
log4j.appender.CONSOLE=org.apache.log4j.ConsoleAppender
log4j.appender.CONSOLE.layout=org.apache.log4j.PatternLayout
log4j.appender.CONSOLE.layout.ConversionPattern=%d{yyyy-MM-dd-HH-mm} [%X{TRACE_ID}] [%t] [%c] [%p] - %m%n
log4j.logger.info=info
log4j.appender.info=org.apache.log4j.DailyRollingFileAppender
log4j.appender.info.layout=org.apache.log4j.PatternLayout
log4j.appender.info.layout.ConversionPattern=%d{yyyy-MM-dd-HH-mm} [%X{TRACE_ID}] [%t] [%c] [%p] - %m%n
log4j.appender.info.datePattern='.'yyyy-MM-dd
log4j.appender.info.Threshold = info
log4j.appender.info.append=true
log4j.appender.info.File=d://springboot3/logs/api_services_info.log
log4j.logger.error=error
log4j.appender.error=org.apache.log4j.DailyRollingFileAppender
log4j.appender.error.layout=org.apache.log4j.PatternLayout
log4j.appender.error.layout.ConversionPattern=%d{yyyy-MM-dd-HH-mm} [%X{TRACE_ID}] [%t] [%c] [%p] - %m%n
log4j.appender.error.datePattern='.'yyyy-MM-dd
log4j.appender.error.Threshold = error
log4j.appender.error.append=true
log4j.appender.error.File=d://springboot3/logs/error/api_services_error.log
log4j.logger.DEBUG=DEBUG
log4j.appender.DEBUG=org.apache.log4j.DailyRollingFileAppender
log4j.appender.DEBUG.layout=org.apache.log4j.PatternLayout
log4j.appender.DEBUG.layout.ConversionPattern=%d{yyyy-MM-dd-HH-mm} [%X{TRACE_ID}] [%t] [%c] [%p] - %m%n
log4j.appender.DEBUG.datePattern='.'yyyy-MM-dd
log4j.appender.DEBUG.Threshold = DEBUG
log4j.appender.DEBUG.append=true
log4j.appender.DEBUG.File=d://springboot3/logs/debug/api_services_debug.log

这将在每个日志消息前添加一个包含traceId的前缀。

  1. 测试
    现在,我们可以测试我们的配置了。以下是一个简单的Spring MVC控制器,用于测试目的:
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("demo")
public class DemoController {

    Logger logger = LoggerFactory.getLogger(DemoController.class);

    @RequestMapping("add")
    public String add(String name) {
        logger.info("Adding test item [{}]", name);
        return "Add successful: " + name;
    }
}

当您执行一个请求到/demo/add时,您应该在日志中看到类似以下的输出:

2024-04-03-15-01 [b0fd5581437a4fedada9ce377cc412a9] [http-nio-8080-exec-2] [com.sun.log4jtest.controller.DemoController] [INFO] - Adding test item [zhangsan]

可以看到,我们成功地在日志消息中添加了traceId。

以上就是如何在Spring MVC和Log4j中使用MDC和拦截器来为日志消息添加traceId的步骤。希望对你有所帮助!

### 如何在 Spring 框架中设置包含 `requestId` 的日志记录 为了实现在 Spring 应用中的日志记录中加入唯一标识符 `requestId`,可以利用 Logback 和 Mapped Diagnostic Context (MDC) 技术来完成这一需求。以下是具体方法: #### 配置 Logback-Spring.xml 文件 首先,在项目的资源目录 (`src/main/resources`) 下创建或修改 `logback-spring.xml` 文件,定义日志格式以便能够动态插入 `requestId`。 ```xml <configuration> <!-- 定义控制台输出 --> <appender name="CONSOLE" class="ch.qos.logback.core.ConsoleAppender"> <encoder> <pattern>%d{yyyy-MM-dd HH:mm:ss} [%thread] %-5level %logger{36} - requestId=%X{requestId} %msg%n</pattern> </encoder> </appender> <!-- 设置根日志级别 --> <root level="info"> <appender-ref ref="CONSOLE"/> </root> </configuration> ``` 上述配置中 `%X{requestId}` 表示从 MDC 中获取键为 `requestId` 的值,并将其嵌入到每条日志消息中[^1]。 --- #### 使用拦截器注入 `requestId` 为了让每次 HTTP 请求都有一个唯一的 `requestId`,可以通过自定义拦截器的方式实现。下面是一个简单的拦截器类: ```java import org.slf4j.MDC; import javax.servlet.http.HttpServletRequest; import java.util.UUID; public class RequestIdInterceptor extends HandlerInterceptorAdapter { @Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { String traceId = request.getHeader("X-Trace-Id"); // 尝试从请求头读取 X-Trace-Id if (traceId == null || traceId.isEmpty()) { traceId = UUID.randomUUID().toString(); // 如果不存在,则生成一个新的 ID } MDC.put("requestId", traceId); // 将 requestId 放入 MDC return true; } @Override public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception { MDC.clear(); // 清理当前线程的 MDC 数据 } } ``` 接着注册该拦截器至 Spring MVC 配置中: ```java @Configuration public class WebConfig implements WebMvcConfigurer { @Bean public RequestIdInterceptor requestIdInterceptor() { return new RequestIdInterceptor(); } @Override public void addInterceptors(InterceptorRegistry registry) { registry.addInterceptor(requestIdInterceptor()); } } ``` 这样就可以确保每个 Controller 请求都会自动携带 `requestId` 并写入日志[^2]。 --- #### 处理异步任务和线程池场景下的 `requestId` 由于 Java 默认不会跨线程传递上下文数据(如 MDC),因此对于多线程环境(例如定时任务、线程池执行的任务等),需要手动复制 MDC 上下文。 ##### 方法一:使用 `ThreadPoolTaskExecutor` 自定义线程工厂 如果项目中有使用 Spring 提供的 `@Async` 注解或者 ThreadPool 来管理并发任务,可以在初始化线程池时指定一个继承自 `ThreadFactory` 的实现类,用于拷贝父线程的 MDC 到子线程。 ```java @Bean(name = "taskExecutor") public TaskExecutor taskExecutor() { ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor(); executor.setCorePoolSize(5); executor.setMaxPoolSize(10); executor.setThreadNamePrefix("CustomThread-"); // 设置自定义 ThreadFactory executor.setThreadFactory(new CustomizableThreadFactory((r) -> { Map<String, String> contextMap = MDC.getCopyOfContextMap(); Thread thread = Executors.defaultThreadFactory().newThread(r); thread.setName(thread.getName() + "-with-MDC"); thread.start(); Runnable wrappedRunnable = () -> { try { if (contextMap != null) { MDC.setContextMap(contextMap); // 还原 MDC 上下文 } r.run(); } finally { MDC.clear(); // 执行完毕清理 MDC } }; return thread; })); executor.initialize(); return executor; } ``` ##### 方法二:显式传播 MDC 另一种方式是在启动新线程之前主动保存现有 MDC 状态,并在线程结束前恢复它。 ```java private static final ExecutorService EXECUTOR_SERVICE = Executors.newFixedThreadPool(5); public void executeWithMdc(Runnable command) { Map<String, String> currentMdcContext = MDC.getCopyOfContextMap(); EXECUTOR_SERVICE.submit(() -> { try { if (currentMdcContext != null) { MDC.setContextMap(currentMdcContext); } command.run(); } finally { MDC.clear(); } }); } ``` 这种方法适用于更复杂的业务逻辑。 --- #### 总结 通过以上步骤,即可成功地在 Spring Boot 应用程序中集成基于 `requestId` 的分布式跟踪机制。这不仅有助于排查问题,还能提升系统的可观测性和调试效率。
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值