From 4b7d0dd3a86208f40c351fc602aeb7b9b7c114d1 Mon Sep 17 00:00:00 2001
From: ramiroaisen <52116153+ramiroaisen@users.noreply.github.com>
Date: Mon, 17 Jun 2024 21:05:42 -0300
Subject: [PATCH] fix: remove unnecesary receiver field from GracefulShutdown

---
 src/server/graceful.rs | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/src/server/graceful.rs b/src/server/graceful.rs
index d649ac1..4616bc4 100644
--- a/src/server/graceful.rs
+++ b/src/server/graceful.rs
@@ -19,19 +19,18 @@ use tokio::sync::watch;
 /// A graceful shutdown utility
 pub struct GracefulShutdown {
     tx: watch::Sender<()>,
-    rx: watch::Receiver<()>,
 }
 
 impl GracefulShutdown {
     /// Create a new graceful shutdown helper.
     pub fn new() -> Self {
-        let (tx, rx) = watch::channel(());
-        Self { tx, rx }
+        let (tx, _) = watch::channel(());
+        Self { tx }
     }
 
     /// Wrap a future for graceful shutdown watching.
     pub fn watch<C: GracefulConnection>(&self, conn: C) -> impl Future<Output = C::Output> {
-        let mut rx = self.rx.clone();
+        let mut rx = self.tx.subscribe();
         GracefulConnectionFuture::new(conn, async move {
             let _ = rx.changed().await;
             // hold onto the rx until the watched future is completed
@@ -45,8 +44,7 @@ impl GracefulShutdown {
     /// connections have shutdown.
     pub async fn shutdown(self) {
         // drop the rx immediately, or else it will hold us up
-        let Self { tx, rx } = self;
-        drop(rx);
+        let Self { tx } = self;
 
         // signal all the watched futures about the change
         let _ = tx.send(());