Close recv stream after receiving block

[dnsseed-rust] / src / main.rs

diff --git a/src/main.rs b/src/main.rs
index 0f6b931c5057127e7668d4d2730a0da5d9df3b92..ae0bb5b833ce37510b099072bd97068ac7bac532 100644 (file)
--- a/src/main.rs
+++ b/src/main.rs
@@ -156,6 +156,7 @@ pub fn scan_node(scan_time: Instant, node: SocketAddr, manual: bool) {
                                                return future::err(());
                                        }
                                        check_set_flag!(recvd_block, "block");
+                                       return future::err(());
                                },
                                _ => {},
                        }
@@ -211,9 +212,11 @@ fn poll_dnsseeds() {
 
 fn scan_net() {
        tokio::spawn(future::lazy(|| {
+               let printer = unsafe { PRINTER.as_ref().unwrap() };
                let store = unsafe { DATA_STORE.as_ref().unwrap() };
 
                let mut scan_nodes = store.get_next_scan_nodes();
+               printer.add_line(format!("Got {} addresses to scan", scan_nodes.len()), false);
                let per_iter_time = Duration::from_millis(1000 / store.get_u64(U64Setting::ConnsPerSec));
                let start_time = Instant::now();
                let mut iter_time = start_time;