torsh-profiler 0.1.2

Performance profiling and monitoring for ToRSh
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
69
70
71
72
73
74

//! Alert rules and rule management
//!
//! This module contains the alert rule definitions and helper functions
//! for creating common alert rules.

use super::core::{AlertCondition, AlertSeverity};
use super::notifications::NotificationChannel;
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use std::time::Duration;

/// Alert rule configuration
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct AlertRule {
    pub id: String,
    pub name: String,
    pub description: String,
    pub condition: AlertCondition,
    pub severity: AlertSeverity,
    pub channels: Vec<NotificationChannel>,
    pub enabled: bool,
    pub cooldown_duration: Duration,
    pub max_alerts_per_hour: u32,
    pub tags: HashMap<String, String>,
}

/// Create a duration-based alert rule
pub fn create_duration_alert_rule(
    operation: String,
    max_duration_ms: f64,
    severity: AlertSeverity,
    channels: Vec<NotificationChannel>,
) -> AlertRule {
    AlertRule {
        id: format!("duration_{}_{}", operation, max_duration_ms),
        name: format!("Duration threshold for {}", operation),
        description: format!(
            "Alert when operation '{}' exceeds {}ms duration",
            operation, max_duration_ms
        ),
        condition: AlertCondition::DurationThreshold {
            operation,
            max_duration_ns: (max_duration_ms * 1_000_000.0) as u64,
        },
        severity,
        channels,
        enabled: true,
        cooldown_duration: Duration::from_secs(300),
        max_alerts_per_hour: 10,
        tags: HashMap::new(),
    }
}

/// Create a memory-based alert rule
pub fn create_memory_alert_rule(
    max_memory_mb: u64,
    severity: AlertSeverity,
    channels: Vec<NotificationChannel>,
) -> AlertRule {
    AlertRule {
        id: format!("memory_{}", max_memory_mb),
        name: format!("Memory threshold {}MB", max_memory_mb),
        description: format!("Alert when memory usage exceeds {}MB", max_memory_mb),
        condition: AlertCondition::MemoryThreshold {
            max_memory_bytes: max_memory_mb * 1024 * 1024,
        },
        severity,
        channels,
        enabled: true,
        cooldown_duration: Duration::from_secs(300),
        max_alerts_per_hour: 10,
        tags: HashMap::new(),
    }
}