awfufu/traudit
1
0
Fork 0
mirror of https://github.com/awfufu/traudit synced 2026-03-01 05:29:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: refine signal handling and add config check to systemd reload

Browse Source
This commit is contained in:
氟氟
2026-01-27 13:52:38 +08:00
parent a71e950734
commit 897237755f
4 changed files with 9 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Cargo.lock generated
View File

@@ -3887,7 +3887,7 @@ dependencies = [
[[package]] [[package]]
name = "traudit" name = "traudit"
version = "0.0.6" version = "0.0.7"
dependencies = [ dependencies = [
"anyhow", "anyhow",
"async-trait", "async-trait",

2
Cargo.toml
View File

@@ -1,6 +1,6 @@
[package] [package]
name = "traudit" name = "traudit"
version = "0.0.6" version = "0.0.7"
edition = "2021" edition = "2021"
authors = ["awfufu"] authors = ["awfufu"]
description = "A reverse proxy that streams audit records directly to databases." description = "A reverse proxy that streams audit records directly to databases."

14
src/main.rs
View File

@@ -136,20 +136,18 @@ async fn main() -> anyhow::Result<()> {
Err(e) => error!("failed to spawn new process: {}", e), Err(e) => error!("failed to spawn new process: {}", e),
} }
// Initiate graceful shutdown for this process // Initiate graceful shutdown for this process
info!("shutting down old process gracefully..."); info!("shutting down old process gracefully (draining connections)...");
let _ = shutdown_tx.send(());
// Wait for server to finish (graceful drain)
let _ = server_handle.await;
} }
_ = sigint.recv() => { _ = sigint.recv() => {
info!("received SIGINT, shutdown..."); info!("received SIGINT, exiting immediately...");
} }
_ = sigterm.recv() => { _ = sigterm.recv() => {
info!("received SIGTERM, shutdown..."); info!("received SIGTERM, exiting immediately...");
} }
} }
// Send shutdown signal to server components
let _ = shutdown_tx.send(());
// Wait for server to finish (graceful drain)
let _ = server_handle.await;
Ok(()) Ok(())
} }

1
traudit.service
View File

@@ -7,6 +7,7 @@ Type=notify
RuntimeDirectory=traudit RuntimeDirectory=traudit
WorkingDirectory=/run/traudit WorkingDirectory=/run/traudit
ExecStart=/usr/bin/traudit -f /etc/traudit/config.yaml ExecStart=/usr/bin/traudit -f /etc/traudit/config.yaml
ExecReload=/usr/bin/traudit -f /etc/traudit/config.yaml -t
ExecReload=/bin/kill -HUP $MAINPID ExecReload=/bin/kill -HUP $MAINPID
KillSignal=SIGINT KillSignal=SIGINT
Restart=on-failure Restart=on-failure