From 89e17b0ffabdf6ec5cf30d89eac7bbfe6e6fa14c Mon Sep 17 00:00:00 2001
From: Jedrzej Stuczynski <jedrzej.stuczynski@gmail.com>
Date: Thu, 9 Jan 2020 16:30:56 +0000
Subject: [PATCH] Added number of test packets to healthcheck config

---
 validator-config.toml                       | 3 ++-
 validator/src/validator/config/mod.rs       | 3 +++
 validator/src/validator/health_check/mod.rs | 7 ++++++-
 3 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/validator-config.toml b/validator-config.toml
index 68226e268c..96c6ed6951 100644
--- a/validator-config.toml
+++ b/validator-config.toml
@@ -1,4 +1,5 @@
 [healthcheck]
 
 directory-server = "http://localhost:8080"
-interval = 10.0
\ No newline at end of file
+interval = 10.0
+test-packets-per-node = 2
\ No newline at end of file
diff --git a/validator/src/validator/config/mod.rs b/validator/src/validator/config/mod.rs
index d5a636616b..7e25d24c13 100644
--- a/validator/src/validator/config/mod.rs
+++ b/validator/src/validator/config/mod.rs
@@ -12,4 +12,7 @@ pub struct HealthCheck {
     pub directory_server: String,
 
     pub interval: f64, // in seconds
+
+    #[serde(rename(deserialize = "test-packets-per-node"))]
+    pub num_test_packets: usize,
 }
diff --git a/validator/src/validator/health_check/mod.rs b/validator/src/validator/health_check/mod.rs
index 206e5ab2ca..40a73944d3 100644
--- a/validator/src/validator/health_check/mod.rs
+++ b/validator/src/validator/health_check/mod.rs
@@ -130,6 +130,7 @@ impl std::fmt::Display for NodeScore {
 pub(crate) struct HealthChecker {
     directory_client: directory_client::Client,
     interval: Duration,
+    num_test_packets: usize,
 }
 
 impl HealthChecker {
@@ -142,6 +143,7 @@ impl HealthChecker {
         HealthChecker {
             directory_client: directory_client::Client::new(directory_client_config),
             interval: Duration::from_secs_f64(config.interval),
+            num_test_packets: config.num_test_packets,
         }
     }
 
@@ -166,7 +168,10 @@ impl HealthChecker {
     }
 
     pub async fn run(self) -> Result<(), HealthCheckerError> {
-        debug!("healthcheck will run every {:?}", self.interval,);
+        debug!(
+            "healthcheck will run every {:?} and will send {} packets to each node",
+            self.interval, self.num_test_packets
+        );
 
         loop {
             match self.do_check() {