From be6c15a2008645568a8f28182a371d08a51ff944 Mon Sep 17 00:00:00 2001 From: Akira Hayakawa Date: Fri, 1 Nov 2024 20:10:20 +0900 Subject: [PATCH] Use tokio::sync::Mutex to protext response_cache --- sorock/src/process/command_log/consumer.rs | 2 +- sorock/src/process/command_log/mod.rs | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/sorock/src/process/command_log/consumer.rs b/sorock/src/process/command_log/consumer.rs index 4ed8600c..0d224e96 100644 --- 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 index fed5c246..9dd380ac 100644 --- 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, + response_cache: tokio::sync::Mutex, user_completions: spin::Mutex>, kern_completions: spin::Mutex>, } @@ -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()) }