retries

aleph-client/Cargo.toml

@@ -10,7 +10,6 @@ anyhow = "1.0"
 codec = { package = 'parity-scale-codec', version = "3.0.0", features = ['derive'] }
 hex = { version = "0.4.3", features = ["alloc"] }
 log = "0.4"
-rayon = "1.5"
 serde_json = { version = "1.0" }
 thiserror = "1.0"
 contract-metadata = "1.5"

aleph-client/src/connections.rs

@@ -1,3 +1,5 @@
+use std::{thread::sleep, time::Duration};
+
 use anyhow::anyhow;
 use codec::Decode;
 use log::info;
@@ -51,12 +53,25 @@ impl SudoCall for RootConnection {
 }
 
 impl Connection {
+    const DEFAULT_RETRIES: u32 = 10;
+    const RETRY_WAIT_SECS: u64 = 1;
+
     pub async fn new(address: String) -> Self {
-        let client = Client::from_url(address)
-            .await
-            .expect("Should connect to the chain");
+        Self::new_with_retries(address, Self::DEFAULT_RETRIES).await
+    }
 
-        Self { client }
+    pub async fn new_with_retries(address: String, mut retries: u32) -> Self {
+        loop {
+            let client = Client::from_url(&address).await;
+            match (retries, client) {
+                (_, Ok(client)) => return Self { client },
+                (0, Err(e)) => panic!("{:?}", e),
+                _ => {
+                    sleep(Duration::from_secs(Self::RETRY_WAIT_SECS));
+                    retries -= 1;
+                }
+            }
+        }
     }
 
     pub async fn get_storage_entry<T: DecodeWithMetadata, Defaultable, Iterable>(