use std::fmt;
use crate::{CacheManager, HttpResponse, Result};
use foyer::{HybridCache, HybridCacheBuilder};
use http_cache_semantics::CachePolicy;
use serde::{Deserialize, Serialize};
#[cfg_attr(docsrs, doc(cfg(feature = "manager-foyer")))]
#[derive(Clone)]
pub struct FoyerManager {
cache: HybridCache<String, Vec<u8>>,
}
impl fmt::Debug for FoyerManager {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f.debug_struct("FoyerManager").finish_non_exhaustive()
}
}
#[derive(Debug, Deserialize, Serialize)]
struct Store {
response: HttpResponse,
policy: CachePolicy,
}
impl FoyerManager {
pub fn new(cache: HybridCache<String, Vec<u8>>) -> Self {
Self { cache }
}
pub async fn in_memory(capacity: usize) -> Result<Self> {
let cache: HybridCache<String, Vec<u8>> = HybridCacheBuilder::new()
.memory(capacity)
.storage() .build()
.await
.map_err(|e| crate::HttpCacheError::cache(e.to_string()))?;
Ok(Self { cache })
}
pub async fn close(&self) -> Result<()> {
self.cache
.close()
.await
.map_err(|e| crate::HttpCacheError::cache(e.to_string()))?;
Ok(())
}
}
impl CacheManager for FoyerManager {
async fn get(
&self,
cache_key: &str,
) -> Result<Option<(HttpResponse, CachePolicy)>> {
match self.cache.get(&cache_key.to_string()).await {
Ok(Some(entry)) => {
match postcard::from_bytes::<Store>(entry.value()) {
Ok(store) => Ok(Some((store.response, store.policy))),
Err(e) => {
log::debug!(
"Failed to deserialize cache entry for key '{}': {}",
cache_key,
e
);
Ok(None)
}
}
}
Ok(None) => Ok(None),
Err(_) => Ok(None),
}
}
async fn put(
&self,
cache_key: String,
response: HttpResponse,
policy: CachePolicy,
) -> Result<HttpResponse> {
let data = Store { response, policy };
let bytes = postcard::to_allocvec(&data)?;
self.cache.insert(cache_key, bytes);
Ok(data.response)
}
async fn delete(&self, cache_key: &str) -> Result<()> {
self.cache.remove(&cache_key.to_string());
Ok(())
}
}