aljaz/cdk
1
0
Fork 0
mirror of https://github.com/aljazceru/cdk.git synced 2026-02-23 14:06:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

Do not fallback to HTTP on first error (#1061)

Browse Source 
Instead use the same failure counter.

Fixes #496
This commit is contained in:
C
2025-09-10 06:57:58 -03:00
committed by GitHub
parent 21b4080810
commit 053a3a7e43
1 changed files with 21 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
crates/cdk/src/wallet/subscription/ws.rs
View File

@@ -69,8 +69,6 @@ pub async fn ws_main(
};
tracing::debug!("Connected to {}", url);
failure_count = 0;
let (mut write, mut read) = ws_stream.split();
let req_id = AtomicUsize::new(0);
@@ -147,23 +145,31 @@ pub async fn ws_main(
WsMessageOrResponse::Response(response) => {
tracing::debug!("Received response from server: {:?}", response);
subscription_requests.remove(&response.id);
// reset connection failure after a successful response from the serer
failure_count = 0;
}
WsMessageOrResponse::ErrorResponse(error) => {
tracing::error!("Received error from server: {:?}", error);
if subscription_requests.contains(&error.id) {
tracing::error!(
"Falling back to HTTP client"
);
return http_main(
active_subscriptions.into_keys(),
http_client,
subscriptions,
new_subscription_recv,
on_drop,
wallet,
)
.await;
if subscription_requests.contains(&error.id) {
failure_count += 1;
if failure_count > MAX_ATTEMPT_FALLBACK_HTTP {
tracing::error!(
"Falling back to HTTP client"
);
return http_main(
active_subscriptions.into_keys(),
http_client,
subscriptions,
new_subscription_recv,
on_drop,
wallet,
)
.await;
}
break; // break connection to force a reconnection, to attempt to recover form this error
}
}
}