Add Server::serve_with_graceful_shutdown

djc · Jun 12, 2023 · 2ad9b9b · 2ad9b9b
1 parent 0f8ccf8
commit 2ad9b9b
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/mendes/src/application.rs b/mendes/src/application.rs
@@ -1,6 +1,7 @@
 use std::borrow::Cow;
 #[cfg(feature = "with-http-body")]
 use std::error::Error as StdError;
+use std::future::Future;
 use std::net::SocketAddr;
 use std::str;
 use std::str::FromStr;
@@ -692,4 +693,9 @@ pub trait Server: Application {
     type ServerError;
 
     async fn serve(self, addr: &SocketAddr) -> Result<(), Self::ServerError>;
+    async fn serve_with_graceful_shutdown(
+        self,
+        addr: &SocketAddr,
+        signal: impl Future<Output = ()> + Send,
+    ) -> Result<(), Self::ServerError>;
 }
diff --git a/mendes/src/hyper.rs b/mendes/src/hyper.rs
@@ -31,6 +31,17 @@ where
             .serve(ApplicationService::new(self))
             .await
     }
+
+    async fn serve_with_graceful_shutdown(
+        self,
+        addr: &SocketAddr,
+        signal: impl Future<Output = ()> + Send,
+    ) -> Result<(), hyper::Error> {
+        hyper::Server::bind(addr)
+            .serve(ApplicationService::new(self))
+            .with_graceful_shutdown(signal)
+            .await
+    }
 }
 
 struct ApplicationService<A>(Arc<A>);