This is an automated email from the ASF dual-hosted git repository.

zouxinyi pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/doris.git


The following commit(s) were added to refs/heads/master by this push:
     new ad754cb58f9 [fix](fe ut) Fix set traceid failed #26808
ad754cb58f9 is described below

commit ad754cb58f936d181de8f4c42f128c3651fb397c
Author: zhiqiang <[email protected]>
AuthorDate: Sat Nov 11 20:55:10 2023 -0600

    [fix](fe ut) Fix set traceid failed #26808
    
    related to #26605
---
 fe/fe-core/src/main/java/org/apache/doris/qe/ConnectContext.java      | 4 ++++
 .../src/main/java/org/apache/doris/qe/VariableVarCallbacks.java       | 4 ++++
 2 files changed, 8 insertions(+)

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 989151ae1a6..81c8f7c53e5 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 8726f894f8c..f1dc5e69c3c 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;
+                }
             }
         }
     }


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to