OpenSource/axum
OpenSource/axum
1
0
Fork 0
mirror of https://github.com/tokio-rs/axum.git synced 2024-10-23 17:36:39 +02:00
Code Issues Projects Releases Packages Wiki Activity

Simplify graceful shutdown (#673)

Browse source
This commit is contained in:
Jonas Platte 2021-12-28 16:23:07 +01:00 committed by GitHub
parent 5698fb8be9
commit 616a43aaa3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

0
examples/graceful_shutdown/Cargo.toml → examples/graceful-shutdown/Cargo.toml
View file

36
examples/graceful_shutdown/src/main.rs → examples/graceful-shutdown/src/main.rs
View file

@ -7,6 +7,7 @@
use axum::{response::Html, routing::get, Router};
use std::net::SocketAddr;
use tokio::signal;
#[tokio::main]
async fn main() {
@ -27,29 +28,28 @@ async fn handler() -> Html<&'static str> {
Html("<h1>Hello, World!</h1>")
}
#[cfg(unix)]
pub async fn shutdown_signal() {
use std::io;
use tokio::signal::unix::SignalKind;
async fn shutdown_signal() {
let ctrl_c = async {
signal::ctrl_c()
.await
.expect("failed to install Ctrl+C handler");
};
async fn terminate() -> io::Result<()> {
tokio::signal::unix::signal(SignalKind::terminate())?
#[cfg(unix)]
let terminate = async {
signal::unix::signal(signal::unix::SignalKind::terminate())
.expect("failed to install signal handler")
.recv()
.await;
Ok(())
}
};
#[cfg(not(unix))]
let terminate = std::future::pending::<()>();
tokio::select! {
_ = terminate() => {},
_ = tokio::signal::ctrl_c() => {},
_ = ctrl_c => {},
_ = terminate => {},
}
println!("signal received, starting graceful shutdown")
}
#[cfg(windows)]
pub async fn shutdown_signal() {
tokio::signal::ctrl_c()
.await
.expect("faild to install CTRL+C handler");
println!("signal received, starting graceful shutdown")
println!("signal received, starting graceful shutdown");
}