logos-co/nomos-node
1
0
Fork 0
mirror of synced 2025-02-24 21:48:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add command to retrieve libp2p network info (#281)

Browse Source 
* 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::{ use nomos_libp2p::{
libp2p::{ libp2p::{
gossipsub::{self, Message}, gossipsub::{self, Message},
@ -6,10 +7,9 @@ use nomos_libp2p::{
BehaviourEvent, Swarm, SwarmConfig, SwarmEvent, BehaviourEvent, Swarm, SwarmConfig, SwarmEvent,
}; };
use overwatch_rs::{overwatch::handle::OverwatchHandle, services::state::NoState}; use overwatch_rs::{overwatch::handle::OverwatchHandle, services::state::NoState};
use serde::{Deserialize, Serialize};
use tokio::sync::{broadcast, mpsc}; use tokio::sync::{broadcast, mpsc};
use super::NetworkBackend;
macro_rules! log_error { macro_rules! log_error {
($e:expr) => { ($e:expr) => {
if let Err(e) = $e { if let Err(e) = $e {
@ -23,6 +23,14 @@ pub struct Libp2p {
commands_tx: mpsc::Sender<Command>, 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)] #[derive(Debug)]
pub enum EventKind { pub enum EventKind {
Message, Message,
@ -38,6 +46,7 @@ pub enum Command {
Broadcast { topic: Topic, message: Vec<u8> }, Broadcast { topic: Topic, message: Vec<u8> },
Subscribe(Topic), Subscribe(Topic),
Unsubscribe(Topic), Unsubscribe(Topic),
Info { reply: oneshot::Sender<Libp2pInfo> },
} }
pub type Topic = String; pub type Topic = String;
@ -129,6 +138,18 @@ impl NetworkBackend for Libp2p {
tracing::debug!("unsubscribing to topic: {topic}"); tracing::debug!("unsubscribing to topic: {topic}");
log_error!(swarm.unsubscribe(&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));
}
}; };
} }
} }