fix: revert revert retry get (#86)

* fix: revert revert retry get * fix: Add Send + Sync * fix: client ownerships
raiden-rs · Aug 3, 2020 · c07462f · c07462f
1 parent eb9d381
commit c07462f
Show file tree

Hide file tree

Showing 8 changed files with 419 additions and 14 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/raiden-derive/src/lib.rs b/raiden-derive/src/lib.rs
@@ -142,6 +142,7 @@ pub fn derive_raiden(input: TokenStream) -> TokenStream {
             )*
             table_prefix: String,
             table_suffix: String,
+            retry_condition: ::raiden::RetryCondition,
         }
 
         #attr_names
@@ -174,15 +175,21 @@ pub fn derive_raiden(input: TokenStream) -> TokenStream {
                     table_prefix: "".to_owned(),
                     table_suffix: "".to_owned(),
                     client,
+                    retry_condition: ::raiden::RetryCondition::new(),
                 }
             }
 
-            pub fn table_prefix(mut self, prefix: impl Into<String>) -> Self {
+            pub fn with_retries(&mut self, s: Box<dyn ::raiden::retry::RetryStrategy + Send + Sync>) -> &mut Self {
+                self.retry_condition.strategy = s;
+                self
+            }
+
+            pub fn table_prefix(&mut self, prefix: impl Into<String>) -> &mut Self {
                 self.table_prefix = prefix.into();
                 self
             }
 
-            pub fn table_suffix(mut self, suffix: impl Into<String>) -> Self {
+            pub fn table_suffix(&mut self, suffix: impl Into<String>) -> &mut Self {
                 self.table_suffix = suffix.into();
                 self
             }