89 lines
2.8 KiB
Rust
89 lines
2.8 KiB
Rust
use argh::FromArgs;
|
|
use log::{error, info};
|
|
use pagerduty_rs::{eventsv2async::EventsV2 as PagerdutyAsyncEventsV2, types::Event};
|
|
use web3_proxy::{
|
|
config::TopConfig,
|
|
pagerduty::{pagerduty_event_for_config, trigger_pagerduty_alert},
|
|
};
|
|
|
|
#[derive(FromArgs, PartialEq, Debug, Eq)]
|
|
/// Quickly create a pagerduty alert
|
|
#[argh(subcommand, name = "pagerduty")]
|
|
pub struct PagerdutySubCommand {
|
|
#[argh(positional)]
|
|
/// short description of the alert
|
|
summary: String,
|
|
|
|
#[argh(option)]
|
|
/// the class/type of the event
|
|
class: Option<String>,
|
|
|
|
#[argh(option)]
|
|
/// the component of the event
|
|
component: Option<String>,
|
|
|
|
#[argh(option)]
|
|
/// deduplicate alerts based on this key.
|
|
/// If there are no open incidents with this key, a new incident will be created.
|
|
/// If there is an open incident with a matching key, the new event will be appended to that incident's Alerts log as an additional Trigger log entry.
|
|
dedup_key: Option<String>,
|
|
|
|
#[argh(option, default = "\"web3-proxy\".to_string()")]
|
|
/// a cluster or grouping of sources.
|
|
/// For example, sources "ethereum-proxy" and "polygon-proxy" might both be part of "web3-proxy".
|
|
group: String,
|
|
}
|
|
|
|
impl PagerdutySubCommand {
|
|
pub async fn main(
|
|
self,
|
|
pagerduty_async: Option<PagerdutyAsyncEventsV2>,
|
|
top_config: Option<TopConfig>,
|
|
) -> anyhow::Result<()> {
|
|
let event = top_config
|
|
.map(|top_config| {
|
|
pagerduty_event_for_config(
|
|
top_config,
|
|
self.class.clone(),
|
|
self.component.clone(),
|
|
Some(self.group.clone()),
|
|
self.summary.clone(),
|
|
None,
|
|
None::<()>,
|
|
)
|
|
})
|
|
.unwrap_or_else(|| {
|
|
trigger_pagerduty_alert(
|
|
"web3-proxy".to_string(),
|
|
None,
|
|
self.class,
|
|
None,
|
|
self.component,
|
|
Some(self.group),
|
|
None,
|
|
self.summary,
|
|
None,
|
|
None::<()>,
|
|
)
|
|
});
|
|
|
|
if let Some(pagerduty_async) = pagerduty_async {
|
|
info!(
|
|
"sending to pagerduty: {}",
|
|
serde_json::to_string_pretty(&event)?
|
|
);
|
|
|
|
if let Err(err) = pagerduty_async.event(Event::AlertTrigger(event)).await {
|
|
error!("Failed sending to pagerduty: {}", err);
|
|
}
|
|
} else {
|
|
info!(
|
|
"would send to pagerduty if PAGERDUTY_INTEGRATION_KEY were set: {}",
|
|
serde_json::to_string_pretty(&event)?
|
|
);
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
}
|