use super::PathInfoService; use crate::{ blobservice::BlobService, directoryservice::DirectoryService, nar::calculate_size_and_sha256, proto, Error, }; use std::{ collections::HashMap, sync::{Arc, RwLock}, }; pub struct MemoryPathInfoService { db: Arc>>, blob_service: Arc, directory_service: Arc, } impl MemoryPathInfoService { pub fn new( blob_service: Arc, directory_service: Arc, ) -> Self { Self { db: Default::default(), blob_service, directory_service, } } } impl PathInfoService for MemoryPathInfoService { fn get(&self, digest: [u8; 20]) -> Result, Error> { let db = self.db.read().unwrap(); match db.get(&digest) { None => Ok(None), Some(path_info) => Ok(Some(path_info.clone())), } } fn put(&self, path_info: proto::PathInfo) -> Result { // Call validate on the received PathInfo message. match path_info.validate() { Err(e) => Err(Error::InvalidRequest(format!( "failed to validate PathInfo: {}", e ))), // In case the PathInfo is valid, and we were able to extract a NixPath, store it in the database. // This overwrites existing PathInfo objects. Ok(nix_path) => { let mut db = self.db.write().unwrap(); db.insert(nix_path.digest, path_info.clone()); Ok(path_info) } } } fn calculate_nar(&self, root_node: &proto::node::Node) -> Result<(u64, [u8; 32]), Error> { calculate_size_and_sha256( root_node, self.blob_service.clone(), self.directory_service.clone(), ) .map_err(|e| Error::StorageError(e.to_string())) } }