shockrah/freechat
shockrah/freechat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

handling ctrl-c

Browse Source
This commit is contained in:
shockrah 2020-06-02 00:50:09 -07:00
parent 9033c80369
commit c7c6f530a0
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
server/src/main.rs
View File

@ -66,6 +66,12 @@ async fn main_responder(request: Request<Body>) -> Result<Response<Body>, hyper:
Ok(response) Ok(response)
} }
async fn shutdown_signal() {
tokio::signal::ctrl_c()
.await
.expect("Failed to capture ctrl-c signal");
}
#[tokio::main] #[tokio::main]
async fn main() { async fn main() {
dotenv().ok(); dotenv().ok();
@ -77,4 +83,8 @@ async fn main() {
}); });
let server = Server::bind(&addr).serve(service); let server = Server::bind(&addr).serve(service);
let graceful_shutdown = server.with_graceful_shutdown(shutdown_signal());
if let Err(e) = graceful_shutdown.await {
eprintln!("Server shutdown error: {}", e);
}
} }