47 lines
1 KiB
Rust
47 lines
1 KiB
Rust
|
//! Low memory caching stuff
|
||
|
|
|
||
|
|
use std::path::{Path, PathBuf};
|
||
|
|
|
||
|
|
use async_trait::async_trait;
|
||
|
|
use bytes::Bytes;
|
||
|
|
use futures::Stream;
|
||
|
|
use lru::LruCache;
|
||
|
|
|
||
|
|
use super::{fs::FromFsStream, Cache, CacheKey};
|
||
|
|
|
||
|
|
pub struct LowMemCache {
|
||
|
|
on_disk: LruCache<CacheKey, ()>,
|
||
|
|
disk_path: PathBuf,
|
||
|
|
disk_max_size: usize,
|
||
|
|
disk_cur_size: usize,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl LowMemCache {
|
||
|
|
pub fn new(disk_max_size: usize, disk_path: PathBuf) -> Self {
|
||
|
|
Self {
|
||
|
|
on_disk: LruCache::unbounded(),
|
||
|
|
disk_path,
|
||
|
|
disk_max_size,
|
||
|
|
disk_cur_size: 0,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#[async_trait]
|
||
|
|
impl Cache for LowMemCache {
|
||
|
|
async fn get_stream(&mut self, key: &CacheKey) -> Option<Result<FromFsStream, std::io::Error>> {
|
||
|
|
if self.on_disk.get(key).is_some() {
|
||
|
|
super::fs::read_file(&Path::new(&key.to_string())).await
|
||
|
|
} else {
|
||
|
|
None
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
async fn put_stream(
|
||
|
|
&mut self,
|
||
|
|
key: CacheKey,
|
||
|
|
image: impl Stream<Item = Result<Bytes, reqwest::Error>> + Unpin + Send + 'static,
|
||
|
|
) {
|
||
|
|
}
|
||
|
|
}
|