diff --git a/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectContext.java b/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectContext.java index 989151ae1a..81c8f7c53e 100644 --- a/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectContext.java +++ b/fe/fe-core/src/main/java/org/apache/doris/qe/ConnectContext.java @@ -699,6 +699,10 @@ public class ConnectContext { } } + public void setTraceId(String traceId) { + this.traceId = traceId; + } + public String traceId() { return traceId; } diff --git a/fe/fe-core/src/main/java/org/apache/doris/qe/VariableVarCallbacks.java b/fe/fe-core/src/main/java/org/apache/doris/qe/VariableVarCallbacks.java index 8726f894f8..f1dc5e69c3 100644 --- a/fe/fe-core/src/main/java/org/apache/doris/qe/VariableVarCallbacks.java +++ b/fe/fe-core/src/main/java/org/apache/doris/qe/VariableVarCallbacks.java @@ -63,6 +63,10 @@ public class VariableVarCallbacks { if (innerParts.length != 2) { continue; } + if (innerParts[0].equals("trace_id")) { + ConnectContext.get().setTraceId(innerParts[1]); + break; + } } } }