Implementing udp support partially...in theory. Still untested :)

src/shutdown.rs

@@ -11,39 +11,48 @@ use tokio::sync::broadcast;
 /// received or not.
 #[derive(Debug)]
 pub(crate) struct Shutdown {
-   /// `true` if the shutdown signal has been received
-   shutdown: bool,
+    /// `true` if the shutdown signal has been received
+    shutdown: bool,
 
-   /// The receive half of the channel used to listen for shutdown.
-   notify: broadcast::Receiver<()>,
+    /// The receive half of the channel used to listen for shutdown.
+    notify: broadcast::Receiver<()>,
 }
 
 impl Shutdown {
-   /// Create a new `Shutdown` backed by the given `broadcast::Receiver`.
-   pub(crate) fn new(notify: broadcast::Receiver<()>) -> Shutdown {
-      Shutdown {
-         shutdown: false,
-         notify,
-      }
-   }
+    /// Create a new `Shutdown` backed by the given `broadcast::Receiver`.
+    pub(crate) fn new(notify: broadcast::Receiver<()>) -> Shutdown {
+        Shutdown {
+            shutdown: false,
+            notify,
+        }
+    }
 
-   /// Returns `true` if the shutdown signal has been received.
-   // pub(crate) fn is_shutdown(&self) -> bool {
-   //    self.shutdown
-   // }
+    /// Returns `true` if the shutdown signal has been received.
+    // pub(crate) fn is_shutdown(&self) -> bool {
+    //    self.shutdown
+    // }
 
-   /// Receive the shutdown notice, waiting if necessary.
-   pub(crate) async fn recv(&mut self) {
-      // If the shutdown signal has already been received, then return
-      // immediately.
-      if self.shutdown {
-         return;
-      }
+    /// Receive the shutdown notice, waiting if necessary.
+    pub(crate) async fn recv(&mut self) {
+        // If the shutdown signal has already been received, then return
+        // immediately.
+        if self.shutdown {
+            return;
+        }
 
-      // Cannot receive a "lag error" as only one value is ever sent.
-      let _ = self.notify.recv().await;
+        // Cannot receive a "lag error" as only one value is ever sent.
+        let _ = self.notify.recv().await;
 
-      // Remember that the signal has been received.
-      self.shutdown = true;
-   }
+        // Remember that the signal has been received.
+        self.shutdown = true;
+    }
+}
+
+impl Clone for Shutdown {
+    fn clone(&self) -> Self {
+        Shutdown {
+            shutdown: self.shutdown,
+            notify: self.notify.resubscribe(),
+        }
+    }
 }