|
|
|
|
@@ -17,10 +17,7 @@ async fn main() -> anyhow::Result<()> {
|
|
|
|
|
tracing::info!(%addr, "mangalord listening");
|
|
|
|
|
let listener = tokio::net::TcpListener::bind(addr).await?;
|
|
|
|
|
axum::serve(listener, router)
|
|
|
|
|
.with_graceful_shutdown(async {
|
|
|
|
|
let _ = tokio::signal::ctrl_c().await;
|
|
|
|
|
tracing::info!("ctrl-c received; shutting down");
|
|
|
|
|
})
|
|
|
|
|
.with_graceful_shutdown(shutdown_signal())
|
|
|
|
|
.await?;
|
|
|
|
|
|
|
|
|
|
// Drain background tasks (crawler daemon) before exiting so Chromium
|
|
|
|
|
@@ -30,3 +27,33 @@ async fn main() -> anyhow::Result<()> {
|
|
|
|
|
}
|
|
|
|
|
Ok(())
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// Wait for either Ctrl-C (interactive shell) or SIGTERM (Docker /
|
|
|
|
|
/// Kubernetes / Podman / systemd stop) and log which arrived. Without
|
|
|
|
|
/// the SIGTERM branch, `docker compose stop` runs out its grace period
|
|
|
|
|
/// and skips straight to SIGKILL — the daemon never gets the
|
|
|
|
|
/// `daemon.shutdown().await` path, leaking Chromium.
|
|
|
|
|
async fn shutdown_signal() {
|
|
|
|
|
use tokio::signal::unix::{signal, SignalKind};
|
|
|
|
|
let mut sigterm = match signal(SignalKind::terminate()) {
|
|
|
|
|
Ok(s) => s,
|
|
|
|
|
Err(e) => {
|
|
|
|
|
// SignalKind::terminate() is supported on every Unix the
|
|
|
|
|
// tokio runtime runs on; if registration fails we still
|
|
|
|
|
// honour Ctrl-C so the process is at least
|
|
|
|
|
// interactive-shutdownable.
|
|
|
|
|
tracing::warn!(error = %e, "could not install SIGTERM handler; falling back to ctrl_c only");
|
|
|
|
|
let _ = tokio::signal::ctrl_c().await;
|
|
|
|
|
tracing::info!("ctrl-c received; shutting down");
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
};
|
|
|
|
|
tokio::select! {
|
|
|
|
|
_ = tokio::signal::ctrl_c() => {
|
|
|
|
|
tracing::info!("ctrl-c received; shutting down");
|
|
|
|
|
}
|
|
|
|
|
_ = sigterm.recv() => {
|
|
|
|
|
tracing::info!("SIGTERM received; shutting down");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|