diff options
| author | Alex Auvolat <alex@adnab.me> | 2021-10-13 17:12:13 +0200 |
|---|---|---|
| committer | Alex Auvolat <alex@adnab.me> | 2021-10-13 17:12:13 +0200 |
| commit | 70839d70d86354232f168e63ce4062219acb85c7 (patch) | |
| tree | 9c956af0339aa048f487c3a4e54c320be8d13647 /src/util.rs | |
| parent | 8dede69dee20b812ad1dcab5b374c60232409f4f (diff) | |
| download | netapp-70839d70d86354232f168e63ce4062219acb85c7.tar.gz netapp-70839d70d86354232f168e63ce4062219acb85c7.zip | |
Try to handle termination and closing of stuff properly
Diffstat (limited to 'src/util.rs')
| -rw-r--r-- | src/util.rs | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/src/util.rs b/src/util.rs index ba485bf..e5b57ec 100644 --- a/src/util.rs +++ b/src/util.rs @@ -1,5 +1,7 @@ use serde::Serialize; +use log::info; + use tokio::sync::watch; pub type NodeID = sodiumoxide::crypto::sign::ed25519::PublicKey; @@ -38,3 +40,15 @@ pub async fn await_exit(mut must_exit: watch::Receiver<bool>) { } } } + +pub fn watch_ctrl_c() -> watch::Receiver<bool> { + let (send_cancel, watch_cancel) = watch::channel(false); + tokio::spawn(async move { + tokio::signal::ctrl_c() + .await + .expect("failed to install CTRL+C signal handler"); + info!("Received CTRL+C, shutting down."); + send_cancel.send(true).unwrap(); + }); + watch_cancel +} |