mii/wasmer
1
0
Fork 0
mirror of https://github.com/mii443/wasmer.git synced 2025-12-07 05:08:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improved cache system on Windows

Browse Source
This commit is contained in:
Syrus Akbary
2020-06-11 22:23:20 -07:00
parent f088136810
commit b6a8313ea2
3 changed files with 59 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
lib/cache/src/filesystem.rs vendored
View File

@@ -31,6 +31,7 @@ use wasmer::{DeserializeError, Module, SerializeError, Store};
/// ```
pub struct FileSystemCache {
path: PathBuf,
ext: Option<String>,
}
impl FileSystemCache {
@@ -41,7 +42,7 @@ impl FileSystemCache {
let metadata = path.metadata()?;
if metadata.is_dir() {
if !metadata.permissions().readonly() {
Ok(Self { path })
Ok(Self { path, ext: None })
} else {
// This directory is readonly.
Err(io::Error::new(
@@ -62,9 +63,17 @@ impl FileSystemCache {
} else {
// Create the directory and any parent directories if they don't yet exist.
create_dir_all(&path)?;
Ok(Self { path })
Ok(Self { path, ext: None })
}
}
/// Set's the extension for this cached file.
///
/// This is needed for loading Native files from Windows, as otherwise
/// loadding the library will fail (it requires to have a `.dll` extension)
pub fn set_cache_extension(&mut self, ext: Option<impl AsRef<str>>) {
self.ext = ext.map(|ext| ext.as_ref().to_string());
}
}
impl Cache for FileSystemCache {
@@ -72,20 +81,25 @@ impl Cache for FileSystemCache {
type SerializeError = SerializeError;
unsafe fn load(&self, store: &Store, key: WasmHash) -> Result<Module, Self::DeserializeError> {
let filename = key.to_string();
let mut new_path_buf = self.path.clone();
new_path_buf.push(filename);
Module::deserialize_from_file(&store, new_path_buf)
let filename = if let Some(ref ext) = self.ext {
format!("{}.{}", key.to_string(), ext)
} else {
key.to_string()
};
let path = self.path.join(filename);
Module::deserialize_from_file(&store, path)
}
fn store(&mut self, key: WasmHash, module: &Module) -> Result<(), Self::SerializeError> {
let filename = key.to_string();
let mut new_path_buf = self.path.clone();
let filename = if let Some(ref ext) = self.ext {
format!("{}.{}", key.to_string(), ext)
} else {
key.to_string()
};
let path = self.path.join(filename);
let mut file = File::create(path)?;
let buffer = module.serialize()?;
new_path_buf.push(filename);
let mut file = File::create(new_path_buf)?;
file.write_all(&buffer)?;
Ok(())