add method for TraceContextUtil

3 years ago · f406bb358d
parent fbfdecaef1
commit f406bb358d
1 changed files with 9 additions and 0 deletions
--- a/hippo4j-core/src/main/java/cn/hippo4j/core/toolkit/TraceContextUtil.java
+++ b/hippo4j-core/src/main/java/cn/hippo4j/core/toolkit/TraceContextUtil.java
@ -42,6 +42,15 @@ public class TraceContextUtil {
        return val;
    }

+    /**
+     * set.
+     *
+     * @return
+     */
+    public static void set(String trace) {
+        MDC.put(EXECUTE_TIMEOUT_TRACE_KEY, trace);
+    }
+
    /**
     * Set execute timeout trace key.
     *