Mercurial > public > mercurial-scm > hg

diff rust/chg/src/main.rs @ 40120:89742f1fa6cb

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
rust-chg: install signal handlers to forward signals to server I use sync::Once as a synchronization primitive because it's quite easy to use, and is good enough to prevent data race in these C functions.
author Yuya Nishihara <yuya@tcha.org>
date Tue, 25 Sep 2018 20:55:03 +0900
parents 6bdee4bc181a
children af52181f71ff
line wrap: on
line diff
--- a/rust/chg/src/main.rs	Mon Sep 24 22:19:49 2018 +0900
+++ b/rust/chg/src/main.rs	Tue Sep 25 20:55:03 2018 +0900
@@ -10,6 +10,7 @@
 
 use chg::{ChgClientExt, ChgUiHandler};
 use chg::locator;
+use chg::procutil;
 use futures::sync::oneshot;
 use std::env;
 use std::io;
@@ -38,9 +39,16 @@
             client.attach_io(io::stdin(), io::stdout(), io::stderr())
         })
         .and_then(|client| {
+            let pid = client.server_spec().process_id.unwrap();
+            let pgid = client.server_spec().process_group_id;
+            procutil::setup_signal_handler_once(pid, pgid)?;
+            Ok(client)
+        })
+        .and_then(|client| {
             client.run_command_chg(handler, env::args_os().skip(1))
         })
         .map(|(_client, _handler, code)| {
+            procutil::restore_signal_handler_once()?;
             Ok(code)
         })
         .or_else(|err| Ok(Err(err)))  // pass back error to caller