From 22545d336097441d9dccd8a995450c44144b524b Mon Sep 17 00:00:00 2001
From: Lijun Wang <83639177+lijunwangs@users.noreply.github.com>
Date: Sat, 11 Jan 2025 13:50:25 -0800
Subject: [PATCH 1/2] Update metrics before breaking connections

---
 streamer/src/nonblocking/quic.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/streamer/src/nonblocking/quic.rs b/streamer/src/nonblocking/quic.rs
index aca98bb4b45996..3e8457aa6b044b 100644
--- a/streamer/src/nonblocking/quic.rs
+++ b/streamer/src/nonblocking/quic.rs
@@ -1167,6 +1167,8 @@ async fn handle_connection(
                         CONNECTION_CLOSE_CODE_INVALID_STREAM.into(),
                         CONNECTION_CLOSE_REASON_INVALID_STREAM,
                     );
+                    stats.total_streams.fetch_sub(1, Ordering::Relaxed);
+                    stream_load_ema.update_ema_if_needed();            
                     break 'conn;
                 }
             }

From 904d5f5f05cab83806b9ebde013468493edf7a25 Mon Sep 17 00:00:00 2001
From: Lijun Wang <83639177+lijunwangs@users.noreply.github.com>
Date: Sat, 11 Jan 2025 13:55:57 -0800
Subject: [PATCH 2/2] Update metrics before breaking connections

---
 streamer/src/nonblocking/quic.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/streamer/src/nonblocking/quic.rs b/streamer/src/nonblocking/quic.rs
index 3e8457aa6b044b..d2087a77e2987b 100644
--- a/streamer/src/nonblocking/quic.rs
+++ b/streamer/src/nonblocking/quic.rs
@@ -1168,7 +1168,7 @@ async fn handle_connection(
                         CONNECTION_CLOSE_REASON_INVALID_STREAM,
                     );
                     stats.total_streams.fetch_sub(1, Ordering::Relaxed);
-                    stream_load_ema.update_ema_if_needed();            
+                    stream_load_ema.update_ema_if_needed();
                     break 'conn;
                 }
             }