add proper backoff wrapper

Author: keyvankhademi

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "pyth-lazer-client"
-version = "1.0.0"
+version = "2.0.0"
 edition = "2021"
 description = "A Rust client for Pyth Lazer"
 license = "Apache-2.0"

lazer/sdk/rust/client/Cargo.toml

@@ -5,6 +5,26 @@ use backoff::{
     ExponentialBackoff, ExponentialBackoffBuilder,
 };
 
+#[derive(Debug)]
+pub struct PythLazerExponentialBackoff {
+    initial_interval: Duration,
+    randomization_factor: f64,
+    multiplier: f64,
+    max_interval: Duration,
+}
+
+impl From<PythLazerExponentialBackoff> for ExponentialBackoff {
+    fn from(val: PythLazerExponentialBackoff) -> Self {
+        ExponentialBackoffBuilder::default()
+            .with_initial_interval(val.initial_interval)
+            .with_randomization_factor(val.randomization_factor)
+            .with_multiplier(val.multiplier)
+            .with_max_interval(val.max_interval)
+            .with_max_elapsed_time(None)
+            .build()
+    }
+}
+
 #[derive(Debug)]
 pub struct PythLazerExponentialBackoffBuilder {
     initial_interval: Duration,
@@ -57,13 +77,12 @@ impl PythLazerExponentialBackoffBuilder {
         self
     }
 
-    pub fn build(&self) -> ExponentialBackoff {
-        ExponentialBackoffBuilder::default()
-            .with_initial_interval(self.initial_interval)
-            .with_randomization_factor(self.randomization_factor)
-            .with_multiplier(self.multiplier)
-            .with_max_interval(self.max_interval)
-            .with_max_elapsed_time(None)
-            .build()
+    pub fn build(&self) -> PythLazerExponentialBackoff {
+        PythLazerExponentialBackoff {
+            initial_interval: self.initial_interval,
+            randomization_factor: self.randomization_factor,
+            multiplier: self.multiplier,
+            max_interval: self.max_interval,
+        }
     }
 }

lazer/sdk/rust/client/src/backoff.rs

@@ -1,7 +1,9 @@
 use std::time::Duration;
 
 use crate::{
-    resilient_ws_connection::PythLazerResilientWSConnection, ws_connection::AnyResponse,
+    backoff::{PythLazerExponentialBackoff, PythLazerExponentialBackoffBuilder},
+    resilient_ws_connection::PythLazerResilientWSConnection,
+    ws_connection::AnyResponse,
     CHANNEL_CAPACITY,
 };
 use anyhow::{bail, Result};
@@ -43,13 +45,10 @@ impl PythLazerClient {
         endpoints: Vec<Url>,
         access_token: String,
         num_connections: usize,
-        backoff: ExponentialBackoff,
+        backoff: PythLazerExponentialBackoff,
         timeout: Duration,
         channel_capacity: usize,
     ) -> Result<Self> {
-        if backoff.max_elapsed_time.is_some() {
-            bail!("max_elapsed_time is not supported in Pyth Lazer client");
-        }
         if endpoints.is_empty() {
             bail!("At least one endpoint must be provided");
         }
@@ -58,7 +57,7 @@ impl PythLazerClient {
             access_token,
             num_connections,
             ws_connections: Vec::with_capacity(num_connections),
-            backoff,
+            backoff: backoff.into(),
             timeout,
             channel_capacity,
         })
@@ -128,7 +127,7 @@ pub struct PythLazerClientBuilder {
     endpoints: Vec<Url>,
     access_token: String,
     num_connections: usize,
-    backoff: ExponentialBackoff,
+    backoff: PythLazerExponentialBackoff,
     timeout: Duration,
     channel_capacity: usize,
 }
@@ -142,7 +141,7 @@ impl PythLazerClientBuilder {
                 .collect(),
             access_token,
             num_connections: DEFAULT_NUM_CONNECTIONS,
-            backoff: ExponentialBackoff::default(),
+            backoff: PythLazerExponentialBackoffBuilder::default().build(),
             timeout: DEFAULT_TIMEOUT,
             channel_capacity: CHANNEL_CAPACITY,
         }
@@ -158,7 +157,7 @@ impl PythLazerClientBuilder {
         self
     }
 
-    pub fn with_backoff(mut self, backoff: ExponentialBackoff) -> Self {
+    pub fn with_backoff(mut self, backoff: PythLazerExponentialBackoff) -> Self {
         self.backoff = backoff;
         self
     }