Alexey Velikiy
3 years ago
4 changed files with 19634 additions and 8 deletions
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,49 @@
|
||||
use hastic::config::Config; |
||||
use hastic::services::data_service::Segment; |
||||
use hastic::services::metric_service::MetricService; |
||||
use anyhow; |
||||
|
||||
use subbeat::metric::{Metric, MetricResult}; |
||||
|
||||
|
||||
struct AnalyticUnit { |
||||
metric_service: MetricService |
||||
} |
||||
|
||||
impl AnalyticUnit { |
||||
fn new(config: &Config) -> AnalyticUnit { |
||||
AnalyticUnit{ |
||||
metric_service: MetricService::new( |
||||
&config.prom_url, |
||||
&config.query |
||||
) |
||||
} |
||||
} |
||||
|
||||
pub async fn get_detections(&self, from: u64, to: u64) -> anyhow::Result<Vec<Segment>> { |
||||
let prom = self.metric_service.get_prom(); |
||||
let mr = prom.query(from, to, 10).await?; |
||||
|
||||
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.0 { |
||||
if from.is_some() { |
||||
continue; |
||||
} else { |
||||
from = Some(*t); |
||||
} |
||||
} else { |
||||
if from.is_some() { |
||||
result.push(Segment{ id: None, from: from.unwrap(), to: *t }); |
||||
from = None; |
||||
} |
||||
} |
||||
} |
||||
|
||||
Ok(result) |
||||
} |
||||
} |
Loading…
Reference in new issue