Ecosystem/serai
Ecosystem/serai
1
0
Fork 0
mirror of https://github.com/serai-dex/serai.git synced 2025-01-03 09:29:46 +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 // TODO: We should also connect to random peers from random nets as needed for
// cosigning // cosigning
// Define a buffer, `to_retry`, so we can exhaust this channel before sending more down // Drain the chainnel, de-duplicating any networks in it
// it let mut connect_to_network_networks = HashSet::new();
let mut to_retry = vec![];
while let Some(network) = connect_to_network_recv.recv().await { 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 let Ok(mut nodes) = serai.p2p_validators(network).await {
// If there's an insufficient amount of nodes known, connect to all yet add it // If there's an insufficient amount of nodes known, connect to all yet add it
// back and break // back and break
@ -480,7 +482,8 @@ impl LibP2p {
network, network,
nodes.len() nodes.len()
); );
to_retry.push(network); // Retry this later
connect_to_network_send.send(network).unwrap();
for node in nodes { for node in nodes {
connect(network, node).await; 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 // Sleep 60 seconds before moving to the next iteration
tokio::time::sleep(core::time::Duration::from_secs(60)).await; tokio::time::sleep(core::time::Duration::from_secs(60)).await;
} }