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.
77 lines
2.0 KiB
77 lines
2.0 KiB
use crate::{config::Config, utils::get_random_str}; |
|
|
|
use super::{ |
|
metric_service::MetricService, |
|
segments_service::{Segment, SegmentType, ID_LENGTH}, |
|
}; |
|
|
|
use subbeat::metric::Metric; |
|
|
|
use anyhow; |
|
|
|
#[derive(Clone)] |
|
pub struct AnalyticService { |
|
metric_service: MetricService, |
|
} |
|
|
|
impl AnalyticService { |
|
pub fn new(config: &Config) -> AnalyticService { |
|
AnalyticService { |
|
metric_service: MetricService::new(&config.prom_url, &config.query), |
|
} |
|
} |
|
|
|
pub async fn get_detections( |
|
&self, |
|
from: u64, |
|
to: u64, |
|
step: u64, |
|
) -> anyhow::Result<Vec<Segment>> { |
|
let prom = self.metric_service.get_prom(); |
|
let mr = prom.query(from, to, step).await?; |
|
|
|
if mr.data.keys().len() == 0 { |
|
return Ok(Vec::new()); |
|
} |
|
|
|
let key = mr.data.keys().nth(0).unwrap(); |
|
let ts = &mr.data[key]; |
|
|
|
let mut result = Vec::<Segment>::new(); |
|
let mut from: Option<u64> = None; |
|
for (t, v) in ts { |
|
if *v > 100_000.0 { |
|
if from.is_some() { |
|
continue; |
|
} else { |
|
from = Some(*t); |
|
} |
|
} else { |
|
if from.is_some() { |
|
result.push(Segment { |
|
// TODO: persist detections together with id |
|
id: Some(get_random_str(ID_LENGTH)), |
|
from: from.unwrap(), |
|
to: *t, |
|
segment_type: SegmentType::Detection, |
|
}); |
|
from = None; |
|
} |
|
} |
|
} |
|
|
|
// TODO: don't repeat myself |
|
if from.is_some() { |
|
result.push(Segment { |
|
id: Some(get_random_str(ID_LENGTH)), |
|
from: from.unwrap(), |
|
to, |
|
segment_type: SegmentType::Detection, |
|
}); |
|
} |
|
|
|
// TODO: decide what to do it from is Some() in the end |
|
|
|
Ok(result) |
|
} |
|
}
|
|
|