logos-co
/
nomos-node
mirror of
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add command to retrieve libp2p network info (#281) 

* Add command to retrieve libp2p network info

* fix field name

* fix
This commit is contained in:
Giacomo Pasini 2023-08-01 17:04:18 +02:00 committed by GitHub
parent fac42cd31d
commit c2ca46e6a8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 23 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
nomos-services/network/src/backends/libp2p.rs
View File

@ -1,3 +1,4 @@
use super::NetworkBackend;
use nomos_libp2p::{
libp2p::{
gossipsub::{self, Message},
@ -6,10 +7,9 @@ use nomos_libp2p::{
BehaviourEvent, Swarm, SwarmConfig, SwarmEvent,
};
use overwatch_rs::{overwatch::handle::OverwatchHandle, services::state::NoState};
use serde::{Deserialize, Serialize};
use tokio::sync::{broadcast, mpsc};
use super::NetworkBackend;
macro_rules! log_error {
($e:expr) => {
if let Err(e) = $e {
@ -23,6 +23,14 @@ pub struct Libp2p {
commands_tx: mpsc::Sender<Command>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Libp2pInfo {
pub listen_addresses: Vec<Multiaddr>,
pub n_peers: usize,
pub n_connections: u32,
pub n_pending_connections: u32,
}
#[derive(Debug)]
pub enum EventKind {
Message,
@ -38,6 +46,7 @@ pub enum Command {
Broadcast { topic: Topic, message: Vec<u8> },
Subscribe(Topic),
Unsubscribe(Topic),
Info { reply: oneshot::Sender<Libp2pInfo> },
}
pub type Topic = String;
@ -129,6 +138,18 @@ impl NetworkBackend for Libp2p {
tracing::debug!("unsubscribing to topic: {topic}");
log_error!(swarm.unsubscribe(&topic));
}
Command::Info { reply } => {
let swarm = swarm.swarm();
let network_info = swarm.network_info();
let counters = network_info.connection_counters();
let info = Libp2pInfo {
listen_addresses: swarm.listeners().cloned().collect(),
n_peers: network_info.num_peers(),
n_connections: counters.num_connections(),
n_pending_connections: counters.num_pending(),
};
log_error!(reply.send(info));
}
};
}
}