fix: update memory cache before persistent store to prevent race condition

When multiple concurrent operations access the state store, there was a race
condition where GET requests could read stale cached data while the persistent
store was being updated. This fix ensures the memory cache is updated first,
preventing other threads from reading stale data during the update window.

This change affects both update() and store() methods in StateStore.

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: Ian Clarke

crates/core/src/wasm_runtime/state_store.rs

@@ -84,12 +84,13 @@ where
                 .map_err(Into::into)?
                 .ok_or_else(|| StateStoreError::MissingContract(*key))?;
         }
-        self.store
-            .store(*key, state.clone())
-            .await
-            .map_err(Into::into)?;
+        // Update memory cache first to prevent race condition where GET requests
+        // could read stale cached data while persistent store is being updated
         let cost = state.size() as i64;
-        self.state_mem_cache.insert(*key, state, cost).await;
+        self.state_mem_cache.insert(*key, state.clone(), cost).await;
+
+        // Then update persistent store
+        self.store.store(*key, state).await.map_err(Into::into)?;
         Ok(())
     }
 
@@ -99,12 +100,12 @@ where
         state: WrappedState,
         params: Parameters<'static>,
     ) -> Result<(), StateStoreError> {
-        self.store
-            .store(key, state.clone())
-            .await
-            .map_err(Into::into)?;
+        // Update memory cache first to prevent race condition
         let cost = state.size() as i64;
-        self.state_mem_cache.insert(key, state, cost).await;
+        self.state_mem_cache.insert(key, state.clone(), cost).await;
+
+        // Then update persistent stores
+        self.store.store(key, state).await.map_err(Into::into)?;
         self.store
             .store_params(key, params.clone())
             .await