Cuprate/cuprate
Cuprate/cuprate
1
0
Fork 0
mirror of https://github.com/Cuprate/cuprate.git synced 2025-01-22 10:44:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

use JoinSet in connection maintainer

Browse source
This commit is contained in:
Boog900 2024-05-13 23:39:36 +01:00
parent 37e7c29c48
commit e59bb0a1f5
No known key found for this signature in database
GPG key ID: 42AB1287CB0041C2
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
p2p/cuprate-p2p/src/connection_maintainer.rs
View file

@ -5,10 +5,10 @@
//! no connected peer has.
use std::sync::Arc;
use futures::{stream::FuturesUnordered, StreamExt};
use rand::{distributions::Bernoulli, prelude::*};
use tokio::{
sync::{mpsc, OwnedSemaphorePermit, Semaphore},
task::JoinSet,
time::{sleep, timeout},
};
use tower::{Service, ServiceExt};
@ -111,7 +111,7 @@ where
let mut allowed_errors = seeds.len();
let mut handshake_futs = FuturesUnordered::new();
let mut handshake_futs = JoinSet::new();
for seed in seeds {
info!("Getting peers from seed node: {}", seed);
@ -131,10 +131,10 @@ where
}),
);
// Spawn the handshake on a separate task with a timeout, so we don't get stuck connecting to a peer.
handshake_futs.push(tokio::spawn(fut));
handshake_futs.spawn(fut);
}
while let Some(res) = handshake_futs.next().await {
while let Some(res) = handshake_futs.join_next().await {
if matches!(res, Err(_) | Ok(Err(_)) | Ok(Ok(Err(_)))) {
allowed_errors -= 1;
}
@ -231,7 +231,7 @@ where
return Ok(());
}
tracing::debug!("Permit available making outbound connection.");
tracing::debug!("Permit available, making outbound connection.");
let req = if self.peer_type_gen.sample(&mut thread_rng()) {
AddressBookRequest::TakeRandomGrayPeer { height: None }