From 3193a259ad40b7620a461a571208eb81e8a4b76d Mon Sep 17 00:00:00 2001 From: Simon Cruanes Date: Fri, 30 Aug 2024 13:27:04 -0400 Subject: [PATCH] fix: catch TLS.Not_set --- src/core/types_.ml | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/core/types_.ml b/src/core/types_.ml index afafbba6..e7d738a8 100644 --- a/src/core/types_.ml +++ b/src/core/types_.ml @@ -27,8 +27,13 @@ let[@inline] get_current_fiber () : fiber option = match TLS.get_exn k_cur_fiber with | f when f != _dummy_fiber -> Some f | _ -> None + | exception TLS.Not_set -> None + +let error_get_current_fiber_ = + "Moonpool: get_current_fiber was called outside of a fiber." let[@inline] get_current_fiber_exn () : fiber = match TLS.get_exn k_cur_fiber with | f when f != _dummy_fiber -> f - | _ -> failwith "Moonpool: get_current_fiber was called outside of a fiber." + | _ -> failwith error_get_current_fiber_ + | exception TLS.Not_set -> failwith error_get_current_fiber_