nodedb 0.0.0-beta.1

Local-first, real-time, edge-to-cloud hybrid database for multi-modal workloads
Documentation 
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
62
63
64
65
66
67
68

//! Per-tenant quota usage tracking.

/// Per-tenant resource quota usage.
#[derive(Debug)]
pub struct TenantQuotaMetrics {
    pub tenant_id: u32,
    pub memory_bytes_used: u64,
    pub memory_bytes_limit: u64,
    pub storage_bytes_used: u64,
    pub storage_bytes_limit: u64,
    pub qps_current: u64,
    pub qps_limit: u64,
    pub connections_active: u64,
    pub connections_limit: u64,
}

impl TenantQuotaMetrics {
    /// Whether any quota is exceeded.
    pub fn is_over_quota(&self) -> bool {
        (self.memory_bytes_limit > 0 && self.memory_bytes_used > self.memory_bytes_limit)
            || (self.storage_bytes_limit > 0 && self.storage_bytes_used > self.storage_bytes_limit)
            || (self.qps_limit > 0 && self.qps_current > self.qps_limit)
            || (self.connections_limit > 0 && self.connections_active > self.connections_limit)
    }

    /// Utilization as a percentage (0–100) of the most constrained resource.
    pub fn max_utilization_pct(&self) -> u8 {
        let mut max = 0.0f64;
        if self.memory_bytes_limit > 0 {
            max = max.max(self.memory_bytes_used as f64 / self.memory_bytes_limit as f64);
        }
        if self.storage_bytes_limit > 0 {
            max = max.max(self.storage_bytes_used as f64 / self.storage_bytes_limit as f64);
        }
        if self.qps_limit > 0 {
            max = max.max(self.qps_current as f64 / self.qps_limit as f64);
        }
        (max * 100.0).min(100.0) as u8
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn tenant_quota_check() {
        let q = TenantQuotaMetrics {
            tenant_id: 1,
            memory_bytes_used: 800,
            memory_bytes_limit: 1000,
            storage_bytes_used: 500,
            storage_bytes_limit: 1000,
            qps_current: 50,
            qps_limit: 100,
            connections_active: 5,
            connections_limit: 10,
        };
        assert!(!q.is_over_quota());
        assert_eq!(q.max_utilization_pct(), 80);

        let over = TenantQuotaMetrics {
            memory_bytes_used: 1100,
            ..q
        };
        assert!(over.is_over_quota());
    }
}