Springboot给每个接口设置traceId,并添加到返回结果中

原理

slf4j有个MDC的类,是ThreadLocal的实现,保存在这里的变量会绑定到某个请求线程,于是在该请求的线程里的日志代码都可以使用设入的变量。 

实现

一、引入依赖

这个是可选项,用于生成唯一uid,我人懒,就直接用这个工具,你们要自定义uid格式的,就可以不用导入。

        <!-- hutool工具包 -->
        <dependency>
            <groupId>cn.hutool</groupId>
            <artifactId>hutool-all</artifactId>
            <version>5.7.20</version>
        </dependency>

二、设置拦截器

因为要给每一个请求都利用所以建议利用MDC绑定traceId,所以需要用拦截器拦截每一个请求。

先给设置一个拦截器配置InterceptorConfig.class并实现重写WebMvcConfigurer,如下:

@Configuration
public class InterceptorConfig implements WebMvcConfigurer {

    /**
     * http请求拦截器
     * @param registry
     */
    @Override
    public void addInterceptors(InterceptorRegistry registry) {

        registry.addInterceptor(new TraceIdInterceptor());
    }
}

里面的new TraceIdInterceptor()就是我们的核心配置类了,用于MDC绑定traceId到线程中,首先,新建一个TraceIdInterceptor.class,代码及注释如下:

@Component
public class TraceIdInterceptor implements HandlerInterceptor {

    /**
     * preHandle为请求前拦截
     */
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {

        // 在日志框架中的MDC中添加请求的唯一标识(用于返回接口及保存mysql)
        String traceId = String.valueOf(UUID.randomUUID());

        // 绑定key值到线程中
        MDC.put("traceId", traceId);
        response.addHeader("traceId",traceId);

        // 继续执行接口请求
        return true;
    }
}

三、设置返回体

然后通过【MDC.get("traceId")】实现获取traceId,这个可以用在统一返回的结果包装类里,如下: 

/**
 * 接口返回统一包装
 */
@Data
@NoArgsConstructor
@AllArgsConstructor
public class Result {
    private String traceId;
    private Integer code;
    private String msg;
    private Object data;

    /**
     * 成功,且返回体有数据
     * @param code
     * @param msg
     * @param data
     * @return
     */
    public static Result success(Integer code, String msg, Object data) {

        // 获取MDC中绑定的traceId
        String traceId = MDC.get("traceId");
        return new Result(traceId, code, msg, data);
    }

    /**
     * 成功,但返回体没数据
     * @param code
     * @param msg
     * @return
     */
    public static Result success(Integer code, String msg) {

        // 获取MDC中绑定的traceId
        String traceId = MDC.get("traceId");
        return new Result(traceId, code, msg, null);
    }
}

四、效果测试

Spring Boot中的过滤器(Filter)是用来处理HTTP请求之前或之后的一些操作,例如添加、修改或删除请求头设置响应状态等。如果想要拦截`TRACE`请求,你可以创建自定义的`WebFilter`在其中检查请求的方法(`HttpServletRequest`的`getMethod()`方法)。下面是一个简单的示例: ```java import org.springframework.stereotype.Component; import org.springframework.web.filter.OncePerRequestFilter; @Component public class TraceRequestFilter extends OncePerRequestFilter { @Override protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain chain) throws ServletException, IOException { String method = request.getMethod(); if ("TRACE".equals(method)) { // 检查是否为TRACE请求 System.out.println("Detected TRACE request"); // 这可以添加对TRACE请求的特殊处理,比如记录日志或返回特定响应 // 然后可以选择不继续传递给下一个Filter或Controller处理 return; // 或者chain.doFilter(request, response); } chain.doFilter(request, response); // 如果不是TRACE,正常执行后续过滤链 } } ``` 为了应用这个过滤器,你需要将其添加到`WebMvcConfigurer`或`ServletComponentScan`配置中。例如,在`application.yml`或`application.properties`中: ```yaml spring: mvc: additional-message-code-packages: com.example.yourpackage.filters ``` 然后在`@Configuration`类中扫描到你的过滤器: ```java @Configuration public class WebConfig implements WebMvcConfigurer { @Autowired private TraceRequestFilter traceRequestFilter; @Override public void addFilters(FilterRegistrationBean[] registrations) { FilterRegistrationBean registration = new FilterRegistrationBean(); registration.setFilter(traceRequestFilter); registration.addUrlPatterns("/*"); // 添加你要拦截的URL模式 registration.setName("traceRequestFilter"); registrations.add(registration); } } ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值