Use tokio::sync::Mutex to protext response_cache

akiradeveloper · Nov 1, 2024 · be6c15a · be6c15a
1 parent 22f1fac
commit be6c15a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sorock/src/process/command_log/consumer.rs b/sorock/src/process/command_log/consumer.rs
@@ -63,7 +63,7 @@ impl CommandLog {
         };
 
         if do_process {
-            let mut response_cache = self.response_cache.lock();
+            let mut response_cache = self.response_cache.lock().await;
             debug!("process user@{process_index}");
             match command {
                 Command::Snapshot { .. } => {

diff --git a/sorock/src/process/command_log/mod.rs b/sorock/src/process/command_log/mod.rs
@@ -29,7 +29,7 @@ pub struct Inner {
     pub membership_pointer: AtomicU64,
 
     app: App,
-    response_cache: spin::Mutex<ResponseCache>,
+    response_cache: tokio::sync::Mutex<ResponseCache>,
     user_completions: spin::Mutex<BTreeMap<Index, completion::UserCompletion>>,
     kern_completions: spin::Mutex<BTreeMap<Index, completion::KernCompletion>>,
 }
@@ -50,7 +50,7 @@ impl CommandLog {
             snapshot_lock: tokio::sync::RwLock::new(()),
             user_completions: spin::Mutex::new(BTreeMap::new()),
             kern_completions: spin::Mutex::new(BTreeMap::new()),
-            response_cache: spin::Mutex::new(ResponseCache::new()),
+            response_cache: tokio::sync::Mutex::new(ResponseCache::new()),
         };
         Self(inner.into())
     }