e4378f0143
Co-authored-by: edef <edef@edef.eu> Change-Id: I233206e8aae35504ca0519ac88178dfc5596bedb Reviewed-on: https://cl.tvl.fyi/c/depot/+/12439 Reviewed-by: flokli <flokli@flokli.de> Autosubmit: yuka <yuka@yuka.dev> Tested-by: BuildkiteCI
52 lines
1.6 KiB
Rust
52 lines
1.6 KiB
Rust
use tonic::async_trait;
|
|
use tvix_castore::B3Digest;
|
|
|
|
mod import;
|
|
mod renderer;
|
|
pub mod seekable;
|
|
pub use import::ingest_nar;
|
|
pub use import::ingest_nar_and_hash;
|
|
pub use renderer::calculate_size_and_sha256;
|
|
pub use renderer::write_nar;
|
|
pub use renderer::SimpleRenderer;
|
|
use tvix_castore::Node;
|
|
|
|
#[async_trait]
|
|
pub trait NarCalculationService: Send + Sync {
|
|
/// Return the nar size and nar sha256 digest for a given root node.
|
|
/// This can be used to calculate NAR-based output paths.
|
|
async fn calculate_nar(&self, root_node: &Node)
|
|
-> Result<(u64, [u8; 32]), tvix_castore::Error>;
|
|
}
|
|
|
|
#[async_trait]
|
|
impl<A> NarCalculationService for A
|
|
where
|
|
A: AsRef<dyn NarCalculationService> + Send + Sync,
|
|
{
|
|
async fn calculate_nar(
|
|
&self,
|
|
root_node: &Node,
|
|
) -> Result<(u64, [u8; 32]), tvix_castore::Error> {
|
|
self.as_ref().calculate_nar(root_node).await
|
|
}
|
|
}
|
|
|
|
/// Errors that can encounter while rendering NARs.
|
|
#[derive(Debug, thiserror::Error)]
|
|
pub enum RenderError {
|
|
#[error("failure talking to a backing store client: {0}")]
|
|
StoreError(#[source] std::io::Error),
|
|
|
|
#[error("unable to find directory {0}, referred from {1:?}")]
|
|
DirectoryNotFound(B3Digest, bytes::Bytes),
|
|
|
|
#[error("unable to find blob {0}, referred from {1:?}")]
|
|
BlobNotFound(B3Digest, bytes::Bytes),
|
|
|
|
#[error("unexpected size in metadata for blob {0}, referred from {1:?} returned, expected {2}, got {3}")]
|
|
UnexpectedBlobMeta(B3Digest, bytes::Bytes, u32, u32),
|
|
|
|
#[error("failure using the NAR writer: {0}")]
|
|
NARWriterError(std::io::Error),
|
|
}
|