|
@@ -1,9 +1,5 @@
|
|
|
package cn.iocoder.dashboard.framework.tracer.core.util;
|
|
|
|
|
|
-import cn.hutool.core.util.StrUtil;
|
|
|
-import cn.hutool.extra.spring.SpringUtil;
|
|
|
-import cn.iocoder.dashboard.framework.tracer.core.ITrace;
|
|
|
-import io.netty.util.concurrent.FastThreadLocal;
|
|
|
import org.apache.skywalking.apm.toolkit.trace.TraceContext;
|
|
|
|
|
|
import java.util.UUID;
|
|
@@ -15,36 +11,6 @@ import java.util.UUID;
|
|
|
*/
|
|
|
public class TracerUtils {
|
|
|
|
|
|
- /**
|
|
|
- * 维护请求线程对应的TraceId
|
|
|
- */
|
|
|
- private final static FastThreadLocal<String> traceIdMap = new FastThreadLocal<>();
|
|
|
-
|
|
|
- /**
|
|
|
- * 保存链路流水号
|
|
|
- *
|
|
|
- * @param traceId 链路流水号
|
|
|
- */
|
|
|
- public static void saveThreadTraceId(String traceId) {
|
|
|
- traceIdMap.set(traceId);
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * 根据线程获取链路流水号
|
|
|
- *
|
|
|
- * @return 链路流水号
|
|
|
- */
|
|
|
- public static String getThreadTraceId() {
|
|
|
- return traceIdMap.get();
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * 根据线程删除链路流水
|
|
|
- */
|
|
|
- public static void deleteThreadTraceId() {
|
|
|
- traceIdMap.remove();
|
|
|
- }
|
|
|
-
|
|
|
/**
|
|
|
* 私有化构造方法
|
|
|
*/
|
|
@@ -54,71 +20,13 @@ public class TracerUtils {
|
|
|
/**
|
|
|
* 获得链路追踪编号
|
|
|
* <p>
|
|
|
- * 一般来说,通过链路追踪编号,可以将访问日志,错误日志,链路追踪日志,logger 打印日志等,结合在一起,从而进行排错。
|
|
|
+ * 直接返回skywalking 的TraceId 如果不存在的话为空字符串""
|
|
|
* <p>
|
|
|
- * 默认情况下,我们使用 Apache SkyWalking 的 traceId 作为链路追踪编号。当然,可能会存在并未引入 Skywalking 的情况,此时使用 UUID 。
|
|
|
- * 方法获取顺序: skywalking - > map -> 扩展接口 -> default
|
|
|
- * 项目整体获取traceId的优先级 skywalking > 自定义实现接口 > 默认
|
|
|
- * map中的traceId 当且仅当其他情况为产生时才会出现.
|
|
|
*
|
|
|
* @return 链路追踪编号
|
|
|
*/
|
|
|
public static String getTraceId() {
|
|
|
- String traceId;
|
|
|
- // 通过 SkyWalking 获取链路编号
|
|
|
- try {
|
|
|
- traceId = TraceContext.traceId();
|
|
|
- if (StrUtil.isNotBlank(traceId)) {
|
|
|
- return traceId;
|
|
|
- }
|
|
|
- } catch (Throwable ignore) {
|
|
|
- }
|
|
|
- // 尝试从map中获取
|
|
|
- traceId = traceIdMap.get();
|
|
|
- if (StrUtil.isNotBlank(traceId)) {
|
|
|
- return traceId;
|
|
|
- }
|
|
|
- // 通过自定义扩展的tracer产生traceId, 在Spring容器加载完成前会获取不到对应的Bean
|
|
|
- ITrace tracer = null;
|
|
|
- try {
|
|
|
- tracer = getTracer();
|
|
|
- } catch (Throwable ignore) {
|
|
|
- }
|
|
|
- if (null != tracer) {
|
|
|
- try {
|
|
|
- return tracer.getTraceId();
|
|
|
- } catch (Throwable ignored) {
|
|
|
- }
|
|
|
- }
|
|
|
- // TODO 芋艿 多次调用会问题
|
|
|
-
|
|
|
- return defaultTraceId();
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * 仅仅获取Skywalking 中的traceId, 若无skywalking,则会返回""
|
|
|
- *
|
|
|
- * @return skywalking 的traceId
|
|
|
- */
|
|
|
- public static String getSkywalkingTraceId() {
|
|
|
return TraceContext.traceId();
|
|
|
}
|
|
|
- /**
|
|
|
- * 从Spring 容器中获取 ITrace 类,返回可以为null
|
|
|
- *
|
|
|
- * @return ITrace
|
|
|
- */
|
|
|
- private static ITrace getTracer() {
|
|
|
- return SpringUtil.getBean(ITrace.class);
|
|
|
- }
|
|
|
-
|
|
|
- /**
|
|
|
- * 默认生成TraceId规则为UUID
|
|
|
- *
|
|
|
- * @return UUID
|
|
|
- */
|
|
|
- private static String defaultTraceId() {
|
|
|
- return "UUID:" + UUID.randomUUID().toString();
|
|
|
- }
|
|
|
|
|
|
}
|