handle flight error path

risingwavelabs · Mar 25, 2024 · dba9f93 · dba9f93
1 parent 5eb63e8
commit dba9f93
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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,
+        }
     }
 }