3

xinyiZzz · Dec 13, 2024 · f5d0417 · f5d0417
1 parent dbe7491
commit f5d0417
Showing 1 changed file with 2 additions and 0 deletions.
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
@@ -819,6 +819,7 @@ public void addReturnResultFromRemoteExecutor(StmtExecutor executor) {
     }
 
     public void finalizeArrowFlightSqlRequest() {
+        setThreadLocalInfo();
         for (StmtExecutor asynExecutor : returnResultFromRemoteExecutor) {
             asynExecutor.finalizeQuery();
         }
@@ -837,6 +838,7 @@ public void finalizeArrowFlightSqlRequest() {
             }
         }
 
+        remove();
         setCommand(MysqlCommand.COM_SLEEP);
         clear();
     }