tsarevvv/rust-ipfs
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Return time since connection if no round trip yet

Browse Source 
Fixes #178
This commit is contained in:
Jack Maloney 2021-12-02 01:17:17 -06:00 committed by Joonas Koivunen
parent 2ca9704823
commit 828bfc4b75
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/p2p/swarm.rs
View File

@ -10,7 +10,7 @@ use libp2p::swarm::{
};
use std::collections::{hash_map::Entry, HashMap, HashSet, VecDeque};
use std::convert::{TryFrom, TryInto};
use std::time::Duration;
use std::time::{Duration, Instant};
/// A description of currently active connection.
#[derive(Clone, Debug, PartialEq, Eq)]
@ -51,6 +51,7 @@ pub struct SwarmApi {
connections: HashMap<MultiaddrWithoutPeerId, PeerId>,
roundtrip_times: HashMap<PeerId, Duration>,
connected_peers: HashMap<PeerId, Vec<MultiaddrWithoutPeerId>>,
connected_times: HashMap<PeerId, Instant>,
/// The connections which have been requested, but the swarm/network is yet to ask for
/// addresses; currently filled in the order of adding, with the default size of one.
@ -80,7 +81,13 @@ impl SwarmApi {
self.connected_peers
.iter()
.filter_map(move |(peer, conns)| {
let rtt = self.roundtrip_times.get(peer).cloned();
let rtt = self.roundtrip_times.get(peer).cloned().or_else(|| {
// If no rountrip time yet, just return time since connected.
// See: https://github.com/rs-ipfs/rust-ipfs/issues/178
self.connected_times.get(peer).and_then(|connected_time| {
Instant::now().checked_duration_since(*connected_time)
})
});
conns.first().map(|any| Connection {
addr: MultiaddrWithPeerId::from((any.clone(), *peer)),
@ -121,6 +128,9 @@ impl SwarmApi {
handler,
});
// store this for returning the time since connecting started before ping is available
self.connected_times.insert(addr.peer_id, Instant::now());
self.pending_addresses
.entry(addr.peer_id)
.or_insert_with(|| Vec::with_capacity(1))
@ -291,6 +301,8 @@ impl NetworkBehaviour for SwarmApi {
closed_addr
);
self.connected_times.remove(peer_id);
if let ConnectedPoint::Dialer { .. } = endpoint {
let addr = MultiaddrWithPeerId::from((closed_addr, peer_id.to_owned()));