From e28c33f1e0ceae401ec4c1f40c5aa094fbd58349 Mon Sep 17 00:00:00 2001 From: Kushagra Udai Date: Sat, 23 Nov 2024 00:19:37 -0800 Subject: [PATCH 1/2] Fix metrics being emitted more than once. --- CHANGELOG.md | 4 ++++ Cargo.toml | 2 +- src/lib.rs | 10 +++++----- src/metrics.rs | 14 +++++++++----- 4 files changed, 19 insertions(+), 11 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 7141b3c..d410e8a 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -6,6 +6,10 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), ## Unreleased +## 0.8.2 - 2024-11-22 + +- Fix metrics being emitted more than once. + ## 0.8.1 - 2024-11-20 ### Changed diff --git a/Cargo.toml b/Cargo.toml index 3a7ca18..d94393f 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -1,6 +1,6 @@ [package] name = "rs-consul" -version = "0.8.1" +version = "0.8.2" authors = ["Roblox"] edition = "2021" description = "This crate provides access to a set of strongly typed apis to interact with consul (https://www.consul.io/)" diff --git a/src/lib.rs b/src/lib.rs index ba649ea..bea4f19 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -420,7 +420,7 @@ impl Consul { #[cfg(feature = "metrics")] { metrics_info_wrapper.set_status(code); - drop(metrics_info_wrapper.clone()); + metrics_info_wrapper.emit_metrics(); } ConsulError::UnexpectedResponseCode( code, @@ -438,7 +438,7 @@ impl Consul { #[cfg(feature = "metrics")] { metrics_info_wrapper.set_status(StatusCode::OK); - drop(metrics_info_wrapper.clone()); + metrics_info_wrapper.emit_metrics(); } return Ok(response); } @@ -822,7 +822,7 @@ impl Consul { #[cfg(feature = "metrics")] { metrics_info_wrapper.set_status(StatusCode::REQUEST_TIMEOUT); - drop(metrics_info_wrapper.clone()); + metrics_info_wrapper.emit_metrics(); } Err(ConsulError::TimeoutExceeded(dur)) } @@ -833,7 +833,7 @@ impl Consul { let response = response.inspect_err(|_| { #[cfg(feature = "metrics")] - drop(metrics_info_wrapper.clone()); + metrics_info_wrapper.emit_metrics(); })?; #[cfg(feature = "trace")] @@ -844,7 +844,7 @@ impl Consul { #[cfg(feature = "metrics")] { metrics_info_wrapper.set_status(status); - drop(metrics_info_wrapper); + metrics_info_wrapper.emit_metrics(); } let mut response_body = response diff --git a/src/metrics.rs b/src/metrics.rs index abfa60f..27507a8 100644 --- a/src/metrics.rs +++ b/src/metrics.rs @@ -65,16 +65,20 @@ impl MetricInfoWrapper { pub fn set_status(&mut self, status: StatusCode) { self.metrics.status = Some(status); } + + pub fn emit_metrics(&mut self) { + if let Some(sender) = self.sender.take() { + let mut metrics = self.metrics; + metrics.duration = Some(self.start.elapsed()); + let _ = sender.send(metrics); + } + } } #[cfg(feature = "metrics")] impl Drop for MetricInfoWrapper { fn drop(&mut self) { - if let Some(sender) = self.sender.take() { - let mut metrics = self.metrics; - metrics.duration = Some(self.start.elapsed()); - let _ = sender.send(self.metrics); - } + self.emit_metrics(); } } From 17350f29dd44e18d5b6eb5120ba12b938c500efd Mon Sep 17 00:00:00 2001 From: GitHub Action Bot Date: Sat, 23 Nov 2024 08:20:29 +0000 Subject: [PATCH 2/2] Commit from GitHub Actions (Format Code) --- src/metrics.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/metrics.rs b/src/metrics.rs index 27507a8..45fbe89 100644 --- a/src/metrics.rs +++ b/src/metrics.rs @@ -65,7 +65,7 @@ impl MetricInfoWrapper { pub fn set_status(&mut self, status: StatusCode) { self.metrics.status = Some(status); } - + pub fn emit_metrics(&mut self) { if let Some(sender) = self.sender.take() { let mut metrics = self.metrics;