refactor(tvix/glue): drop ingest_entries_sync
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:
parent
bd32024047
commit
156a5a0fb6
2 changed files with 20 additions and 18 deletions
|
@ -102,12 +102,15 @@ async fn filtered_ingest(
|
|||
|
||||
pin_mut!(entries_stream);
|
||||
|
||||
state
|
||||
.ingest_entries_sync(entries_stream)
|
||||
.map_err(|err| ErrorKind::IO {
|
||||
path: Some(path.to_path_buf()),
|
||||
error: err.into(),
|
||||
})
|
||||
state.tokio_handle.block_on(async {
|
||||
state
|
||||
.ingest_entries(entries_stream)
|
||||
.await
|
||||
.map_err(|err| ErrorKind::IO {
|
||||
path: Some(path.to_path_buf()),
|
||||
error: err.into(),
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
#[builtins(state = "Rc<TvixStoreIO>")]
|
||||
|
|
|
@ -273,21 +273,20 @@ 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,
|
||||
entries_stream,
|
||||
)
|
||||
.await
|
||||
.map_err(|err| std::io::Error::new(io::ErrorKind::Other, err))
|
||||
})
|
||||
tvix_castore::import::ingest_entries(
|
||||
&self.blob_service,
|
||||
&self.directory_service,
|
||||
entries_stream,
|
||||
)
|
||||
.await
|
||||
.map_err(|err| std::io::Error::new(io::ErrorKind::Other, err))
|
||||
}
|
||||
|
||||
pub(crate) async fn node_to_path_info(
|
||||
|
|
Loading…
Reference in a new issue