fix(p2p): rework ticks of bootstrap query interval

Signed-off-by: Simon Paitrault <[email protected]>
topos-protocol · Mar 21, 2024 · 369cfb8 · 369cfb8
1 parent b2a1af0
commit 369cfb8
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/crates/topos-p2p/src/behaviour/discovery.rs b/crates/topos-p2p/src/behaviour/discovery.rs
@@ -80,7 +80,10 @@ impl DiscoveryBehaviour {
             next_bootstrap_query: if known_peers.is_empty() {
                 None
             } else {
-                Some(Box::pin(tokio::time::interval(config.bootstrap_interval)))
+                let mut interval = tokio::time::interval(config.bootstrap_interval);
+                interval.set_missed_tick_behavior(tokio::time::MissedTickBehavior::Delay);
+
+                Some(Box::pin(interval))
             },
             health_status: if known_peers.is_empty() {
                 HealthStatus::Healthy
@@ -105,9 +108,14 @@ impl DiscoveryBehaviour {
     }
 
     /// Change the interval of the next bootstrap queries
-    pub fn change_interval(&mut self, duration: Duration) -> Result<(), P2PError> {
+    pub async fn change_interval(&mut self, duration: Duration) -> Result<(), P2PError> {
         if let Some(interval) = self.next_bootstrap_query.as_mut() {
-            interval.set(tokio::time::interval(duration));
+            let mut new_interval = tokio::time::interval(duration);
+            // Delay the next tick
+            new_interval.set_missed_tick_behavior(tokio::time::MissedTickBehavior::Delay);
+            // ignore first tick
+            _ = new_interval.tick().await;
+            interval.set(new_interval);
         }
 
         Ok(())

diff --git a/crates/topos-p2p/src/runtime/handle_event/discovery.rs b/crates/topos-p2p/src/runtime/handle_event/discovery.rs
@@ -53,7 +53,8 @@ impl EventHandler<Box<Event>> for Runtime {
                 behaviour.discovery.health_status = HealthStatus::Unhealthy;
                 _ = behaviour
                     .discovery
-                    .change_interval(self.config.discovery.fast_bootstrap_interval);
+                    .change_interval(self.config.discovery.fast_bootstrap_interval)
+                    .await;
             }
 
             Event::OutboundQueryProgressed {
@@ -115,7 +116,8 @@ impl EventHandler<Box<Event>> for Runtime {
                 behaviour.discovery.health_status = HealthStatus::Healthy;
                 _ = behaviour
                     .discovery
-                    .change_interval(self.config.discovery.bootstrap_interval);
+                    .change_interval(self.config.discovery.bootstrap_interval)
+                    .await;
             }
 
             Event::OutboundQueryProgressed {