Add .await

mdeloof · Jan 8, 2025 · 17e52c2 · 17e52c2
1 parent e94e183
commit 17e52c2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/statig/src/awaitable/inner.rs b/statig/src/awaitable/inner.rs
@@ -43,7 +43,7 @@ where
 
     /// Transition from the current state to the given target state.
     pub async fn transition(&mut self, mut target: M::State, context: &mut M::Context<'_>) {
-        M::before_transition(&mut self.shared_storage, &self.state, &target);
+        M::before_transition(&mut self.shared_storage, &self.state, &target).await;
 
         // Get the transition path we need to perform from one state to the next.
         let (exit_levels, enter_levels) = self.state.transition_path(&mut target);

diff --git a/statig/src/awaitable/state.rs b/statig/src/awaitable/state.rs
@@ -121,7 +121,8 @@ where
                             shared_storage,
                             StateOrSuperstate::Superstate(&superstate),
                             event,
-                        );
+                        )
+                        .await;
 
                         let response = superstate
                             .call_handler(shared_storage, event, context)
@@ -131,7 +132,8 @@ where
                             shared_storage,
                             StateOrSuperstate::Superstate(&superstate),
                             event,
-                        );
+                        )
+                        .await;
 
                         match response {
                             Response::Handled => break Response::Handled,