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);
    }
}

四、效果测试

  • 2
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值