feat: add retry for do_get

databendlabs · Aug 22, 2024 · 0357267 · 0357267
1 parent 2c45ee4
commit 0357267
Show file tree

Hide file tree

Showing 14 changed files with 700 additions and 184 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -252,6 +252,7 @@ http = "1"
 itertools = "0.10.5"
 jsonb = "0.4.1"
 jwt-simple = "0.11.0"
+hyper = "0.14.20"
 match-template = "0.0.1"
 mysql_async = { version = "0.34", default-features = false, features = ["native-tls-tls"] }
 object_store_opendal = "0.46"

diff --git a/src/common/exception/Cargo.toml b/src/common/exception/Cargo.toml
@@ -31,6 +31,7 @@ serde_json = { workspace = true }
 tantivy = { workspace = true }
 thiserror = { workspace = true }
 tonic = { workspace = true }
+hyper = {workspace = true}
 
 [package.metadata.cargo-machete]
 ignored = ["geos"]

diff --git a/src/common/exception/src/exception_into.rs b/src/common/exception/src/exception_into.rs
@@ -325,6 +325,13 @@ impl From<tonic::Status> for ErrorCode {
             tonic::Code::Unknown => {
                 let details = status.details();
                 if details.is_empty() {
+                    if status.source().map_or(false, |e| e.is::<hyper::Error>()) {
+                        return ErrorCode::CannotConnectNode(format!(
+                            "{}, source: {:?}",
+                            status.message(),
+                            status.source()
+                        ));
+                    }
                     return ErrorCode::UnknownException(format!(
                         "{}, source: {:?}",
                         status.message(),