1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
#![cfg(not(feature = "web"))] use crate::fetch::{FetchCancelReason, FetchEngine, FetchProcess, FetchProcessReader, FetchStatus}; use std::path::{Path, PathBuf}; #[derive(Default, Clone)] pub struct FsFetchEngine { root_path: PathBuf, } impl FsFetchEngine { pub fn new<S: AsRef<Path>>(root_path: &S) -> Self { Self { root_path: root_path.as_ref().into(), } } } impl FetchEngine for FsFetchEngine { fn fetch(&mut self, path: &str) -> Result<Box<dyn FetchProcessReader>, FetchStatus> { #[cfg(feature = "parallel")] { let path = self.root_path.join(path); let process = FetchProcess::new_start(); let mut p = process.clone(); rayon::spawn(move || { if let Ok(bytes) = std::fs::read(path) { p.done(bytes); } else { p.cancel(FetchCancelReason::Error); } }); Ok(Box::new(process)) } #[cfg(not(feature = "parallel"))] { if let Ok(bytes) = std::fs::read(self.root_path.join(path)) { Ok(Box::new(FetchProcess::new_done(bytes))) } else { Err(FetchStatus::Canceled(FetchCancelReason::Error)) } } } }