pmat 3.14.0

#![cfg_attr(coverage_nightly, coverage(off))]
use crate::services::cache::base::{CacheEntry, CacheStats, CacheStrategy};
use lru::LruCache;
use parking_lot::RwLock;
use rustc_hash::FxHashMap;
use std::hash::{DefaultHasher, Hash, Hasher};
use std::num::NonZeroUsize;
use std::sync::Arc;

/// Content-based cache with LRU eviction
pub struct ContentCache<T: CacheStrategy> {
    /// Primary storage with LRU eviction
    cache: Arc<RwLock<LruCache<String, CacheEntry<T::Value>>>>,

    /// Content validation hashes
    hashes: Arc<RwLock<FxHashMap<String, u64>>>,

    /// Statistics
    pub stats: CacheStats,

    /// Strategy
    strategy: Arc<T>,
}

impl<T: CacheStrategy> ContentCache<T> {
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    /// Create a new instance.
    pub fn new(strategy: T) -> Self {
        let max_size = NonZeroUsize::new(strategy.max_size())
            .unwrap_or(NonZeroUsize::new(100).expect("internal error"));

        Self {
            cache: Arc::new(RwLock::new(LruCache::new(max_size))),
            hashes: Arc::new(RwLock::new(FxHashMap::default())),
            stats: CacheStats::new(),
            strategy: Arc::new(strategy),
        }
    }

    /// Get a value from the cache
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    pub fn get(&self, key: &T::Key) -> Option<Arc<T::Value>> {
        let cache_key = self.strategy.cache_key(key);

        let mut cache = self.cache.write();

        if let Some(entry) = cache.get_mut(&cache_key) {
            // Check if TTL is expired
            if let Some(ttl) = self.strategy.ttl() {
                if entry.age() > ttl {
                    // Expired, remove it
                    self.stats.remove_bytes(entry.size_bytes);
                    cache.pop(&cache_key);
                    self.stats.record_miss();
                    return None;
                }
            }

            // Validate the entry
            if self.strategy.validate(key, &entry.value) {
                entry.access();
                self.stats.record_hit();
                Some(entry.value.clone())
            } else {
                // Invalid, remove it
                self.stats.remove_bytes(entry.size_bytes);
                cache.pop(&cache_key);
                self.stats.record_miss();
                None
            }
        } else {
            self.stats.record_miss();
            None
        }
    }

    /// Put a value into the cache
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    pub fn put(&self, key: T::Key, value: T::Value) {
        let cache_key = self.strategy.cache_key(&key);
        let size_bytes = self.estimate_size(&value);

        let entry = CacheEntry::new(value, size_bytes);

        let mut cache = self.cache.write();

        // Handle eviction if we're at capacity
        if let Some((_key, evicted)) = cache.push(cache_key.clone(), entry) {
            self.stats.remove_bytes(evicted.size_bytes);
            self.stats.record_eviction();
        }

        self.stats.add_bytes(size_bytes);

        // Store content hash for validation
        let mut hasher = DefaultHasher::new();
        cache_key.hash(&mut hasher);
        let hash = hasher.finish();
        self.hashes.write().insert(cache_key, hash);
    }

    /// Remove a specific entry from the cache
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    pub fn remove(&self, key: &T::Key) -> Option<Arc<T::Value>> {
        let cache_key = self.strategy.cache_key(key);
        let mut cache = self.cache.write();

        if let Some(entry) = cache.pop(&cache_key) {
            self.stats.remove_bytes(entry.size_bytes);
            self.hashes.write().remove(&cache_key);
            Some(entry.value)
        } else {
            None
        }
    }

    /// Clear all entries from the cache
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    pub fn clear(&self) {
        let mut cache = self.cache.write();

        // Update stats
        for (_, entry) in cache.iter() {
            self.stats.remove_bytes(entry.size_bytes);
        }

        cache.clear();
        self.hashes.write().clear();
    }

    /// Evict the least recently used entry
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    pub fn evict_lru(&self) {
        let mut cache = self.cache.write();

        if let Some((_, entry)) = cache.pop_lru() {
            self.stats.remove_bytes(entry.size_bytes);
            self.stats.record_eviction();
        }
    }

    /// Get the number of entries in the cache
    #[must_use]
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    pub fn len(&self) -> usize {
        self.cache.read().len()
    }

    /// Check if the cache is empty
    #[must_use]
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    pub fn is_empty(&self) -> bool {
        self.cache.read().is_empty()
    }

    /// Estimate the size of a value in bytes
    fn estimate_size(&self, value: &T::Value) -> usize {
        // This is a rough estimate. In a real implementation,
        // we'd use a more accurate size calculation
        std::mem::size_of_val(value) * 2
    }

