rust 45 lines · 8 steps

Graceful shutdown in an Axum server

Wire an Axum server to drain connections cleanly when it receives Ctrl+C or SIGTERM.

Explained by highlit
Axum Intermediate graceful-shutdown signal-handling async tokio select
1use std::time::Duration;
2 
3use axum::{routing::get, Router};
4use tokio::net::TcpListener;
5use tokio::signal;
6 
7async fn shutdown_signal() {
8 let ctrl_c = async {
9 signal::ctrl_c()
10 .await
11 .expect("failed to install Ctrl+C handler");
12 };
13 
14 #[cfg(unix)]
15 let terminate = async {
16 signal::unix::signal(signal::unix::SignalKind::terminate())
17 .expect("failed to install SIGTERM handler")
18 .recv()
19 .await;
20 };
21 
22 #[cfg(not(unix))]
23 let terminate = std::future::pending::<()>();
24 
25 tokio::select! {
26 _ = ctrl_c => {},
27 _ = terminate => {},
28 }
29 
30 tracing::info!("shutdown signal received, draining connections");
31}
32 
33pub async fn serve() -> anyhow::Result<()> {
34 let app = Router::new().route("/health", get(|| async { "ok" }));
35 
36 let listener = TcpListener::bind("0.0.0.0:3000").await?;
37 tracing::info!("listening on {}", listener.local_addr()?);
38 
39 axum::serve(listener, app)
40 .with_graceful_shutdown(shutdown_signal())
41 .await?;
42 
43 tokio::time::sleep(Duration::from_millis(250)).await;
44 Ok(())
45}
01 / 01
STEP 01

Walkthrough

Space play ←→ step click any line
Three takeaways
  1. 1Graceful shutdown means racing two signal futures and stopping the server when either fires.
  2. 2Platform differences like SIGTERM are handled with cfg attributes so the code compiles everywhere.
  3. 3with_graceful_shutdown lets Axum finish in-flight requests instead of dropping connections abruptly.

Related explainers

rust 
use serde::Deserialize;
 
#[derive(Debug, Deserialize)]
#[serde(untagged)]

Parsing flexible JSON shapes with serde

deserialization enums json
Intermediate 6 steps
python 
from fastapi import FastAPI, WebSocket, WebSocketDisconnect
 
app = FastAPI()
 

Building a WebSocket chat with FastAPI

websockets broadcast connection-management
Intermediate 9 steps
go 
package streaming
 
import (
	"bufio"

Streaming NDJSON logs over HTTP in Go

http-streaming channels select
Advanced 10 steps
rust 
use std::cmp::{Ordering, Reverse};
use std::collections::BinaryHeap;
use std::fs::File;
use std::io::{self, BufRead, BufReader, BufWriter, Lines, Write};

K-way merge of sorted logs in Rust

binary-heap k-way-merge streaming-io
Intermediate 8 steps
rust 
use axum::{
    extract::{Path, State},
    response::sse::{Event, KeepAlive, Sse},
};

Streaming import progress with SSE in Axum

server-sent-events streams watch-channel
Advanced 7 steps
rust 
use std::f64::consts::PI;
 
#[derive(Debug, Clone, Copy)]
pub struct LatLng {

Building geographic bounding boxes in Rust

geospatial value-types option
Intermediate 7 steps

Share this explainer

Here's the card — post it anywhere.

Graceful shutdown in an Axum server — share card
Made with highlit — turn any snippet into a walkthrough like this in about a minute.
Explain your code