From e28c33f1e0ceae401ec4c1f40c5aa094fbd58349 Mon Sep 17 00:00:00 2001
From: Kushagra Udai <kudai@roblox.com>
Date: Sat, 23 Nov 2024 00:19:37 -0800
Subject: [PATCH] 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();
     }
 }