diff options
| -rw-r--r-- | src/main.rs | 45 |
1 files changed, 45 insertions, 0 deletions
diff --git a/src/main.rs b/src/main.rs index d4221dc..b7dee25 100644 --- a/src/main.rs +++ b/src/main.rs @@ -33,6 +33,7 @@ use std::net::ToSocketAddrs; use getopts::Options; use std::os::unix::io::IntoRawFd; +use nix::sys::signal; fn run(binds: Vec<TcpListener>, mut prog: Command, number: u64) { @@ -58,6 +59,26 @@ fn run(binds: Vec<TcpListener>, mut prog: Command, number: u64) { for mut c in children { c.wait().unwrap(); } + println!("Done."); +} + +static mut interrupted: bool = false; + +extern fn handle_hup(_: i32) { + println!("This is where I would restart all children gracefully?"); +} + +extern fn handle_int(_: i32) { + let first = unsafe { + let tmp = !interrupted; + interrupted = true; + tmp + }; + if first { + println!("Waiting for childred to shutdown gracefully (Ctrl-C again to bail)"); + } else { + panic!(); + } } fn print_usage(opts: Options) { @@ -112,6 +133,30 @@ fn main() { } builder.init().unwrap(); + println!("Registering signal handlers..."); + let mut mask_hup = signal::SigSet::empty(); + mask_hup.add(signal::Signal::SIGHUP); + mask_hup.add(signal::Signal::SIGUSR2); + let hup_action = signal::SigAction::new( + signal::SigHandler::Handler(handle_hup), + signal::SaFlags::empty(), + mask_hup); + + let mut mask_int = signal::SigSet::empty(); + mask_int.add(signal::Signal::SIGINT); + mask_int.add(signal::Signal::SIGTERM); + let int_action = signal::SigAction::new( + signal::SigHandler::Handler(handle_int), + signal::SaFlags::empty(), + mask_int); + + unsafe { + signal::sigaction(signal::Signal::SIGHUP, &hup_action).unwrap(); + signal::sigaction(signal::Signal::SIGUSR2, &hup_action).unwrap(); + signal::sigaction(signal::Signal::SIGINT, &int_action).unwrap(); + signal::sigaction(signal::Signal::SIGTERM, &int_action).unwrap(); + } + let binds: Vec<TcpListener> = sock_addrs.iter().map(|sa| { TcpListener::bind(sa).unwrap() }).collect(); |