657 fast ping on startup

CHANGELOG.md

@@ -8,6 +8,7 @@ Bob versions changelog
 - Include bobc into release builds (#569)
 - Added blob-info and index-info features to brt (#356)
 - Support for files, file name patterns, key ranges and a 'exists' subcommand to 'bobc' (#539)
+- Fast ping at the start (#657)
 
 #### Changed
 - Change locks to sync where possible (#472)

bob/Cargo.toml

@@ -43,6 +43,7 @@ tonic = { version = "0.6", features = ["prost"] }
 tower = "0.4"
 tower-service = "0.3"
 uuid = "1.1.2"
+coarsetime = "0.1.22"
 
 [dependencies.pearl]
 version = "0.15.0"

bob/src/link_manager.rs

@@ -2,6 +2,9 @@ use crate::prelude::*;
 
 use termion::color;
 
+const FAST_PING_PERIOD_MS: u64 = 100;
+const FAST_PING_DURATION_SEC: u64 = 60;
+
 #[derive(Debug)]
 pub(crate) struct LinkManager {
     nodes: Arc<[Node]>,
@@ -21,28 +24,63 @@ impl LinkManager {
     }
 
     async fn checker_task(factory: Factory, nodes: Arc<[Node]>, period: Duration) {
+        let now = coarsetime::Clock::now_since_epoch().as_secs();
+        Self::checker(
+            &factory,
+            &nodes,
+            Duration::from_millis(FAST_PING_PERIOD_MS),
+            || {
+                let ts = coarsetime::Clock::now_since_epoch().as_secs();
+                ts > now && ts - now > FAST_PING_DURATION_SEC
+            },
+            period.as_millis() as usize / FAST_PING_PERIOD_MS as usize,
+        )
+        .await;
+        Self::checker(&factory, &nodes, period, || false, 1).await;
+    }
+
+    async fn checker(
+        factory: &Factory,
+        nodes: &[Node],
+        period: Duration,
+        should_stop: impl Fn() -> bool,
+        log_iteration_div: usize,
+    ) {
         let mut interval = interval(period);
-        loop {
+        let mut i = 1;
+        while !should_stop() {
+            let log = i % log_iteration_div == 0;
+            if log {
+                i = 1;
+            } else {
+                i += 1;
+            }
             interval.tick().await;
             let mut err_cnt = 0;
             let mut status = String::from("Node status: ");
             for node in nodes.iter() {
                 if let Err(e) = node.check(&factory).await {
-                    error!(
-                        "No connection to {}:[{}] - {}",
-                        node.name(),
-                        node.address(),
-                        e
-                    );
-                    status += &format!("{}{:<10} ", color::Fg(color::Red), node.name());
+                    if log {
+                        error!(
+                            "No connection to {}:[{}] - {}",
+                            node.name(),
+                            node.address(),
+                            e
+                        );
+                        status += &format!("{}{:<10} ", color::Fg(color::Red), node.name());
+                    }
                     err_cnt += 1;
                 } else {
-                    status += &format!("{}{:<10} ", color::Fg(color::Green), node.name());
+                    if log {
+                        status += &format!("{}{:<10} ", color::Fg(color::Green), node.name());
+                    }
                 }
             }
-            info!("{}{}", status, color::Fg(color::Reset));
-            let cnt = nodes.len() - err_cnt;
-            gauge!(AVAILABLE_NODES_COUNT, cnt as f64);
+            if log {
+                info!("{}{}", status, color::Fg(color::Reset));
+                let cnt = nodes.len() - err_cnt;
+                gauge!(AVAILABLE_NODES_COUNT, cnt as f64);
+            }
         }
     }