From defca492d8bd2efd05d03bca8e94031fa4e90e5c Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" <41898282+github-actions[bot]@users.noreply.github.com> Date: Mon, 25 Mar 2024 13:58:00 +0800 Subject: [PATCH] fix(expr): handle another `Tonic` error path for `always_retry_on_network_error` (#15876) (#15878) Co-authored-by: Noel Kwan <47273164+kwannoel@users.noreply.github.com> Co-authored-by: xxchan --- src/expr/udf/src/error.rs | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/expr/udf/src/error.rs b/src/expr/udf/src/error.rs index 17306f172970d..fc6733052b137 100644 --- 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(_)) + ) } }