summaryrefslogtreecommitdiff
path: root/src/handle_unwind.rs
blob: 42b6fc5cff26911908bf1c1ac36ba2db9ffa94ab (plain)
use std::panic::{catch_unwind, resume_unwind, AssertUnwindSafe};

/// Runs `try_fn`. If it unwinds, it will run `catch` and then continue
/// unwinding. This is used instead of `scopeguard` to ensure the `catch`
/// function doesn't run if the thread is already panicking. The unwind
/// must specifically be caused by the `try_fn`
pub fn handle_unwind<R, F: FnOnce() -> R, G: FnOnce()>(try_fn: F, catch: G) -> R {
	let try_fn = AssertUnwindSafe(try_fn);
	catch_unwind(try_fn).unwrap_or_else(|e| {
		catch();
		resume_unwind(e)
	})
}