5

xinyiZzz · Dec 17, 2024 · 19ce8bd · 19ce8bd
1 parent 47414e2
commit 19ce8bd
Show file tree

Hide file tree

Showing 2 changed files with 5 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
@@ -751,6 +751,10 @@ public boolean getIsExecStatusDone() {
         return isExecStatusDone;
     }
 
+    public void resetFinalizeArrowFlightSqlRequestFinished() {
+        this.finalizeArrowFlightSqlRequestFinished = false;
+    }
+
     public boolean getFinalizeArrowFlightSqlRequestFinished() {
         return finalizeArrowFlightSqlRequestFinished;
     }

diff --git a/fe/fe-core/src/main/java/org/apache/doris/service/arrowflight/FlightSqlConnectProcessor.java b/fe/fe-core/src/main/java/org/apache/doris/service/arrowflight/FlightSqlConnectProcessor.java
@@ -94,6 +94,7 @@ public void handleQuery(String query) throws ConnectionException {
         prepare(command);
 
         ctx.setRunningQuery(query);
+        ctx.resetFinalizeArrowFlightSqlRequestFinished();
         super.handleQuery(query);
     }