From: Matt Corallo Date: Fri, 23 Aug 2019 04:50:24 +0000 (-0400) Subject: Print stats about routing table size in UI X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=commitdiff_plain;h=c49b9c568cbde2bb7d24726deaa66da11a9d8aab;p=dnsseed-rust Print stats about routing table size in UI --- diff --git a/src/bgp_client.rs b/src/bgp_client.rs index 1fa9250..ed07bdc 100644 --- a/src/bgp_client.rs +++ b/src/bgp_client.rs @@ -20,7 +20,7 @@ use tokio::timer::Delay; use futures::sync::mpsc; -use crate::printer::Printer; +use crate::printer::{Printer, Stat}; struct Route { path: Vec, @@ -287,6 +287,8 @@ impl BGPClient { route_table.announce(r, Arc::clone(&path_arc)); } } + printer.set_stat(Stat::V4RoutingTableSize(route_table.v4_table.len())); + printer.set_stat(Stat::V6RoutingTableSize(route_table.v6_table.len())); }, _ => {} } diff --git a/src/printer.rs b/src/printer.rs index 0bea277..933603c 100644 --- a/src/printer.rs +++ b/src/printer.rs @@ -11,12 +11,16 @@ pub enum Stat { HeaderCount(u64), NewConnection, ConnectionClosed, + V4RoutingTableSize(usize), + V6RoutingTableSize(usize), } struct Stats { lines: LinkedList, header_count: u64, connection_count: u64, + v4_table_size: usize, + v6_table_size: usize, } pub struct Printer { @@ -29,6 +33,8 @@ impl Printer { lines: LinkedList::new(), header_count: 0, connection_count: 0, + v4_table_size: 0, + v6_table_size: 0, })); let thread_arc = Arc::clone(&stats); std::thread::spawn(move || { @@ -83,6 +89,10 @@ impl Printer { ).as_bytes()).expect("stdout broken?"); } + out.write_all(format!( + "\nBGP Routing Table: {} v4 paths, {} v6 paths\n", + stats.v4_table_size, stats.v6_table_size).as_bytes()).expect("stdout broken?"); + out.write_all(b"\nCommands:\n").expect("stdout broken?"); out.write_all(b"q: quit\n").expect("stdout broken?"); out.write_all(format!( @@ -124,6 +134,8 @@ impl Printer { Stat::HeaderCount(c) => self.stats.lock().unwrap().header_count = c, Stat::NewConnection => self.stats.lock().unwrap().connection_count += 1, Stat::ConnectionClosed => self.stats.lock().unwrap().connection_count -= 1, + Stat::V4RoutingTableSize(c) => self.stats.lock().unwrap().v4_table_size = c, + Stat::V6RoutingTableSize(c) => self.stats.lock().unwrap().v6_table_size = c, } } }