fix(expr): handle another Tonic error path for `always_retry_on_net…

…work_error` (#15876) (#15878) Co-authored-by: Noel Kwan <[email protected]> Co-authored-by: xxchan <[email protected]>
risingwavelabs · Mar 25, 2024 · defca49 · defca49
1 parent 37a8404
commit defca49
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/expr/udf/src/error.rs b/src/expr/udf/src/error.rs
@@ -57,7 +57,10 @@ impl Error {
     }
 
     pub fn is_tonic_error(&self) -> bool {
-        matches!(self.inner(), ErrorInner::Tonic(_))
+        matches!(
+            self.inner(),
+            ErrorInner::Tonic(_) | ErrorInner::Flight(FlightError::Tonic(_))
+        )
     }
 }