absolutebi/serai
1
0
Fork 0
mirror of https://github.com/serai-dex/serai.git synced 2025-12-08 12:19:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Track total peer count in the coordinator

Browse Source
This commit is contained in:
Luke Parker
2024-03-23 18:02:48 -04:00
parent 5ea3b1bf97
commit b7d49af1d5
1 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
coordinator/src/p2p.rs
View File

@@ -394,6 +394,7 @@ impl LibP2p {
async move { async move {
let mut set_for_genesis = HashMap::new(); let mut set_for_genesis = HashMap::new();
let mut connected_peers = 0;
loop { loop {
let time_since_last = Instant::now().duration_since(time_of_last_p2p_message); let time_since_last = Instant::now().duration_since(time_of_last_p2p_message);
tokio::select! { tokio::select! {
@@ -437,15 +438,25 @@ impl LibP2p {
} }
Some(SwarmEvent::ConnectionEstablished { peer_id, connection_id, .. }) => { Some(SwarmEvent::ConnectionEstablished { peer_id, connection_id, .. }) => {
if &peer_id == swarm.local_peer_id() { if &peer_id == swarm.local_peer_id() {
log::warn!("established a libp2p connection to ourselves");
swarm.close_connection(connection_id); swarm.close_connection(connection_id);
} else if swarm.is_connected(&peer_id) {} else { continue;
}
connected_peers += 1;
log::debug!( log::debug!(
"connection established to peer {} in connection ID {}", "connection established to peer {} in connection ID {}, connected peers: {}",
&peer_id, &peer_id,
&connection_id, &connection_id,
connected_peers,
); );
swarm.behaviour_mut().gossipsub.add_explicit_peer(&peer_id)
} }
Some(SwarmEvent::ConnectionClosed { peer_id, .. }) => {
connected_peers -= 1;
log::debug!(
"connection with peer {peer_id} closed, connected peers: {}",
connected_peers,
);
} }
Some(SwarmEvent::Behaviour(BehaviorEvent::Gossipsub( Some(SwarmEvent::Behaviour(BehaviorEvent::Gossipsub(
GsEvent::Message { propagation_source, message, .. }, GsEvent::Message { propagation_source, message, .. },