diff options
author | Florian Klink <flokli@flokli.de> | 2024-05-10T05·31+0300 |
---|---|---|
committer | clbot <clbot@tvl.fyi> | 2024-05-11T15·09+0000 |
commit | 25d97c5b55758c9e62ab3f090669dea7a10ab387 (patch) | |
tree | 2dce6d4be3f8bb90075288f6b97b16eb82ca39eb /tvix/castore/src | |
parent | 84114cf02c7d742ff4bbfb1952734839d7e013eb (diff) |
refactor(tvix/castore/directoryservice/memory): use tokio RwLock r/8114
We don't want to block here, and this also means there's no poisoning to deal with. Change-Id: Ic375571970c48beace0005ae2c012135086a4d67 Reviewed-on: https://cl.tvl.fyi/c/depot/+/11613 Autosubmit: flokli <flokli@flokli.de> Tested-by: BuildkiteCI Reviewed-by: Connor Brewster <cbrewster@hey.com>
Diffstat (limited to '')
-rw-r--r-- | tvix/castore/src/directoryservice/memory.rs | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tvix/castore/src/directoryservice/memory.rs b/tvix/castore/src/directoryservice/memory.rs index 2cbbbd1b1657..3d1c5009be9b 100644 --- a/tvix/castore/src/directoryservice/memory.rs +++ b/tvix/castore/src/directoryservice/memory.rs @@ -1,7 +1,8 @@ use crate::{proto, B3Digest, Error}; use futures::stream::BoxStream; use std::collections::HashMap; -use std::sync::{Arc, RwLock}; +use std::sync::Arc; +use tokio::sync::RwLock; use tonic::async_trait; use tracing::{instrument, warn}; @@ -17,7 +18,7 @@ pub struct MemoryDirectoryService { impl DirectoryService for MemoryDirectoryService { #[instrument(skip(self, digest), fields(directory.digest = %digest))] async fn get(&self, digest: &B3Digest) -> Result<Option<proto::Directory>, Error> { - let db = self.db.read()?; + let db = self.db.read().await; match db.get(digest) { // The directory was not found, return @@ -62,7 +63,7 @@ impl DirectoryService for MemoryDirectoryService { } // store it - let mut db = self.db.write()?; + let mut db = self.db.write().await; db.insert(digest.clone(), directory); Ok(digest) |