Merge pull request #23 from chainbound/fix/backoff-timeout

Fix backoff timeouts
chainbound · Oct 23, 2023 · 3929e4e · 3929e4e
2 parents 817b9cc + e58a8b7
commit 3929e4e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/msg-transport/src/durable/session.rs b/msg-transport/src/durable/session.rs
@@ -181,10 +181,18 @@ where
  reconnect_status.current_attempt = Some(attempt);
  }
  SessionState::Processing(_) => {
- error!("Session was disconnected from {}", self.endpoint);
+ error!(
+ "Session was disconnected from {} during processing stage",
+ self.endpoint
+ );
+ let attempt = Box::pin(async move {
+ tokio::time::sleep(Duration::from_millis(100)).await;
+ Io::establish(endpoint).await
+ });
+
  self.state = SessionState::Disconnected(ReconnectStatus {
  attempts: 0,
- current_attempt: Some(Io::establish(self.endpoint)),
+ current_attempt: Some(attempt),
  });
  }
  SessionState::Terminated(_) => {

diff --git a/msg/Cargo.toml b/msg/Cargo.toml
@@ -30,10 +30,6 @@ futures.workspace = true
 tracing.workspace = true
 rand.workspace = true
 
-[[bench]]
-name = "req_res_localhost"
-harness = false
-
 [[bench]]
 name = "reqrep"
 harness = false