github/awatcher
1
1
Fork 0
mirror of https://github.com/2e3s/awatcher.git synced 2025-06-04 02:20:15 +00:00
Code Issues 17 Packages Projects Releases Wiki Activity

Add 1 more second to reconnect

Browse Source 
Also, code is cleaner.
This commit is contained in:
Demmie 2024-06-11 14:27:10 -04:00
parent 214ff1c06e
commit 1f7ed75a0b
No known key found for this signature in database
GPG Key ID: B06DAA3D432C6E9A
1 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
watchers/src/report_client.rs
View File

@ -39,24 +39,22 @@ impl ReportClient {
Fut: Future<Output = Result<T, E>>,
E: std::error::Error + Send + Sync + 'static,
{
let mut interval = tokio::time::interval(tokio::time::Duration::from_secs(1));
let mut attempts = 0;
loop {
for (attempt, &secs) in [1, 2].iter().enumerate() {
match f().await {
Ok(val) => return Ok(val),
Err(e)
if attempts < 3
&& e.to_string()
.contains("tcp connect error: Connection refused") =>
if e.to_string()
.contains("tcp connect error: Connection refused") =>
{
warn!("Failed to connect, retrying: {}", e);
warn!("Failed to connect on attempt #{attempt}, retrying: {}", e);
attempts += 1;
interval.tick().await;
tokio::time::sleep(tokio::time::Duration::from_secs(secs)).await;
}
Err(e) => return Err(e),
}
}
f().await
}
pub async fn ping(