pezkuwichain/pezkuwi-subxt
1
0
Fork 0
mirror of https://github.com/pezkuwichain/pezkuwi-subxt.git synced 2026-05-31 08:41:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Equivocation detection loop: Reorganize block checking logic as state machine (#2555) (#2557)

Browse Source 
* Split `reconnect_failed_client()` logic

* Reorganize block checking logic as state machine

This way we'll be able to save the state in case of a failure
This commit is contained in:
Serban Iorga
2023-09-12 16:38:26 +03:00
committed by Bastian Köcher
parent 5e5543a35a
commit c102e812eb
4 changed files with 340 additions and 217 deletions
Download Patch File Download Diff File Expand all files Collapse all files
bridges/relays/utils/src/relay_loop.rs
+25 -32
View File
@@ -35,6 +35,25 @@ pub trait Client: 'static + Clone + Send + Sync {
/// Try to reconnect to source node.
async fn reconnect(&mut self) -> Result<(), Self::Error>;
/// Try to reconnect to the source node in an infinite loop until it succeeds.
async fn reconnect_until_success(&mut self, delay: Duration) {
loop {
match self.reconnect().await {
Ok(()) => break,
Err(error) => {
log::warn!(
target: "bridge",
"Failed to reconnect to client. Going to retry in {}s: {:?}",
delay.as_secs(),
error,
);
async_std::task::sleep(delay).await;
},
}
}
}
}
#[async_trait]
@@ -226,44 +245,18 @@ impl<SC, TC, LM> LoopMetrics<SC, TC, LM> {
}
}
/// Deal with the client who has returned connection error.
/// Deal with the clients that have returned connection error.
pub async fn reconnect_failed_client(
failed_client: FailedClient,
reconnect_delay: Duration,
source_client: &mut impl Client,
target_client: &mut impl Client,
) {
loop {
async_std::task::sleep(reconnect_delay).await;
if failed_client == FailedClient::Both || failed_client == FailedClient::Source {
match source_client.reconnect().await {
Ok(()) => (),
Err(error) => {
log::warn!(
target: "bridge",
"Failed to reconnect to source client. Going to retry in {}s: {:?}",
reconnect_delay.as_secs(),
error,
);
continue
},
}
}
if failed_client == FailedClient::Both || failed_client == FailedClient::Target {
match target_client.reconnect().await {
Ok(()) => (),
Err(error) => {
log::warn!(
target: "bridge",
"Failed to reconnect to target client. Going to retry in {}s: {:?}",
reconnect_delay.as_secs(),
error,
);
continue
},
}
}
if failed_client == FailedClient::Source || failed_client == FailedClient::Both {
source_client.reconnect_until_success(reconnect_delay).await;
}
break
if failed_client == FailedClient::Target || failed_client == FailedClient::Both {
target_client.reconnect_until_success(reconnect_delay).await;
}
}