[breaking-change] Improve ping/pong logic (#422)

* v1 impl

* v1 impl

* Fix backoff reset

* fmt

* Add cleaning up

* add unit test

* add unit test

* add unit test

* Use LRU cache

* Use LRU cache

* unwraps -> expect

Author: SozinM

Cargo.lock

@@ -23,6 +23,7 @@ tokio = { version = "1", features = ["full"] }
 eyre = "0.6.12"
 url = "2.2.0"
 sha2 = { version = "0.10", default-features = false }
+backoff = "0.4.0"
 
 # Reth deps
 reth-optimism-payload-builder = { git = "https://github.com/paradigmxyz/reth", tag = "v1.8.3" }

Cargo.toml

@@ -64,6 +64,10 @@ paste = "1.0.15"
 parking_lot = "0.12.3"
 tokio-util = { version = "0.7.13" }
 dashmap = "6.1.0"
+backoff.workspace = true
+uuid = { version = "1.17.0", features = ["v4", "v7"] }
+bytes = "1.10.1"
+lru = "0.16"
 
 [dev-dependencies]
 rand = "0.9.0"

crates/rollup-boost/Cargo.toml

@@ -1,4 +1,6 @@
+use backoff::{ExponentialBackoff, ExponentialBackoffBuilder};
 use clap::Parser;
+use std::time::Duration;
 use url::Url;
 
 #[derive(Parser, Clone, Debug)]
@@ -19,7 +21,59 @@ pub struct FlashblocksArgs {
     #[arg(long, env, default_value = "1112")]
     pub flashblocks_port: u16,
 
-    /// Time used for timeout if builder disconnected
+    /// Websocket connection configuration
+    #[command(flatten)]
+    pub flashblocks_ws_config: FlashblocksWebsocketConfig,
+}
+
+#[derive(Parser, Debug, Clone, Copy)]
+pub struct FlashblocksWebsocketConfig {
+    /// Minimum time for exponential backoff for timeout if builder disconnected
+    #[arg(long, env, default_value = "10")]
+    pub flashblock_builder_ws_initial_reconnect_ms: u64,
+
+    /// Maximum time for exponential backoff for timeout if builder disconnected
     #[arg(long, env, default_value = "5000")]
-    pub flashblock_builder_ws_reconnect_ms: u64,
+    pub flashblock_builder_ws_max_reconnect_ms: u64,
+
+    /// Interval in milliseconds between ping messages sent to upstream servers to detect unresponsive connections
+    #[arg(long, env, default_value = "500")]
+    pub flashblock_builder_ws_ping_interval_ms: u64,
+
+    /// Timeout in milliseconds to wait for pong responses from upstream servers before considering the connection dead
+    #[arg(long, env, default_value = "1500")]
+    pub flashblock_builder_ws_pong_timeout_ms: u64,
+}
+
+impl FlashblocksWebsocketConfig {
+    /// Creates `ExponentialBackoff` use to control builder websocket reconnection time
+    pub fn backoff(&self) -> ExponentialBackoff {
+        ExponentialBackoffBuilder::default()
+            .with_initial_interval(self.initial_interval())
+            .with_max_interval(self.max_interval())
+            .with_randomization_factor(0 as f64)
+            .with_max_elapsed_time(None)
+            .with_multiplier(2.0)
+            .build()
+    }
+
+    /// Returns initial time for exponential backoff
+    pub fn initial_interval(&self) -> Duration {
+        Duration::from_millis(self.flashblock_builder_ws_initial_reconnect_ms)
+    }
+
+    /// Returns maximal time for exponential backoff
+    pub fn max_interval(&self) -> Duration {
+        Duration::from_millis(self.flashblock_builder_ws_max_reconnect_ms)
+    }
+
+    /// Returns ping interval
+    pub fn ping_interval(&self) -> Duration {
+        Duration::from_millis(self.flashblock_builder_ws_ping_interval_ms)
+    }
+
+    /// Returns pong interval
+    pub fn pong_interval(&self) -> Duration {
+        Duration::from_millis(self.flashblock_builder_ws_pong_timeout_ms)
+    }
 }