diff options
Diffstat (limited to 'cli/tokio_util.rs')
-rw-r--r-- | cli/tokio_util.rs | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/cli/tokio_util.rs b/cli/tokio_util.rs index 810b826b4..bf27c39f0 100644 --- a/cli/tokio_util.rs +++ b/cli/tokio_util.rs @@ -13,10 +13,22 @@ pub fn run<F>(future: F) where F: Future<Item = (), Error = ()> + Send + 'static, { + abort_on_panic(); // tokio::runtime::current_thread::run(future) tokio::run(future) } +// Tokio swallows panics. In order to actually crash when we panic, we +// have to set this custom hook. +// https://github.com/tokio-rs/tokio/issues/495 +// https://github.com/tokio-rs/tokio/issues/209 +pub fn abort_on_panic() { + std::panic::set_hook(Box::new(|panic_info| { + eprintln!("{}", panic_info.to_string()); + std::process::abort(); + })); +} + pub fn block_on<F, R, E>(future: F) -> Result<R, E> where F: Send + 'static + Future<Item = R, Error = E>, @@ -40,7 +52,10 @@ where let rt = tokio::runtime::Runtime::new().unwrap(); let mut executor = rt.executor(); let mut enter = tokio_executor::enter().expect("Multiple executors at once"); - tokio_executor::with_default(&mut executor, &mut enter, move |_enter| f()); + tokio_executor::with_default(&mut executor, &mut enter, move |_enter| { + abort_on_panic(); + f() + }); } #[derive(Debug)] |