Implement shutdown, input reading

[dnsseed-rust] / src / printer.rs

diff --git a/src/printer.rs b/src/printer.rs
index a23d023cb2d2457ce895643c6a0f84471e16a163..f51fee55bb116b9e49dd9c5b4f9273061b9a291e 100644 (file)
--- a/src/printer.rs
+++ b/src/printer.rs
@@ -1,9 +1,12 @@
+use std::sync::atomic::Ordering;
 use std::collections::LinkedList;
 use std::sync::{Arc, Mutex};
 use std::io::Write;
 
 use crate::datastore::{Store, AddressState, U64Setting, RegexSetting};
 
+use crate::START_SHUTDOWN;
+
 pub enum Stat {
        HeaderCount(u64),
        NewConnection,
@@ -31,6 +34,9 @@ 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();