mdebray/tvl-depot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor(tvix/glue): drop ingest_entries_sync

Browse source 
Make this function async, and do the block_on on the (single) callsite.

Change-Id: Ib8b0b54ab5370fe02ef95f38a45d8866868a9d60
Reviewed-on: https://cl.tvl.fyi/c/depot/+/11285
Reviewed-by: Connor Brewster <cbrewster@hey.com>
Tested-by: BuildkiteCI
This commit is contained in:
Florian Klink 2024-03-28 22:05:46 +01:00 committed by flokli
parent bd32024047
commit 156a5a0fb6
2 changed files with 20 additions and 18 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tvix/glue/src/builtins/import.rs
View file

@ -102,12 +102,15 @@ async fn filtered_ingest(
pin_mut!(entries_stream);
state.tokio_handle.block_on(async {
state
.ingest_entries_sync(entries_stream)
.ingest_entries(entries_stream)
.await
.map_err(|err| ErrorKind::IO {
path: Some(path.to_path_buf()),
error: err.into(),
})
})
}
#[builtins(state = "Rc<TvixStoreIO>")]

9
tvix/glue/src/tvix_store_io.rs
View file

@ -273,13 +273,13 @@ impl TvixStoreIO {
.map_err(|e| std::io::Error::new(io::ErrorKind::Other, e))
}
/// This forwards the ingestion to the [`tvix_castore::import::ingest_entries`]
/// with a [`tokio::runtime::Handle::block_on`] call for synchronicity.
pub(crate) fn ingest_entries_sync<S>(&self, entries_stream: S) -> io::Result<Node>
/// This forwards the ingestion to the [`tvix_castore::import::ingest_entries`],
/// passing the blob_service and directory_service that's used.
/// The error is mapped to std::io::Error for simplicity.
pub(crate) async fn ingest_entries<S>(&self, entries_stream: S) -> io::Result<Node>
where
S: Stream<Item = DirEntry> + Unpin,
{
self.tokio_handle.block_on(async move {
tvix_castore::import::ingest_entries(
&self.blob_service,
&self.directory_service,
@ -287,7 +287,6 @@ impl TvixStoreIO {
)
.await
.map_err(|err| std::io::Error::new(io::ErrorKind::Other, err))
})
}
pub(crate) async fn node_to_path_info(