From dba9f93d29f88e4eb8c1a526a6ec6acdd4e6e2f5 Mon Sep 17 00:00:00 2001
From: Noel Kwan <noelkwan1998@gmail.com>
Date: Mon, 25 Mar 2024 12:12:39 +0800
Subject: [PATCH] handle flight error path

---
 src/expr/udf/src/error.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/expr/udf/src/error.rs b/src/expr/udf/src/error.rs
index 17306f172970..58e22c1b7930 100644
--- a/src/expr/udf/src/error.rs
+++ b/src/expr/udf/src/error.rs
@@ -57,7 +57,11 @@ impl Error {
     }
 
     pub fn is_tonic_error(&self) -> bool {
-        matches!(self.inner(), ErrorInner::Tonic(_))
+        match self.inner() {
+            ErrorInner::Tonic(_) => true,
+            ErrorInner::Flight(flight_error) => matches!(flight_error, FlightError::Tonic(_)),
+            _ => false,
+        }
     }
 }