aboutsummaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorbnewbold <bnewbold@robocracy.org>2016-10-09 16:32:02 -0700
committerbnewbold <bnewbold@robocracy.org>2016-10-09 16:32:02 -0700
commit89f2f02ab813fe19fd029e95f6a2f6ce65c5a16b (patch)
treefbecf2aa0797888f2ded12b704eb9762743fb00a /src
parent5a84c5287040d57b31b6b6d63f9dc938915e80a2 (diff)
downloadeinhyrningsins-89f2f02ab813fe19fd029e95f6a2f6ce65c5a16b.tar.gz
einhyrningsins-89f2f02ab813fe19fd029e95f6a2f6ce65c5a16b.zip
upate python example with KeyboardInterrupt catching
Diffstat (limited to 'src')
-rw-r--r--src/main.rs48
1 files changed, 40 insertions, 8 deletions
diff --git a/src/main.rs b/src/main.rs
index c1e06b7..e849779 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -87,7 +87,7 @@ impl Offspring {
}
pub fn spawn(&mut self, cfg: &mut EinConfig) -> Result<(), String> {
- if self.state != OffspringState::Expectant && self.state != OffspringState::Dead {
+ if self.is_active() {
return Err(format!("Can't spawn from state: {:?}", self.state));
}
self.process = Some(cfg.cmd.spawn().expect("error spawning"));
@@ -95,6 +95,34 @@ impl Offspring {
self.attempts = 0;
Ok(())
}
+
+ pub fn is_active(&self) -> bool {
+ match self.state {
+ OffspringState::Expectant => false,
+ OffspringState::Infancy => true,
+ OffspringState::Healthy => true,
+ OffspringState::Notified => true,
+ OffspringState::Dead => false,
+ }
+ }
+
+ pub fn signal(&mut self, sig: Signal) {
+ if !self.is_active() {
+ return;
+ }
+ let nix_sig = match sig {
+ Signal::HUP => nix::sys::signal::Signal::SIGHUP,
+ Signal::INT => nix::sys::signal::Signal::SIGINT,
+ _ => { println!("Unexpected signal: {:?}", sig); return; },
+ };
+ match self.process {
+ Some(ref p) => { nix::sys::signal::kill(p.id() as i32, nix_sig).unwrap(); },
+ None => (),
+ }
+ // TODO: abs probably isn't the best way to i32 -> u32 here
+ //nix::sys::signal::kill(num::abs(self.process.unwrap().id()), nix_sig);
+ //let pid = self.process.unwrap().id();
+ }
}
#[derive(Copy, Clone, Debug, PartialEq)]
@@ -107,7 +135,6 @@ fn shepard(mut cfg: EinConfig, signal_rx: Receiver<Signal>) {
//// create timer
let timer = timer::Timer::new();
- // XXX: these signatures are bogus
let (timer_tx, timer_rx): (Sender<TimerAction>, Receiver<TimerAction>) = chan::async();
//// birth the initial set of offspring
@@ -127,11 +154,15 @@ fn shepard(mut cfg: EinConfig, signal_rx: Receiver<Signal>) {
//// infinite select() loop over timers, signals, rpc
loop {
chan_select! {
- timer_rx.recv() => println!("Timer tick'd"),
- signal_rx.recv() -> sig => {
- println!("");
- println!("Signal received! {:?}", sig);
- break;
+ timer_rx.recv() => { println!("Timer tick'd"); "TIMER" },
+ signal_rx.recv() -> sig => match sig.expect("Error with signal handler") {
+ // XXX: Signal::HUP => brood.iter().for_each(|o| o.signal(sig)),
+ Signal::INT | Signal::TERM=> {
+ println!("Notifying children...");
+ // XXX: brood.iter().for_each(|o| o.signal(sig));
+ break;
+ },
+ _ => "Other"
}
}
}
@@ -144,8 +175,8 @@ fn shepard(mut cfg: EinConfig, signal_rx: Receiver<Signal>) {
None => (),
}
}
-*/
println!("Done.");
+*/
}
fn print_usage(opts: Options) {
@@ -209,6 +240,7 @@ fn main() {
builder.init().unwrap();
let binds: Vec<TcpListener> = sock_addrs.iter().map(|sa| {
+ // XXX: SO_REUSE here
TcpListener::bind(sa).unwrap()
}).collect();