#53 Add with_state

audulus · Dec 11, 2023 · 87c0b7e · 87c0b7e
1 parent 18985bc
commit 87c0b7e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/context.rs b/src/context.rs
@@ -136,12 +136,17 @@ pub struct Context {
 
 thread_local! {
     pub(crate) static CONTEXT: RefCell<Context> = RefCell::new(Context::new());
+    pub(crate) static STATE_MAP: RefCell<StateMap> = RefCell::new(StateMap::new());
 }
 
 pub(crate) fn with_context<T>(id: ContextId, f: impl FnOnce(&mut Context) -> T) -> T {
     CONTEXT.with(|cx| f(&mut cx.borrow_mut()))
 }
 
+pub(crate) fn with_state<T>(f: impl FnOnce(&mut StateMap) -> T) -> T {
+    STATE_MAP.with(|map| f(&mut map.borrow_mut()))
+}
+
 impl Default for Context {
     fn default() -> Self {
         Self::new()