dns request

metalbear-co · Dec 30, 2024 · bb063c9 · bb063c9
1 parent a4c9ff7
commit bb063c9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/mirrord/agent/src/dns.rs b/mirrord/agent/src/dns.rs
@@ -18,6 +18,7 @@ use tracing::Level;
 
 use crate::{
     error::{AgentError, AgentResult},
+    metrics::DNS_REQUEST_COUNT,
     watched_task::TaskStatus,
 };
 
@@ -187,6 +188,7 @@ impl DnsApi {
         }
 
         self.responses.push_back(response_rx);
+        DNS_REQUEST_COUNT.inc();
 
         Ok(())
     }
@@ -209,6 +211,8 @@ impl DnsApi {
             }),
         });
 
+        DNS_REQUEST_COUNT.dec();
+
         Ok(GetAddrInfoResponse(response))
     }
 }
diff --git a/mirrord/agent/src/metrics.rs b/mirrord/agent/src/metrics.rs
@@ -17,6 +17,14 @@ pub(crate) static CLIENT_COUNT: LazyLock<IntGauge> = LazyLock::new(|| {
     .expect("Valid at initialization!")
 });
 
+pub(crate) static DNS_REQUEST_COUNT: LazyLock<IntGauge> = LazyLock::new(|| {
+    register_int_gauge!(
+        "mirrord_agent_dns_request_count",
+        "amount of in-progress dns requests in the mirrord-agent"
+    )
+    .expect("Valid at initialization!")
+});
+
 pub(crate) static OPEN_FD_COUNT: LazyLock<IntGauge> = LazyLock::new(|| {
     register_int_gauge!(
         "mirrord_agent_open_fd_count",