use crate::error::CacheError;
#[cfg(feature = "bloom-filter")]
use murmur3::murmur3_32;
const DEFAULT_MAX_KEY_LENGTH: usize = 256;
const DEFAULT_NAMESPACE: &str = "default";
const VALID_KEY_CHARS: &[char] = &[
'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's',
't', 'u', 'v', 'w', 'x', 'y', 'z', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L',
'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', '0', '1', '2', '3', '4',
'5', '6', '7', '8', '9', '-', '_', '.', ':', '/', '@',
];
#[derive(Clone, Debug)]
pub struct KeyGenerator {
namespace: String,
prefix: String,
max_key_length: usize,
}
impl Default for KeyGenerator {
fn default() -> Self {
Self::new()
}
}
impl KeyGenerator {
pub fn new() -> Self {
Self {
namespace: DEFAULT_NAMESPACE.to_string(),
prefix: String::new(),
max_key_length: DEFAULT_MAX_KEY_LENGTH,
}
}
pub fn with_prefix(prefix: &str) -> Self {
Self {
namespace: DEFAULT_NAMESPACE.to_string(),
prefix: prefix.to_string(),
max_key_length: DEFAULT_MAX_KEY_LENGTH,
}
}
pub fn with_namespace(mut self, namespace: &str) -> Self {
self.namespace = namespace.to_string();
self
}
pub fn with_prefix_str(mut self, prefix: &str) -> Self {
self.prefix = prefix.to_string();
self
}
pub fn with_max_key_length(mut self, length: usize) -> Self {
self.max_key_length = length;
self
}
pub fn with_eviction_policy(self, _policy: crate::EvictionPolicy) -> Self {
self
}
pub fn generate(&self, template: &str, params: &[(&str, &str)]) -> String {
let mut result = template.to_string();
for (key, value) in params {
let placeholder = format!("{{{}}}", key);
result = result.replace(&placeholder, value);
}
result
}
pub fn generate_full(&self, template: &str, params: &[(&str, &str)]) -> String {
let key = self.generate(template, params);
let prefixed = self.apply_prefix(&key);
self.namespaced_key(&prefixed)
}
fn apply_prefix(&self, key: &str) -> String {
if self.prefix.is_empty() {
key.to_string()
} else {
format!("{}{}", self.prefix, key)
}
}
pub fn validate_key(&self, key: &str) -> Result<(), CacheError> {
if key.is_empty() {
return Err(CacheError::InvalidInput(
"Cache key cannot be empty".to_string(),
));
}
if key.len() > self.max_key_length {
return Err(CacheError::InvalidInput(format!(
"Cache key exceeds maximum length of {} characters",
self.max_key_length
)));
}
for c in key.chars() {
if !VALID_KEY_CHARS.contains(&c) {
return Err(CacheError::InvalidInput(format!(
"Cache key contains invalid character: '{}'",
c
)));
}
}
Ok(())
}
#[cfg(feature = "bloom-filter")]
pub fn generate_fingerprint(&self, key: &str) -> String {
let key_bytes = key.as_bytes();
let hash = murmur3_32(&mut &key_bytes[..], 0).unwrap_or(0);
format!("_fp{:08x}", hash)
}
#[cfg(feature = "bloom-filter")]
pub fn normalize(&self, key: &str) -> String {
let key = key.trim().to_string();
if key.len() <= self.max_key_length {
key
} else {
let fingerprint = self.generate_fingerprint(&key);
let max_key_length = self.max_key_length.saturating_sub(fingerprint.len());
let truncated = &key[..max_key_length.max(1)];
format!("{}{}", truncated, fingerprint)
}
}
pub fn namespaced_key(&self, key: &str) -> String {
if self.namespace.is_empty() || self.namespace == DEFAULT_NAMESPACE {
key.to_string()
} else {
format!("{}:{}", self.namespace, key)
}
}
}
#[allow(dead_code)]
pub fn generate_cache_key(template: &str, params: &[(&str, &str)]) -> String {
let generator = KeyGenerator::new();
generator.generate(template, params)
}
#[allow(dead_code)]
pub fn generate_namespaced_key(namespace: &str, template: &str, params: &[(&str, &str)]) -> String {
let generator = KeyGenerator::new().with_namespace(namespace);
generator.generate_full(template, params)
}
#[allow(dead_code)]
pub fn validate_cache_key(key: &str) -> Result<(), CacheError> {
let generator = KeyGenerator::new();
generator.validate_key(key)
}
#[cfg(feature = "bloom-filter")]
#[allow(dead_code)]
pub fn normalize_cache_key(key: &str) -> String {
let generator = KeyGenerator::new();
generator.normalize(key)
}