Total runtime timeout, fix saves

[dnsseed-rust] / src / printer.rs

diff --git a/src/printer.rs b/src/printer.rs
index f51fee55bb116b9e49dd9c5b4f9273061b9a291e..4818d9af61ce4076dddc53de9a349e7cc3051c9f 100644 (file)
--- a/src/printer.rs
+++ b/src/printer.rs
@@ -34,14 +34,14 @@ impl Printer {
                std::thread::spawn(move || {
                        loop {
                                std::thread::sleep(std::time::Duration::from_secs(1));
-                               if START_SHUTDOWN.load(Ordering::Relaxed) {
-                                       break;
-                               }
 
                                let stdout = std::io::stdout();
                                let mut out = stdout.lock();
 
                                let stats = thread_arc.lock().unwrap();
+                               if START_SHUTDOWN.load(Ordering::Relaxed) && stats.connection_count == 0 {
+                                       break;
+                               }
 
                                out.write_all(b"\x1b[2J\x1b[;H\n").expect("stdout broken?");
                                for line in stats.lines.iter() {