summaryrefslogtreecommitdiff
path: root/cli/tokio_util.rs
diff options
context:
space:
mode:
Diffstat (limited to 'cli/tokio_util.rs')
-rw-r--r--cli/tokio_util.rs17
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)]