mirrors/krata
1
0
Fork 0
mirror of https://github.com/edera-dev/krata.git synced 2025-08-02 12:50:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

daemon: mask SIGHUP

Browse Source
This commit is contained in:
Alex Zenla 2024-03-05 23:10:31 +00:00
parent 0f85e30149
commit f8e2f50c60
No known key found for this signature in database
GPG Key ID: 067B238899B51269
3 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Cargo.toml
View File

@ -58,6 +58,7 @@ path-absolutize = "3.1.1"
tokio-tun = "0.11.2"
tokio-listener = "0.3.1"
trait-variant = "0.1.1"
signal-hook = "0.3.17"
[workspace.dependencies.uuid]
version = "1.6.1"

1
daemon/Cargo.toml
View File

@ -31,6 +31,7 @@ futures = { workspace = true }
bytes = { workspace = true }
tokio-stream = { workspace = true }
async-trait = { workspace = true }
signal-hook = { workspace = true }
[dependencies.tokio-listener]
workspace = true

9
daemon/bin/daemon.rs
View File

@ -1,3 +1,5 @@
use std::sync::{atomic::AtomicBool, Arc};
use anyhow::{anyhow, Result};
use clap::Parser;
use env_logger::Env;
@ -15,6 +17,7 @@ struct Args {
#[tokio::main(flavor = "multi_thread", worker_threads = 10)]
async fn main() -> Result<()> {
env_logger::Builder::from_env(Env::default().default_filter_or("warn")).init();
mask_sighup()?;
let args = Args::parse();
let Some(listener) = args.listener.bind().await else {
@ -25,3 +28,9 @@ async fn main() -> Result<()> {
daemon.listen(listener?).await?;
Ok(())
}
fn mask_sighup() -> Result<()> {
let flag = Arc::new(AtomicBool::new(false));
signal_hook::flag::register(signal_hook::consts::SIGHUP, flag)?;
Ok(())
}