solidb 1.0.2

A lightweight, high-performance structured database server written in Rust.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

//! In-memory cache for service lookups to avoid hitting RocksDB on every script request.

use crate::scripting::Service;
use dashmap::DashMap;
use std::time::Instant;

/// Cached service entry with expiration
struct CachedService {
    service: Service,
    cached_at: Instant,
}

/// Cache for service metadata, keyed by (database, service_key).
///
/// Services rarely change, so a short TTL (5s) avoids stale data
/// while eliminating ~99% of RocksDB reads on the hot path.
pub struct ServiceCache {
    entries: DashMap<(String, String), CachedService>,
    ttl_secs: u64,
}

impl ServiceCache {
    pub fn new(ttl_secs: u64) -> Self {
        Self {
            entries: DashMap::new(),
            ttl_secs,
        }
    }

    /// Get a cached service if present and not expired.
    pub fn get(&self, db_name: &str, service_key: &str) -> Option<Service> {
        let key = (db_name.to_string(), service_key.to_string());
        if let Some(entry) = self.entries.get(&key) {
            if entry.cached_at.elapsed().as_secs() < self.ttl_secs {
                return Some(entry.service.clone());
            }
            // Expired — drop the ref before removing
            drop(entry);
            self.entries.remove(&key);
        }
        None
    }

    /// Insert or update a cached service.
    pub fn insert(&self, db_name: &str, service_key: &str, service: Service) {
        let key = (db_name.to_string(), service_key.to_string());
        self.entries.insert(
            key,
            CachedService {
                service,
                cached_at: Instant::now(),
            },
        );
    }

    /// Invalidate a specific service entry.
    pub fn invalidate(&self, db_name: &str, service_key: &str) {
        self.entries
            .remove(&(db_name.to_string(), service_key.to_string()));
    }
}