diff --git a/src/service.rs b/src/service.rs index 8f18980..8447d7f 100644 --- a/src/service.rs +++ b/src/service.rs @@ -99,7 +99,10 @@ pub trait ServiceRequest: std::fmt::Debug { if let ClientError::Service(status) = err { matches!( status.code(), - tonic::Code::Unavailable | tonic::Code::DeadlineExceeded | tonic::Code::Unknown + tonic::Code::Unavailable + | tonic::Code::DeadlineExceeded + | tonic::Code::Cancelled + | tonic::Code::Unknown ) } else { false diff --git a/src/service/stream.rs b/src/service/stream.rs index b7ca05e..6882af4 100644 --- a/src/service/stream.rs +++ b/src/service/stream.rs @@ -228,7 +228,10 @@ impl ServiceRequest for AppendServiceRequest { if let ClientError::Service(status) = err { let retryable_error = matches!( status.code(), - tonic::Code::Unavailable | tonic::Code::DeadlineExceeded | tonic::Code::Unknown + tonic::Code::Unavailable + | tonic::Code::DeadlineExceeded + | tonic::Code::Cancelled + | tonic::Code::Unknown ); let policy_compliant = match self.append_retry_policy { AppendRetryPolicy::All => true,