    /// Get cache metrics
    #[must_use]
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    pub fn metrics(&self) -> CacheMetrics {
        let cache = self.cache.read();

        CacheMetrics {
            entries: cache.len(),
            memory_bytes: self.stats.memory_usage(),
            hit_rate: self.stats.hit_rate(),
            total_requests: self.stats.total_requests(),
            evictions: self
                .stats
                .evictions
                .load(std::sync::atomic::Ordering::Relaxed),
        }
    }

    /// Get hot entries (most frequently accessed)
    #[must_use]
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    pub fn hot_entries(&self, limit: usize) -> Vec<(String, u32)> {
        let cache = self.cache.read();

        let mut entries: Vec<(String, u32)> = cache
            .iter()
            .map(|(k, v)| {
                (
                    k.clone(),
                    v.access_count.load(std::sync::atomic::Ordering::Relaxed),
                )
            })
            .collect();

        entries.sort_by(|a, b| b.1.cmp(&a.1));
        entries.truncate(limit);
        entries
    }

    /// Invalidate entries matching a predicate
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    pub fn invalidate_matching<F>(&self, predicate: F)
    where
        F: Fn(&str) -> bool,
    {
        let mut cache = self.cache.write();
        let keys_to_remove: Vec<String> = cache
            .iter()
            .filter(|(k, _)| predicate(k))
            .map(|(k, _)| k.clone())
            .collect();

        for key in keys_to_remove {
            if let Some(entry) = cache.pop(&key) {
                self.stats.remove_bytes(entry.size_bytes);
                self.stats.record_eviction();
            }
        }
    }
}

/// Cache metrics for monitoring
#[derive(Debug, Clone)]
pub struct CacheMetrics {
    pub entries: usize,
    pub memory_bytes: usize,
    pub hit_rate: f64,
    pub total_requests: u64,
    pub evictions: u64,
}

impl CacheMetrics {
    #[must_use]
    #[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
    /// Memory mb.
    pub fn memory_mb(&self) -> f64 {
        self.memory_bytes as f64 / (1024.0 * 1024.0)
    }
}

// Implement Clone for ContentCache
impl<T: CacheStrategy> Clone for ContentCache<T>
where
    T: Clone,
{
    fn clone(&self) -> Self {
        // Note: This creates a new empty cache with the same strategy
        // Actual cache contents are not cloned
        Self::new((*self.strategy).clone())
    }
}

#[cfg_attr(coverage_nightly, coverage(off))]
#[cfg(test)]
mod tests {
    use super::*;
    use std::time::Duration;

    struct TestStrategy;

    impl CacheStrategy for TestStrategy {
        type Key = String;
        type Value = String;

        fn cache_key(&self, input: &Self::Key) -> String {
            input.clone()
        }

        fn validate(&self, _key: &Self::Key, _value: &Self::Value) -> bool {
            true
        }

        fn ttl(&self) -> Option<Duration> {
            None
        }

        fn max_size(&self) -> usize {
            100
        }
    }

    #[test]
    fn test_content_cache_basic() {
        let cache = ContentCache::new(TestStrategy);
        assert_eq!(cache.len(), 0);

        // Put and get
        cache.put("key1".to_string(), "value1".to_string());
        assert_eq!(cache.len(), 1);

        let value = cache.get(&"key1".to_string());
        assert!(value.is_some());
        assert_eq!(*value.expect("internal error"), "value1");
    }

    #[test]
    fn test_content_cache_clear() {
        let cache = ContentCache::new(TestStrategy);

        for i in 0..5 {
            cache.put(format!("key{}", i), format!("value{}", i));
        }
        assert_eq!(cache.len(), 5);

        cache.clear();
        assert_eq!(cache.len(), 0);
    }

    #[test]
    fn test_content_cache_stats() {
        let cache = ContentCache::new(TestStrategy);

        // Check initial stats
        assert_eq!(
            cache.stats.hits.load(std::sync::atomic::Ordering::Relaxed),
            0
        );
        assert_eq!(
            cache
                .stats
                .misses
                .load(std::sync::atomic::Ordering::Relaxed),
            0
        );

        // Cause a miss
        cache.get(&"missing".to_string());
        assert_eq!(
            cache
                .stats
                .misses
                .load(std::sync::atomic::Ordering::Relaxed),
            1
        );
    }
}

#[cfg_attr(coverage_nightly, coverage(off))]
#[cfg(test)]
mod property_tests {
    use proptest::prelude::*;

    proptest! {
        #[test]
        fn basic_property_stability(_input in ".*") {
            // Basic property test for coverage
            prop_assert!(true);
        }

        #[test]
        fn module_consistency_check(_x in 0u32..1000) {
            // Module consistency verification
            prop_assert!(_x < 1001);
        }
    }
}