use crate::error::Result;
use async_trait::async_trait;
use std::time::Duration;
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum BackendKind {
Moka,
DashMap,
Redis,
Chain,
Mock,
Unknown,
}
impl BackendKind {
pub fn is_memory(&self) -> bool {
matches!(self, BackendKind::Moka | BackendKind::DashMap | BackendKind::Mock)
}
pub fn is_distributed(&self) -> bool {
matches!(self, BackendKind::Redis)
}
}
#[async_trait]
pub trait CacheReader: Send + Sync + 'static {
async fn get(&self, key: &str) -> Result<Option<Vec<u8>>>;
async fn exists(&self, key: &str) -> Result<bool>;
async fn ttl(&self, key: &str) -> Result<Option<Duration>>;
async fn len(&self) -> Result<u64>;
async fn is_empty(&self) -> Result<bool> {
Ok(self.len().await?.eq(&0))
}
async fn capacity(&self) -> Result<u64>;
async fn stats(&self) -> Result<std::collections::HashMap<String, String>>;
async fn get_many(&self, keys: &[String]) -> Result<Vec<Option<Vec<u8>>>> {
let mut results = Vec::with_capacity(keys.len());
for key in keys {
results.push(self.get(key).await?);
}
Ok(results)
}
}
#[async_trait]
pub trait CacheWriter: Send + Sync + 'static {
async fn set(&self, key: &str, value: Vec<u8>, ttl: Option<Duration>) -> Result<()>;
async fn delete(&self, key: &str) -> Result<()>;
async fn clear(&self) -> Result<()>;
async fn expire(&self, key: &str, ttl: Duration) -> Result<bool>;
async fn set_many(&self, items: &[(String, Vec<u8>, Option<Duration>)]) -> Result<()> {
for (key, value, ttl) in items {
self.set(key, value.clone(), *ttl).await?;
}
Ok(())
}
async fn delete_many(&self, keys: &[String]) -> Result<()> {
for key in keys {
self.delete(key).await?;
}
Ok(())
}
}
#[async_trait]
pub trait CacheConnector: Send + Sync + 'static {
async fn health_check(&self) -> Result<()>;
async fn shutdown(&self);
fn backend_kind(&self) -> BackendKind;
#[cfg(feature = "lua-script")]
fn as_lua_executor(&self) -> Option<&dyn LuaExecutor> {
None
}
}
#[cfg(feature = "lua-script")]
#[async_trait]
pub trait LuaExecutor: Send + Sync {
async fn eval_lua(&self, script: &str, keys: &[&str], args: &[&str]) -> Result<redis::Value>;
async fn eval_sha(&self, sha: &str, keys: &[&str], args: &[&str]) -> Result<redis::Value>;
async fn script_load(&self, script: &str) -> Result<String>;
}
#[async_trait]
pub trait CacheBackend: CacheReader + CacheWriter + CacheConnector + 'static {}
#[async_trait]
impl<T: CacheReader + CacheWriter + CacheConnector + 'static> CacheBackend for T {}
#[cfg(test)]
mod tests {
use super::*;
use crate::testing::mock::MockBackend;
#[tokio::test]
async fn test_mock_backend() {
let backend = MockBackend::new("mock", 50, false);
backend.set("key1", b"value1".to_vec(), None).await.unwrap();
let value = backend.get("key1").await.unwrap();
assert_eq!(value, Some(b"value1".to_vec()));
assert!(backend.exists("key1").await.unwrap());
assert!(!backend.exists("key2").await.unwrap());
backend.delete("key1").await.unwrap();
assert!(!backend.exists("key1").await.unwrap());
backend.health_check().await.unwrap();
let stats = backend.stats().await.unwrap();
assert_eq!(stats.get("type"), Some(&"mock".to_string()));
}
#[tokio::test]
async fn test_isp_traits() {
let backend = MockBackend::new("mock", 50, false);
let reader: &dyn CacheReader = &backend;
assert!(reader.get("nonexistent").await.unwrap().is_none());
let writer: &dyn CacheWriter = &backend;
writer.set("key", b"value".to_vec(), None).await.unwrap();
let connector: &dyn CacheConnector = &backend;
connector.health_check().await.unwrap();
assert_eq!(connector.backend_kind(), BackendKind::Mock);
}
#[test]
fn test_backend_kind_is_memory_moka() {
assert!(BackendKind::Moka.is_memory());
}
#[test]
fn test_backend_kind_is_memory_dashmap() {
assert!(BackendKind::DashMap.is_memory());
}
#[test]
fn test_backend_kind_is_memory_mock() {
assert!(BackendKind::Mock.is_memory());
}
#[test]
fn test_backend_kind_is_memory_redis_false() {
assert!(!BackendKind::Redis.is_memory());
}
#[test]
fn test_backend_kind_is_memory_chain_false() {
assert!(!BackendKind::Chain.is_memory());
}
#[test]
fn test_backend_kind_is_memory_unknown_false() {
assert!(!BackendKind::Unknown.is_memory());
}
#[test]
fn test_backend_kind_is_distributed_redis() {
assert!(BackendKind::Redis.is_distributed());
}
#[test]
fn test_backend_kind_is_distributed_moka_false() {
assert!(!BackendKind::Moka.is_distributed());
}
#[test]
fn test_backend_kind_is_distributed_dashmap_false() {
assert!(!BackendKind::DashMap.is_distributed());
}
#[test]
fn test_backend_kind_is_distributed_chain_false() {
assert!(!BackendKind::Chain.is_distributed());
}
#[test]
fn test_backend_kind_is_distributed_mock_false() {
assert!(!BackendKind::Mock.is_distributed());
}
#[test]
fn test_backend_kind_is_distributed_unknown_false() {
assert!(!BackendKind::Unknown.is_distributed());
}
#[test]
fn test_backend_kind_debug() {
let kind = BackendKind::Moka;
let debug_str = format!("{:?}", kind);
assert!(debug_str.contains("Moka"));
}
#[test]
fn test_backend_kind_clone() {
let kind = BackendKind::Redis;
let cloned = kind.clone();
assert_eq!(kind, cloned);
}
#[test]
fn test_backend_kind_equality() {
assert_eq!(BackendKind::Moka, BackendKind::Moka);
assert_ne!(BackendKind::Moka, BackendKind::Redis);
}
#[tokio::test]
async fn test_cache_reader_is_empty_default() {
let backend = MockBackend::new("mock", 50, false);
let reader: &dyn CacheReader = &backend;
assert!(reader.is_empty().await.unwrap());
backend.set("key1", b"value1".to_vec(), None).await.unwrap();
assert!(!reader.is_empty().await.unwrap());
}
#[tokio::test]
async fn test_cache_reader_get_many_default() {
let backend = MockBackend::new("mock", 50, false);
backend.set("key1", b"value1".to_vec(), None).await.unwrap();
backend.set("key2", b"value2".to_vec(), None).await.unwrap();
let reader: &dyn CacheReader = &backend;
let keys = vec!["key1".to_string(), "key2".to_string(), "key3".to_string()];
let results = reader.get_many(&keys).await.unwrap();
assert_eq!(results.len(), 3);
assert_eq!(results[0], Some(b"value1".to_vec()));
assert_eq!(results[1], Some(b"value2".to_vec()));
assert_eq!(results[2], None);
}
#[tokio::test]
async fn test_cache_reader_get_many_empty() {
let backend = MockBackend::new("mock", 50, false);
let reader: &dyn CacheReader = &backend;
let keys: Vec<String> = vec![];
let results = reader.get_many(&keys).await.unwrap();
assert!(results.is_empty());
}
#[tokio::test]
async fn test_cache_writer_set_many_default() {
let backend = MockBackend::new("mock", 50, false);
let writer: &dyn CacheWriter = &backend;
let items = vec![
("key1".to_string(), b"value1".to_vec(), None),
("key2".to_string(), b"value2".to_vec(), None),
];
writer.set_many(&items).await.unwrap();
assert!(backend.exists("key1").await.unwrap());
assert!(backend.exists("key2").await.unwrap());
}
#[tokio::test]
async fn test_cache_writer_delete_many_default() {
let backend = MockBackend::new("mock", 50, false);
backend.set("key1", b"value1".to_vec(), None).await.unwrap();
backend.set("key2", b"value2".to_vec(), None).await.unwrap();
let writer: &dyn CacheWriter = &backend;
let keys = vec!["key1".to_string(), "key2".to_string()];
writer.delete_many(&keys).await.unwrap();
assert!(!backend.exists("key1").await.unwrap());
assert!(!backend.exists("key2").await.unwrap());
}
#[tokio::test]
async fn test_cache_connector_backend_kind_mock() {
let backend = MockBackend::new("mock", 50, false);
let connector: &dyn CacheConnector = &backend;
assert_eq!(connector.backend_kind(), BackendKind::Mock);
}
#[tokio::test]
async fn test_cache_connector_shutdown() {
let backend = MockBackend::new("mock", 50, false);
let connector: &dyn CacheConnector = &backend;
connector.shutdown().await;
}
#[tokio::test]
async fn test_cache_backend_trait_object() {
let backend = MockBackend::new("mock", 50, false);
let backend_dyn: &dyn CacheBackend = &backend;
backend_dyn.set("key", b"value".to_vec(), None).await.unwrap();
let value = backend_dyn.get("key").await.unwrap();
assert_eq!(value, Some(b"value".to_vec()));
}
}