nostr/gossip
1
0
Fork 0
mirror of https://github.com/mikedilger/gossip.git synced 2024-09-29 08:21:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Cleaner shutdown by catching signals

Browse Source
This commit is contained in:
Mike Dilger 2024-07-19 13:32:58 +12:00
parent dcd932e256
commit afce8e890a
1 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
gossip-lib/src/overlord/mod.rs
View File

@ -29,6 +29,7 @@ use std::collections::HashMap;
use std::sync::atomic::Ordering;
use std::sync::mpsc;
use std::time::Duration;
use tokio::signal::unix::{signal, SignalKind};
use tokio::sync::broadcast::Sender;
use tokio::sync::mpsc::UnboundedReceiver;
use tokio::sync::watch::Receiver as WatchReceiver;
@ -185,6 +186,10 @@ impl Overlord {
}
}
let mut interrupt_signal = signal(SignalKind::interrupt())?;
let mut quit_signal = signal(SignalKind::quit())?;
let mut terminate_signal = signal(SignalKind::terminate())?;
// Start background tasks
crate::tasks::start_background_tasks();
@ -221,7 +226,23 @@ impl Overlord {
},
task_nextjoined = self.minions.join_next_with_id(), if !self.minions.is_empty() => {
self.handle_task_nextjoined(task_nextjoined).await;
}
},
v = interrupt_signal.recv() => if v.is_some() {
tracing::info!("SIGINT");
let _ = GLOBALS.write_runstate.send(RunState::ShuttingDown);
break;
},
v = quit_signal.recv() => if v.is_some() {
tracing::info!("SIGQUIT");
let _ = GLOBALS.write_runstate.send(RunState::ShuttingDown);
break;
},
v = terminate_signal.recv() => if v.is_some() {
tracing::info!("SIGTERM");
let _ = GLOBALS.write_runstate.send(RunState::ShuttingDown);
break;
},
}
}