Hastic standalone
https://hastic.io
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
89 lines
2.4 KiB
89 lines
2.4 KiB
use subbeat::types::{DatasourceConfig, InfluxConfig, PrometheusConfig}; |
|
|
|
|
|
#[derive(Clone)] |
|
pub struct WebhookAlertingConfig { |
|
endpoint: String |
|
} |
|
|
|
#[derive(Clone)] |
|
pub enum AlertingType { |
|
Webhook(WebhookAlertingConfig) |
|
} |
|
|
|
#[derive(Clone)] |
|
pub struct AlertingConfig { |
|
alerting_type: AlertingType, |
|
interval: u64 // interval in seconds |
|
} |
|
|
|
|
|
#[derive(Clone)] |
|
pub struct Config { |
|
pub port: u16, |
|
pub datasource_config: DatasourceConfig, |
|
pub alerting: Option<AlertingConfig>, |
|
} |
|
|
|
// impl Clone for Config { |
|
// fn clone(&self) -> Self { |
|
// return Config { |
|
// port: self.port, |
|
// datasource_config: self.datasource_config.clone(), |
|
// alerting: self.alerting.clone() |
|
// }; |
|
// } |
|
// } |
|
|
|
fn resolve_datasource(config: &mut config::Config) -> anyhow::Result<DatasourceConfig> { |
|
if config.get::<String>("prometheus.url").is_ok() { |
|
return Ok(DatasourceConfig::Prometheus(PrometheusConfig { |
|
url: config.get("prometheus.url")?, |
|
query: config.get("prometheus.query")?, |
|
})); |
|
} |
|
|
|
if config.get::<String>("influx.url").is_ok() { |
|
return Ok(DatasourceConfig::Influx(InfluxConfig { |
|
url: config.get("influx.url")?, |
|
org_id: config.get("influx.org_id")?, |
|
token: config.get("influx.token")?, |
|
query: config.get("influx.query")?, |
|
})); |
|
} |
|
|
|
return Err(anyhow::format_err!("no datasource found")); |
|
} |
|
|
|
// TODO: use actual config and env variables |
|
impl Config { |
|
pub fn new() -> anyhow::Result<Config> { |
|
|
|
// TODO: parse alerting config |
|
// TODO: throw error on bad config |
|
|
|
let mut config = config::Config::default(); |
|
|
|
if std::path::Path::new("config.toml").exists() { |
|
config.merge(config::File::with_name("config")).unwrap(); |
|
} |
|
config |
|
.merge(config::Environment::with_prefix("HASTIC")) |
|
.unwrap(); |
|
|
|
if config.get::<u16>("port").is_err() { |
|
config.set("port", "8000").unwrap(); |
|
} |
|
|
|
// let mut endpoint = None; |
|
// if config.get::<String>("webhook.endpoint").is_ok() { |
|
// endpoint = Some(config.get("webhook.endpoint").unwrap()); |
|
// } |
|
|
|
Ok(Config { |
|
port: config.get::<u16>("port").unwrap(), |
|
datasource_config: resolve_datasource(&mut config)?, |
|
alerting: None, |
|
}) |
|
} |
|
}
|
|
|