oxigdal-observability 0.1.6

OpenTelemetry-based observability, monitoring, and alerting for OxiGDAL
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

//! Alert deduplication.

use super::Alert;
use crate::error::Result;
use parking_lot::RwLock;
use std::collections::HashMap;
use std::sync::Arc;

/// Alert deduplicator.
pub struct AlertDeduplicator {
    seen_alerts: Arc<RwLock<HashMap<String, Alert>>>,
}

impl AlertDeduplicator {
    /// Create a new alert deduplicator.
    pub fn new() -> Self {
        Self {
            seen_alerts: Arc::new(RwLock::new(HashMap::new())),
        }
    }

    /// Deduplicate alerts based on name and labels.
    pub fn deduplicate(&self, alerts: &[Alert]) -> Result<Vec<Alert>> {
        let mut seen = self.seen_alerts.write();
        let mut deduplicated = Vec::new();

        for alert in alerts {
            let key = format!("{}:{:?}", alert.name, alert.labels);

            if let std::collections::hash_map::Entry::Vacant(e) = seen.entry(key) {
                e.insert(alert.clone());
                deduplicated.push(alert.clone());
            }
        }

        Ok(deduplicated)
    }

    /// Clear deduplication cache.
    pub fn clear(&self) {
        self.seen_alerts.write().clear();
    }
}

impl Default for AlertDeduplicator {
    fn default() -> Self {
        Self::new()
    }
}