7

xinyiZzz · Oct 25, 2023 · 4b9b56e · 4b9b56e
1 parent 84a5166
commit 4b9b56e
Showing 1 changed file with 5 additions and 5 deletions.
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
@@ -69,11 +69,6 @@ public void prepare(MysqlCommand command) {
         ctx.getState().reset();
         executor = null;
 
-        for (StmtExecutor asynExecutor : asynQueryResultExecutor) {
-            asynExecutor.finalizeQuery();
-        }
-        asynQueryResultExecutor.clear();
-
         if (command == null) {
             ErrorReport.report(ErrorCode.ERR_UNKNOWN_COM_ERROR);
             ctx.getState().setError(ErrorCode.ERR_UNKNOWN_COM_ERROR, "Unknown command(" + command.toString() + ")");
@@ -182,6 +177,11 @@ public Schema fetchArrowFlightSchema(int timeoutMs) {
     @Override
     public void close() throws Exception {
         ctx.setCommand(MysqlCommand.COM_SLEEP);
+        // TODO support query profile
+        for (StmtExecutor asynExecutor : asynQueryResultExecutor) {
+            asynExecutor.finalizeQuery();
+        }
+        asynQueryResultExecutor.clear();
         ConnectContext.remove();
     }
 }