Skip to content

Commit

Permalink
fix(sherlock): Moved poll_anomaly_scores sleep to async
Browse files Browse the repository at this point in the history
  • Loading branch information
isala404 committed Mar 16, 2022
1 parent 247ee28 commit 5b49dbf
Show file tree
Hide file tree
Showing 3 changed files with 30 additions and 15 deletions.
24 changes: 12 additions & 12 deletions control-plane/config/samples/lazykoala_v1alpha1_inspector.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,9 @@ metadata:
name: service-1
namespace: default
spec:
deploymentRef: service-1-62bf5632
serviceRef: service-1-62bf5632
modelName: service-1-62bf5632
deploymentRef: service-1-a0b7dd98
serviceRef: service-1-a0b7dd98
modelName: sample_model
namespace: default
---
apiVersion: lazykoala.isala.me/v1alpha1
Expand All @@ -15,9 +15,9 @@ metadata:
name: service-2
namespace: default
spec:
deploymentRef: service-2-62bf5632
serviceRef: service-2-62bf5632
modelName: service-1-62bf5632
deploymentRef: service-2-a0b7dd98
serviceRef: service-2-a0b7dd98
modelName: sample_model
namespace: default
---
apiVersion: lazykoala.isala.me/v1alpha1
Expand All @@ -26,9 +26,9 @@ metadata:
name: service-3
namespace: default
spec:
deploymentRef: service-3-62bf5632
serviceRef: service-3-62bf5632
modelName: service-1-62bf5632
deploymentRef: service-3-a0b7dd98
serviceRef: service-3-a0b7dd98
modelName: sample_model
namespace: default
---
apiVersion: lazykoala.isala.me/v1alpha1
Expand All @@ -37,9 +37,9 @@ metadata:
name: service-4
namespace: default
spec:
deploymentRef: service-4-62bf5632
serviceRef: service-4-62bf5632
modelName: service-1-62bf5632
deploymentRef: service-4-a0b7dd98
serviceRef: service-4-a0b7dd98
modelName: sample_model
namespace: default
---

20 changes: 18 additions & 2 deletions sherlock/src/inference.rs
Original file line number Diff line number Diff line change
@@ -1,4 +1,6 @@
use std::{thread, time::Duration, collections::HashMap, env::var};
use std::{collections::HashMap, env::var};
// use futures::future::join_all;
use tokio::time::{sleep, Duration};
use serde::{Serialize, Deserialize};
use serde_json::json;
use prometheus::GaugeVec;
Expand Down Expand Up @@ -77,12 +79,26 @@ pub async fn poll_anomaly_scores() {
loop {
let services = parse_config().unwrap_or_default();

// let mut tasks = Vec::new();

for (service, args) in services.iter() {
// tasks.push(calculate_anomaly_score(service, args));

if let Err(err) = calculate_anomaly_score(service, args).await {
eprintln!("Error while calculating anomaly score: {}", err);
ANOMLAY_GAUGE.with_label_values(&[service, &args.namespace]).set(-1.0)
}
}
thread::sleep(Duration::from_secs(delay));

// let results = join_all(tasks).await;

// for result in results.iter(){
// if let Err(err) = result {
// eprintln!("Error while calculating anomaly score: {}", err);
// // ANOMLAY_GAUGE.with_label_values(&[service, &args.namespace]).set(-1.0)
// }
// }

sleep(Duration::from_secs(delay)).await;
}
}
1 change: 0 additions & 1 deletion sherlock/src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,6 @@ async fn main() {

tokio::spawn(async { poll_anomaly_scores().await });


let addr = ([0, 0, 0, 0], 9898).into();
println!("Listening on http://{}", addr);

Expand Down

0 comments on commit 5b49dbf

Please sign in to comment.