change heartbeat deadline duration

Author: erhant

compute/src/node/core.rs

@@ -8,7 +8,7 @@ use eyre::{eyre, Result};
 use std::time::Duration;
 use tokio_util::sync::CancellationToken;
 
-use crate::DriaComputeNode;
+use crate::{reqres::HeartbeatRequester, DriaComputeNode};
 
 impl DriaComputeNode {
     /// Runs the main loop of the compute node.
@@ -18,8 +18,6 @@ impl DriaComputeNode {
         const DIAGNOSTIC_REFRESH_INTERVAL_SECS: Duration = Duration::from_secs(45);
         /// Duration between refreshing the available nodes.
         const RPC_LIVENESS_REFRESH_INTERVAL_SECS: Duration = Duration::from_secs(2 * 60);
-        /// Duration between each heartbeat sent to the RPC.
-        const HEARTBEAT_INTERVAL_SECS: Duration = Duration::from_secs(60);
         /// Duration between each specs update sent to the RPC.
         const SPECS_INTERVAL_SECS: Duration = Duration::from_secs(60 * 5);
 
@@ -31,7 +29,7 @@ impl DriaComputeNode {
         rpc_liveness_refresh_interval.tick().await; // move each one tick
 
         // move one tick, and wait at least a third of the diagnostics
-        let mut heartbeat_interval = tokio::time::interval(HEARTBEAT_INTERVAL_SECS);
+        let mut heartbeat_interval = tokio::time::interval(HeartbeatRequester::HEARTBEAT_DEADLINE);
         heartbeat_interval.tick().await;
         heartbeat_interval.reset_after(DIAGNOSTIC_REFRESH_INTERVAL_SECS / 3);
 

compute/src/node/diagnostic.rs

@@ -28,7 +28,7 @@ impl DriaComputeNode {
                 "$DRIA Points: {} total, {} earned in this run, within top {}%",
                 steps.score,
                 earned,
-                steps.percentile.unwrap_or("100%".to_string())
+                steps.percentile.unwrap_or("100".to_string())
             ));
         }
 

compute/src/reqres/heartbeat.rs

@@ -19,16 +19,15 @@ impl IsResponder for HeartbeatRequester {
     type Response = HeartbeatResponse;
 }
 
-/// Any acknowledged heartbeat that is older than this duration is considered dead.
-const HEARTBEAT_DEADLINE_SECS: Duration = Duration::from_secs(20);
-
 impl HeartbeatRequester {
+    /// Any acknowledged heartbeat that is older than this duration is considered dead.
+    pub const HEARTBEAT_DEADLINE: Duration = Duration::from_secs(60);
     pub(crate) async fn send_heartbeat(
         node: &mut DriaComputeNode,
         peer_id: PeerId,
     ) -> Result<OutboundRequestId> {
         let uuid = Uuid::now_v7();
-        let deadline = chrono::Utc::now() + HEARTBEAT_DEADLINE_SECS;
+        let deadline = chrono::Utc::now() + Self::HEARTBEAT_DEADLINE;
 
         let heartbeat_request = HeartbeatRequest {
             heartbeat_id: uuid,