Support no_stdno-std

1 files changed, 28 insertions, 10 deletions

diff --git a/src/handle_unwind.rs b/src/handle_unwind.rs
index 42b6fc5..e00bbbd 100755
--- a/src/handle_unwind.rs
+++ b/src/handle_unwind.rs
@@ -1,13 +1,31 @@
+#[cfg(feature = "std")]
 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)
-	})
+cfg_select! {
+	feature = "std" => {
+		/// 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`
+		///
+		/// This has no effect in `no_std` environments
+		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)
+			})
+		}
+	}
+	_ => {
+		/// 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`
+		///
+		/// This has no effect in `no_std` environments
+		pub fn handle_unwind<R, F: FnOnce() -> R, G: FnOnce()>(try_fn: F, _catch: G) -> R {
+			try_fn()
+		}
+	}
 }