From 507ae5825efe67fc938146058ce27a443b0ef8d4 Mon Sep 17 00:00:00 2001
From: Ning Sun <sunng@protonmail.com>
Date: Wed, 18 Oct 2023 11:22:52 +0800
Subject: [PATCH] fix: make CollectRecordbatch an query error

---
 src/servers/src/error.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/servers/src/error.rs b/src/servers/src/error.rs
index 3efc2d3c65fa..db9724aaec51 100644
--- a/src/servers/src/error.rs
+++ b/src/servers/src/error.rs
@@ -392,7 +392,6 @@ impl ErrorExt for Error {
             Internal { .. }
             | InternalIo { .. }
             | TokioIo { .. }
-            | CollectRecordbatch { .. }
             | StartHttp { .. }
             | StartGrpc { .. }
             | AlreadyStarted { .. }
@@ -403,6 +402,8 @@ impl ErrorExt for Error {
             | GrpcReflectionService { .. }
             | BuildHttpResponse { .. } => StatusCode::Internal,
 
+            CollectRecordbatch { .. } => StatusCode::EngineExecuteQuery,
+
             InsertScript { source, .. }
             | ExecuteScript { source, .. }
             | ExecuteQuery { source, .. }