leyoda/rqbit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

DHT stats

Browse source
This commit is contained in:
Igor Katson 2021-07-14 00:48:53 +01:00
parent 4e31eb6547
commit fe4dcb226f
4 changed files with 140 additions and 101 deletions
Download patch file Download diff file Expand all files Collapse all files

32
crates/dht/src/main.rs
View file

@ -1,7 +1,8 @@
use std::{collections::HashSet, str::FromStr};
use std::{collections::HashSet, str::FromStr, time::Duration};
use anyhow::Context;
use dht::{Dht, Id20};
use log::info;
use tokio_stream::StreamExt;
#[tokio::main]
@ -12,11 +13,28 @@ async fn main() -> anyhow::Result<()> {
let dht = Dht::new().await.context("error initializing DHT")?;
let mut stream = dht.get_peers(info_hash).await?;
let mut seen = HashSet::new();
while let Some(peer) = stream.next().await {
let peer = peer.context("error reading peer stream")?;
if seen.insert(peer) {
log::info!("peer found: {}", peer)
let stats_printer = async move {
loop {
tokio::time::sleep(Duration::from_secs(5)).await;
info!("DHT stats: {:?}", dht.stats());
}
}
Ok(())
Ok::<_, anyhow::Error>(())
};
let peer_printer = async move {
while let Some(peer) = stream.next().await {
let peer = peer.context("error reading peer stream")?;
if seen.insert(peer) {
log::info!("peer found: {}", peer)
}
}
Ok(())
};
let res = tokio::select! {
res = stats_printer => res,
res = peer_printer => res,
};
res
}