feat(health): add kademlia bootstrap healthcheck + fix test span logg…

…ing (#2056)
fluencelabs · Feb 5, 2024 · c082231 · c082231
1 parent d3e4855
commit c082231
Show file tree

Hide file tree

Showing 12 changed files with 288 additions and 184 deletions.
diff --git a/aquamarine/src/vm_pool.rs b/aquamarine/src/vm_pool.rs
@@ -162,6 +162,7 @@ impl<RT: AquaRuntime> VmPool<RT> {
     pub fn poll(&mut self, cx: &mut Context<'_>) {
         let creating_vms = match &mut self.creating_runtimes {
             None => {
+                tracing::debug!("Starting creation {} AVMs", self.pool_size);
                 self.creating_runtimes = Some(
                     (0..self.pool_size)
                         .map(|id| (id, self.create_avm(cx)))

diff --git a/crates/connected-client/src/connected_client.rs b/crates/connected-client/src/connected_client.rs
@@ -28,7 +28,7 @@ use particle_protocol::Particle;
 use serde_json::{Value as JValue, Value};
 use tempfile::TempDir;
 use test_constants::{
-    IDLE_CONNECTION_TIMEOUT, KAD_TIMEOUT, PARTICLE_TTL, SHORT_TIMEOUT, TIMEOUT, TRANSPORT_TIMEOUT,
+    IDLE_CONNECTION_TIMEOUT, PARTICLE_TTL, SHORT_TIMEOUT, TIMEOUT, TRANSPORT_TIMEOUT,
 };
 
 use crate::client::Client;
@@ -43,7 +43,6 @@ pub struct ConnectedClient {
     pub node_address: Multiaddr,
     pub timeout: Duration,
     pub short_timeout: Duration,
-    pub kad_timeout: Duration,
     pub local_vm: tokio::sync::OnceCell<tokio::sync::Mutex<AVM>>,
     pub data_store: Arc<ParticleDataStore>,
     pub particle_ttl: Duration,
@@ -59,10 +58,6 @@ impl ConnectedClient {
         self.short_timeout
     }
 
-    pub fn kad_timeout(&self) -> Duration {
-        self.kad_timeout
-    }
-
     pub fn particle_ttl(&self) -> Duration {
         self.particle_ttl
     }
@@ -201,7 +196,6 @@ impl ConnectedClient {
             node_address,
             timeout: TIMEOUT,
             short_timeout: SHORT_TIMEOUT,
-            kad_timeout: KAD_TIMEOUT,
             local_vm,
             data_store,
             particle_ttl: particle_ttl.unwrap_or(Duration::from_millis(PARTICLE_TTL as u64)),