klaxon 0.2.1

Alerting and monitoring framework
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

use super::issue::{PdIssue, PdIssueFields};
use anyhow::Result;
use futures::future::try_join_all;
use log::*;

pub struct PdReporter {
    pagerduty_api_key: Option<String>,
    to_trigger: Vec<PdIssue>,
    to_resolve: Vec<PdIssue>,
}

impl PdReporter {
    pub fn new(pagerduty_api_key: Option<String>) -> Result<Self> {
        Ok(Self {
            pagerduty_api_key,
            to_trigger: Default::default(),
            to_resolve: Default::default(),
        })
    }

    pub fn trigger(&mut self, issue: PdIssue) {
        self.to_trigger.push(issue);
    }

    pub fn resolve(&mut self, issue: PdIssue) {
        self.to_resolve.push(issue);
    }

    pub async fn finish(self) -> Result<()> {
        let pd_client = match self.pagerduty_api_key {
            None => {
                debug!("🌵 DRY RUN, nothing will be sent to PagerDuty");
                None
            }
            Some(key) => Some(pagerduty_rs::eventsv2async::EventsV2::new(key, None)?),
        };
        let mut tasks = Vec::new();

        for issue in self.to_trigger.into_iter() {
            warn!("🚨 Triggering issue {}: {:?}", issue.dedup_key(), issue);
            let event = pagerduty_rs::types::Event::AlertTrigger(issue.into());
            if let Some(pd_client) = pd_client.as_ref() {
                tasks.push(pd_client.event(event));
            }
        }

        for issue in self.to_resolve.into_iter() {
            info!("✅ Resolving issue {}: {:?}", issue.dedup_key(), issue);
            let event = pagerduty_rs::types::Event::<PdIssueFields>::AlertResolve(issue.into());
            if let Some(pd_client) = pd_client.as_ref() {
                tasks.push(pd_client.event(event));
            }
        }

        debug!("Waiting for {} Pagerduty update tasks", tasks.len());
        try_join_all(tasks).await?;
        debug!("Finished sending updates to PagerDuty");

        Ok(())
    }
}