From 19ce8bdf7cd5acfaba552220a4835d138dd6e597 Mon Sep 17 00:00:00 2001
From: Zou Xinyi <zouxinyi@selectdb.com>
Date: Tue, 17 Dec 2024 20:15:08 +0800
Subject: [PATCH] 5

---
 .../src/main/java/org/apache/doris/qe/ConnectContext.java     | 4 ++++
 .../doris/service/arrowflight/FlightSqlConnectProcessor.java  | 1 +
 2 files changed, 5 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 d347a31803b7cf7..c379d9c1ec710f0 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
@@ -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
index 65d8868de53238b..08481df56e45773 100644
--- 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);
     }