From: Matt Corallo <git@bluematt.me>
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?p=dnsseed-rust;a=commitdiff_plain;h=c49b9c568cbde2bb7d24726deaa66da11a9d8aab

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<u32>,
@@ -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<String>,
 	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,
 		}
 	}
 }