Ecosystem/serai
Ecosystem/serai
1
0
Fork 0
mirror of https://github.com/serai-dex/serai.git synced 2024-12-31 07:59:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Use a HashSet for which networks to try peer finding for

Browse source 
Prevents a flood of retries from individually failed attempts within a batch of
peer connection attempts.
This commit is contained in:
Luke Parker 2024-04-23 10:55:52 -04:00
parent 95591218bb
commit 43dc036660
No known key found for this signature in database
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
coordinator/src/p2p.rs
View file

@ -467,10 +467,12 @@ impl LibP2p {
// TODO: We should also connect to random peers from random nets as needed for
// cosigning
// Define a buffer, `to_retry`, so we can exhaust this channel before sending more down
// it
let mut to_retry = vec![];
// Drain the chainnel, de-duplicating any networks in it
let mut connect_to_network_networks = HashSet::new();
while let Some(network) = connect_to_network_recv.recv().await {
connect_to_network_networks.insert(network);
}
for network in connect_to_network_networks {
if let Ok(mut nodes) = serai.p2p_validators(network).await {
// If there's an insufficient amount of nodes known, connect to all yet add it
// back and break
@ -480,7 +482,8 @@ impl LibP2p {
network,
nodes.len()
);
to_retry.push(network);
// Retry this later
connect_to_network_send.send(network).unwrap();
for node in nodes {
connect(network, node).await;
}
@ -499,9 +502,6 @@ impl LibP2p {
}
}
}
for to_retry in to_retry {
connect_to_network_send.send(to_retry).unwrap();
}
// Sleep 60 seconds before moving to the next iteration
tokio::time::sleep(core::time::Duration::from_secs(60)).await;
}