moniof 0.1.0

Actix middleware to monitor over-fetching and detect N+1 / overfetch database patterns (Mongo + SQL-agnostic)
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

use ahash::AHashMap;
use parking_lot::Mutex;
use std::sync::Arc;
use time::{Duration, OffsetDateTime};

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum QueryKind { Mongo, Sql, Other }

#[derive(Debug)]
pub struct QueryStats {
    pub started_at: OffsetDateTime,
    pub total: usize,
    pub per_key: AHashMap<String, usize>,

    pub total_db_latency_ms: u128,
    pub per_key_latency_ms: AHashMap<String, u128>,
    pub per_key_max_latency_ms: AHashMap<String, u128>,
}

impl QueryStats {
    pub fn new() -> Self {
        Self {
            started_at: OffsetDateTime::now_utc(),
            total: 0,
            per_key: AHashMap::new(),
            total_db_latency_ms: 0,
            per_key_latency_ms: AHashMap::new(),
            per_key_max_latency_ms: AHashMap::new(),
        }
    }

    pub fn record(&mut self, key: &str) {
        self.total += 1;
        *self.per_key.entry(key.to_string()).or_insert(0) += 1;
    }

    pub fn record_latency(&mut self, key: &str, ms: u128) {
        self.total_db_latency_ms += ms;
        *self.per_key_latency_ms.entry(key.to_string()).or_insert(0) += ms;
        let e = self.per_key_max_latency_ms.entry(key.to_string()).or_insert(0);
        if ms > *e { *e = ms; }
    }

    pub fn elapsed(&self) -> Duration {
        OffsetDateTime::now_utc() - self.started_at
    }
}

#[derive(Clone)]
pub struct QueryStatsHandle(pub Arc<Mutex<QueryStats>>);
impl QueryStatsHandle {
    pub fn new() -> Self { Self(Arc::new(Mutex::new(QueryStats::new()))) }
}

// SQL normalization helper (used by sqlx layer)
pub fn normalize_sql(sql: &str) -> String {
    let mut reduced = sql.split_whitespace().collect::<Vec<_>>().join(" ");
    if reduced.len() > 200 { reduced.truncate(200); }
    reduced.to_lowercase()
}