chore: gracefull shutdown on sigterm

Signed-off-by: Henry Gressmann <mail@henrygressmann.de>

Author: explodingcamera

Cargo.lock

@@ -26,7 +26,8 @@ async fn main() -> Result<()> {
     app.run_background_tasks();
 
     tokio::select! {
-        _ = tokio::signal::ctrl_c() => app_copy.shutdown(),
+        biased;
+        _ = liwan::utils::signals::shutdown() => app_copy.shutdown(),
         res = web::start_webserver(app.clone(), s) => res,
         res = tokio::task::spawn_blocking(move || app.clone().events.process(r)) => res?
     }

src/main.rs

@@ -5,6 +5,7 @@ pub mod referrer;
 pub mod refinery_duckdb;
 pub mod refinery_sqlite;
 pub mod seed;
+pub mod signals;
 pub mod useragent;
 pub mod validate;
 

src/utils/mod.rs

@@ -0,0 +1,17 @@
+use eyre::Result;
+use tokio::task::JoinSet;
+
+pub async fn shutdown() -> Result<()> {
+    let mut shutdown = JoinSet::new();
+    shutdown.spawn(tokio::signal::ctrl_c());
+
+    #[cfg(unix)]
+    shutdown.spawn(async {
+        let mut sigterm = tokio::signal::unix::signal(tokio::signal::unix::SignalKind::terminate())?;
+        sigterm.recv().await;
+        Ok(())
+    });
+
+    shutdown.join_next().await.expect("there are tasks")??;
+    Ok(())
+}