From 0836450006e3ef3ec4f150696c164fef7eb701db Mon Sep 17 00:00:00 2001 From: Florian Klink Date: Mon, 27 Mar 2023 17:08:16 +0200 Subject: feat(tvix/store/directorysvc): add put_multiple_start This provides a handle to upload multiple proto::Directory as part of the same closure. Change-Id: I9213dde257a260c8622239918ea541064b270484 Reviewed-on: https://cl.tvl.fyi/c/depot/+/8356 Tested-by: BuildkiteCI Reviewed-by: tazjin Autosubmit: flokli --- tvix/store/src/directoryservice/sled.rs | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'tvix/store/src/directoryservice/sled.rs') diff --git a/tvix/store/src/directoryservice/sled.rs b/tvix/store/src/directoryservice/sled.rs index 44d61ae42ae5..d060232307b1 100644 --- a/tvix/store/src/directoryservice/sled.rs +++ b/tvix/store/src/directoryservice/sled.rs @@ -5,6 +5,7 @@ use prost::Message; use std::path::PathBuf; use tracing::{instrument, warn}; +use super::utils::SimplePutter; use super::{DirectoryService, DirectoryTraverser}; #[derive(Clone)] @@ -97,4 +98,14 @@ impl DirectoryService for SledDirectoryService { fn get_recursive(&self, root_directory_digest: &[u8; 32]) -> Self::DirectoriesIterator { DirectoryTraverser::with(self.clone(), root_directory_digest) } + + type DirectoryPutter = SimplePutter; + + #[instrument(skip_all)] + fn put_multiple_start(&self) -> Self::DirectoryPutter + where + Self: Clone, + { + SimplePutter::new(self.clone()) + } } -- cgit 1.4.1