You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.4 KiB
60 lines
1.4 KiB
//! Unix signals
|
|
use super::*;
|
|
use tokio::{
|
|
signal::unix::{
|
|
self,
|
|
SignalKind,
|
|
},
|
|
};
|
|
|
|
|
|
|
|
pub async fn handle(mut state: State)
|
|
{
|
|
let mut usr1 = unix::signal(SignalKind::user_defined1()).expect("Failed to hook SIGUSR1");
|
|
let mut usr2 = unix::signal(SignalKind::user_defined2()).expect("Failed to hook SIGUSR2");
|
|
let mut quit = unix::signal(SignalKind::quit()).expect("Failed to hook SIGQUIT");
|
|
|
|
loop {
|
|
tokio::select! {
|
|
_ = state.on_shutdown() => {
|
|
break;
|
|
}
|
|
_ = usr1.recv() => {
|
|
info!("Got SIGUSR1. Saving chain immediately.");
|
|
if let Err(e) = save::save_now(&state).await {
|
|
error!("Failed to save chain: {}", e);
|
|
} else{
|
|
trace!("Saved chain okay");
|
|
}
|
|
},
|
|
_ = usr2.recv() => {
|
|
info!("Got SIGUSR1. Loading chain immediately.");
|
|
match save::load(&state.config().file).await {
|
|
Ok(new) => {
|
|
{
|
|
let mut chain = state.chain().write().await;
|
|
*chain = new;
|
|
}
|
|
trace!("Replaced with read chain");
|
|
},
|
|
Err(e) => {
|
|
error!("Failed to load chain from file, keeping current: {}", e);
|
|
},
|
|
}
|
|
},
|
|
_ = quit.recv() => {
|
|
warn!("Got SIGQUIT. Saving chain then aborting.");
|
|
if let Err(e) = save::save_now(&state).await {
|
|
error!("Failed to save chain: {}", e);
|
|
} else{
|
|
trace!("Saved chain okay.");
|
|
}
|
|
error!("Aborting");
|
|
std::process::abort()
|
|
},
|
|
}
|
|
}
|
|
trace!("Graceful shutdown");
|
|
}
|