diff options
author | Florian Klink <flokli@flokli.de> | 2023-03-27T15·08+0200 |
---|---|---|
committer | clbot <clbot@tvl.fyi> | 2023-04-07T09·26+0000 |
commit | 0836450006e3ef3ec4f150696c164fef7eb701db (patch) | |
tree | 04306a0f9b43702177c89283e5d29da9e4c2f6c4 /tvix/store/src/directoryservice/utils.rs | |
parent | 96d7f4f0acc82bc6b12b41ac8d5fbfbd54e41599 (diff) |
feat(tvix/store/directorysvc): add put_multiple_start r/6073
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 <tazjin@tvl.su> Autosubmit: flokli <flokli@flokli.de>
Diffstat (limited to '')
-rw-r--r-- | tvix/store/src/directoryservice/utils.rs | 38 |
1 files changed, 38 insertions, 0 deletions
diff --git a/tvix/store/src/directoryservice/utils.rs b/tvix/store/src/directoryservice/utils.rs index 10edb9ba74dc..7d41e8d3ba01 100644 --- a/tvix/store/src/directoryservice/utils.rs +++ b/tvix/store/src/directoryservice/utils.rs @@ -1,3 +1,4 @@ +use super::DirectoryPutter; use super::DirectoryService; use crate::proto; use crate::Error; @@ -104,3 +105,40 @@ impl<DS: DirectoryService> Iterator for DirectoryTraverser<DS> { } } } + +/// This is a simple implementation of a Directory uploader. +/// TODO: verify connectivity? Factor out these checks into generic helpers? +pub struct SimplePutter<DS: DirectoryService> { + directory_service: DS, + last_directory_digest: Option<[u8; 32]>, +} + +impl<DS: DirectoryService> SimplePutter<DS> { + pub fn new(directory_service: DS) -> Self { + Self { + directory_service, + last_directory_digest: None, + } + } +} + +impl<DS: DirectoryService> DirectoryPutter for SimplePutter<DS> { + fn put(&mut self, directory: proto::Directory) -> Result<(), Error> { + let digest = self.directory_service.put(directory)?; + + // track the last directory digest + self.last_directory_digest = Some(digest); + + Ok(()) + } + + /// We need to be mutable here, as that's the signature of the trait. + fn close(&mut self) -> Result<[u8; 32], Error> { + match self.last_directory_digest { + Some(last_digest) => Ok(last_digest), + None => Err(Error::InvalidRequest( + "no directories sent, can't show root digest".to_string(), + )), + } + } +} |