rustminidb 0.1.0

A lightweight embedded database with native REST API
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113

//! RustMinidb 监控与埋点系统
//!
//! 统一收集：查询次数、执行时间、错误次数、慢查询

use std::sync::atomic::{AtomicU64, Ordering};
use std::sync::Arc;

/// 监控指标
#[derive(Default)]
pub struct Metrics {
    // 查询统计
    pub total_queries: AtomicU64,
    pub successful_queries: AtomicU64,
    pub failed_queries: AtomicU64,
    pub total_query_time_ns: AtomicU64,
    pub slow_queries: AtomicU64,

    // 行统计
    pub rows_inserted: AtomicU64,
    pub rows_deleted: AtomicU64,
    pub rows_returned: AtomicU64,

    // 连接
    pub tables_created: AtomicU64,
    pub tables_dropped: AtomicU64,
}

impl Metrics {
    pub fn new() -> Arc<Self> {
        Arc::new(Metrics::default())
    }

    /// 记录一次查询执行
    pub fn record_query(&self, success: bool, duration: std::time::Duration, rows: usize) {
        self.total_queries.fetch_add(1, Ordering::Relaxed);
        if success {
            self.successful_queries.fetch_add(1, Ordering::Relaxed);
        } else {
            self.failed_queries.fetch_add(1, Ordering::Relaxed);
        }
        self.total_query_time_ns
            .fetch_add(duration.as_nanos() as u64, Ordering::Relaxed);
        self.rows_returned.fetch_add(rows as u64, Ordering::Relaxed);

        // 慢查询阈值：500ms
        if duration.as_millis() > 500 {
            self.slow_queries.fetch_add(1, Ordering::Relaxed);
        }
    }

    /// 获取指标快照
    pub fn snapshot(&self) -> MetricsSnapshot {
        let total = self.total_queries.load(Ordering::Relaxed);
        MetricsSnapshot {
            total_queries: total,
            successful_queries: self.successful_queries.load(Ordering::Relaxed),
            failed_queries: self.failed_queries.load(Ordering::Relaxed),
            avg_query_time_ms: if total > 0 {
                (self.total_query_time_ns.load(Ordering::Relaxed) / total) as f64 / 1_000_000.0
            } else {
                0.0
            },
            slow_queries: self.slow_queries.load(Ordering::Relaxed),
            rows_inserted: self.rows_inserted.load(Ordering::Relaxed),
            rows_deleted: self.rows_deleted.load(Ordering::Relaxed),
            rows_returned: self.rows_returned.load(Ordering::Relaxed),
            tables_created: self.tables_created.load(Ordering::Relaxed),
            tables_dropped: self.tables_dropped.load(Ordering::Relaxed),
        }
    }
}

#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
pub struct MetricsSnapshot {
    pub total_queries: u64,
    pub successful_queries: u64,
    pub failed_queries: u64,
    pub avg_query_time_ms: f64,
    pub slow_queries: u64,
    pub rows_inserted: u64,
    pub rows_deleted: u64,
    pub rows_returned: u64,
    pub tables_created: u64,
    pub tables_dropped: u64,
}

/// 启动 Banner
pub fn print_banner(version: &str, port: u16, db_dir: &str, db_name: &str) {
    let banner = format!(
        r#"
  ╔══════════════════════════════════════════════════╗
  ║                                                  ║
  ║     ____                  _   _  __ _     _      ║
  ║    |  _ \ _   _ _ __ ___ | |_(_)/ _(_) __| |     ║
  ║    | |_) | | | | '_ ` _ \| __| | |_| |/ _` |     ║
  ║    |  _ <| |_| | | | | | | |_| |  _| | (_| |     ║
  ║    |_| \_\\__,_|_| |_| |_|\__|_|_| |_|\__,_|     ║
  ║                                                  ║
  ║    v{:<8}  Lightweight Embedded Database       ║
  ║    Port: {:<6}  Database: {:<16} ║
  ║    Dir:  {:<35}║
  ║                                                  ║
  ║    🌐 Web UI  : http://localhost:{:<4}/          ║
  ║    📡 API     : http://localhost:{:<4}/v1/query  ║
  ║    📁 Storage : redb (ACID, single-file)         ║
  ║    📜 License : BSL-1.1                          ║
  ║                                                  ║
  ╚══════════════════════════════════════════════════╝
"#,
        version, port, db_name, db_dir, port, port
    );
    println!("{}", banner);
}