diff options
Diffstat (limited to 'tvix/store')
-rw-r--r-- | tvix/store/src/bin/tvix-store.rs | 93 | ||||
-rw-r--r-- | tvix/store/src/lib.rs | 1 | ||||
-rw-r--r-- | tvix/store/src/utils.rs | 35 |
3 files changed, 67 insertions, 62 deletions
diff --git a/tvix/store/src/bin/tvix-store.rs b/tvix/store/src/bin/tvix-store.rs index 912417af8146..ca79378eb4fc 100644 --- a/tvix/store/src/bin/tvix-store.rs +++ b/tvix/store/src/bin/tvix-store.rs @@ -11,10 +11,6 @@ use tokio_listener::Listener; use tokio_listener::SystemOptions; use tokio_listener::UserOptions; use tracing_subscriber::prelude::*; -use tvix_castore::blobservice; -use tvix_castore::blobservice::BlobService; -use tvix_castore::directoryservice; -use tvix_castore::directoryservice::DirectoryService; use tvix_castore::import; use tvix_castore::proto::blob_service_server::BlobServiceServer; use tvix_castore::proto::directory_service_server::DirectoryServiceServer; @@ -22,7 +18,6 @@ use tvix_castore::proto::node::Node; use tvix_castore::proto::GRPCBlobServiceWrapper; use tvix_castore::proto::GRPCDirectoryServiceWrapper; use tvix_castore::proto::NamedNode; -use tvix_store::pathinfoservice; use tvix_store::pathinfoservice::PathInfoService; use tvix_store::proto::nar_info; use tvix_store::proto::path_info_service_server::PathInfoServiceServer; @@ -196,18 +191,13 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> { path_info_service_addr, } => { // initialize stores - let blob_service: Arc<dyn BlobService> = - blobservice::from_addr(&blob_service_addr).await?.into(); - let directory_service: Arc<dyn DirectoryService> = - directoryservice::from_addr(&directory_service_addr) - .await? - .into(); - let path_info_service = pathinfoservice::from_addr( - &path_info_service_addr, - blob_service.clone(), - directory_service.clone(), - ) - .await?; + let (blob_service, directory_service, path_info_service) = + tvix_store::utils::construct_services( + blob_service_addr, + directory_service_addr, + path_info_service_addr, + ) + .await?; let listen_address = listen_address .unwrap_or_else(|| "[::]:8000".to_string()) @@ -225,7 +215,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> { GRPCDirectoryServiceWrapper::from(directory_service), )) .add_service(PathInfoServiceServer::new(GRPCPathInfoServiceWrapper::new( - path_info_service, + Arc::from(path_info_service), ))); #[cfg(feature = "tonic-reflection")] @@ -255,18 +245,13 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> { path_info_service_addr, } => { // FUTUREWORK: allow flat for single files? - let blob_service: Arc<dyn BlobService> = - blobservice::from_addr(&blob_service_addr).await?.into(); - let directory_service: Arc<dyn DirectoryService> = - directoryservice::from_addr(&directory_service_addr) - .await? - .into(); - let path_info_service = pathinfoservice::from_addr( - &path_info_service_addr, - blob_service.clone(), - directory_service.clone(), - ) - .await?; + let (blob_service, directory_service, path_info_service) = + tvix_store::utils::construct_services( + blob_service_addr, + directory_service_addr, + path_info_service_addr, + ) + .await?; // Arc the PathInfoService, as we clone it . let path_info_service: Arc<dyn PathInfoService> = path_info_service.into(); @@ -361,27 +346,19 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> { list_root, threads, } => { - let blob_service: Arc<dyn BlobService> = - blobservice::from_addr(&blob_service_addr).await?.into(); - let directory_service: Arc<dyn DirectoryService> = - directoryservice::from_addr(&directory_service_addr) - .await? - .into(); - let path_info_service = pathinfoservice::from_addr( - &path_info_service_addr, - blob_service.clone(), - directory_service.clone(), - ) - .await?; - - // Arc the PathInfoService, as TvixStoreFS requires Clone - let path_info_service: Arc<dyn PathInfoService> = path_info_service.into(); + let (blob_service, directory_service, path_info_service) = + tvix_store::utils::construct_services( + blob_service_addr, + directory_service_addr, + path_info_service_addr, + ) + .await?; let mut fuse_daemon = tokio::task::spawn_blocking(move || { let fs = make_fs( blob_service, directory_service, - path_info_service, + Arc::from(path_info_service), list_root, ); info!(mount_path=?dest, "mounting"); @@ -409,27 +386,19 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> { path_info_service_addr, list_root, } => { - let blob_service: Arc<dyn BlobService> = - blobservice::from_addr(&blob_service_addr).await?.into(); - let directory_service: Arc<dyn DirectoryService> = - directoryservice::from_addr(&directory_service_addr) - .await? - .into(); - let path_info_service = pathinfoservice::from_addr( - &path_info_service_addr, - blob_service.clone(), - directory_service.clone(), - ) - .await?; - - // Arc the PathInfoService, as TvixStoreFS requires Clone - let path_info_service: Arc<dyn PathInfoService> = path_info_service.into(); + let (blob_service, directory_service, path_info_service) = + tvix_store::utils::construct_services( + blob_service_addr, + directory_service_addr, + path_info_service_addr, + ) + .await?; tokio::task::spawn_blocking(move || { let fs = make_fs( blob_service, directory_service, - path_info_service, + Arc::from(path_info_service), list_root, ); info!(socket_path=?socket, "starting virtiofs-daemon"); diff --git a/tvix/store/src/lib.rs b/tvix/store/src/lib.rs index 5b57781c4dab..4bc7b5e02d57 100644 --- a/tvix/store/src/lib.rs +++ b/tvix/store/src/lib.rs @@ -1,6 +1,7 @@ pub mod nar; pub mod pathinfoservice; pub mod proto; +pub mod utils; #[cfg(test)] mod tests; diff --git a/tvix/store/src/utils.rs b/tvix/store/src/utils.rs new file mode 100644 index 000000000000..041a9e683d59 --- /dev/null +++ b/tvix/store/src/utils.rs @@ -0,0 +1,35 @@ +use std::sync::Arc; + +use tvix_castore::{ + blobservice::{self, BlobService}, + directoryservice::{self, DirectoryService}, +}; + +use crate::pathinfoservice::{self, PathInfoService}; + +/// Construct the three store handles from their addrs. +pub async fn construct_services( + blob_service_addr: impl AsRef<str>, + directory_service_addr: impl AsRef<str>, + path_info_service_addr: impl AsRef<str>, +) -> std::io::Result<( + Arc<dyn BlobService>, + Arc<dyn DirectoryService>, + Box<dyn PathInfoService>, +)> { + let blob_service: Arc<dyn BlobService> = blobservice::from_addr(blob_service_addr.as_ref()) + .await? + .into(); + let directory_service: Arc<dyn DirectoryService> = + directoryservice::from_addr(directory_service_addr.as_ref()) + .await? + .into(); + let path_info_service = pathinfoservice::from_addr( + path_info_service_addr.as_ref(), + blob_service.clone(), + directory_service.clone(), + ) + .await?; + + Ok((blob_service, directory_service, path_info_service)) +} |