Use AddressState utils more

[dnsseed-rust] / src / datastore.rs

diff --git a/src/datastore.rs b/src/datastore.rs
index c56a86d7b84eebf927c1ad6126dd19f8eda780bf..4782027e8736eb78c6956ab011f7b1dae690b368 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)]