Better log messages

[dnsseed-rust] / src / datastore.rs

diff --git a/src/datastore.rs b/src/datastore.rs
index c56a86d7b84eebf927c1ad6126dd19f8eda780bf..ba984d23ed167e96d84f786b1469ec73f1ddcd3f 100644 (file)
--- a/src/datastore.rs
+++ b/src/datastore.rs
@@ -80,6 +80,10 @@ impl AddressState {
                        AddressState::WasGood => "Was Good",
                }
        }
+
+       pub fn get_count() -> u8 {
+               11
+       }
 }
 
 #[derive(Hash, PartialEq, Eq)]
@@ -311,11 +315,11 @@ impl Store {
                }
        }
 
-       pub fn set_node_state(&self, addr: SocketAddr, state: AddressState, services: u64) -> bool {
+       pub fn set_node_state(&self, addr: SocketAddr, state: AddressState, services: u64) -> AddressState {
                let mut nodes_lock = self.nodes.write().unwrap();
                let nodes = nodes_lock.borrow_mut();
                let state_ref = nodes.nodes_to_state.get_mut(&addr).unwrap();
-               let ret = state != state_ref.state;
+               let ret = state_ref.state;
                let now = Instant::now();
                if (state_ref.state == AddressState::Good || state_ref.state == AddressState::WasGood)
                                && state != AddressState::Good