set check frequency via config

Author: lamafab

src/lib.rs

@@ -64,6 +64,7 @@ struct Config {
 struct EscalationConfig {
     enabled: bool,
     escalation_window: u64,
+    check_frequency: u64,
 }
 
 #[derive(StructOpt, Debug)]
@@ -110,6 +111,12 @@ pub async fn run() -> Result<()> {
         .unwrap_or(MIN_ESCALATION_WINDOW)
         .max(MIN_ESCALATION_WINDOW);
 
+    let check_frequency = config
+        .escalation
+        .as_ref()
+        .map(|c| c.check_frequency)
+        .unwrap_or(20);
+
     if should_escalate && config.database.is_none() {
         return Err(anyhow!(
             "Escalations require a database configuration, which isn't provided"
@@ -133,7 +140,13 @@ pub async fn run() -> Result<()> {
     let (tx, mut recv) = unbounded_channel();
 
     info!("Adding message processor to system registry");
-    let proc = processor::Processor::new(opt_db, escalation_window, should_escalate, tx.clone());
+    let proc = processor::Processor::new(
+        opt_db,
+        escalation_window,
+        should_escalate,
+        check_frequency,
+        tx.clone(),
+    );
     SystemRegistry::set(proc.start());
 
     info!("Initializing Matrix client");

src/processor.rs

@@ -8,8 +8,6 @@ use std::time::Duration;
 use tokio::sync::mpsc::UnboundedSender;
 use tokio::sync::Mutex;
 
-const CRON_JOB_INTERVAL: u64 = 5;
-
 #[derive(Debug, Clone, Eq, PartialEq, Serialize, Deserialize)]
 pub struct AlertContext {
     pub id: AlertId,
@@ -91,6 +89,7 @@ pub struct Processor {
     should_escalate: bool,
     // Ensures that only one escalation task is running at the time.
     escalation_lock: Arc<Mutex<()>>,
+    check_frequency: u64,
     shutdown_indicator: UnboundedSender<()>,
 }
 
@@ -99,13 +98,15 @@ impl Processor {
         db: Option<Database>,
         escalation_window: u64,
         should_escalate: bool,
+        check_frequency: u64,
         shutdown_indicator: UnboundedSender<()>,
     ) -> Self {
         Processor {
             db: db.map(Arc::new),
             escalation_window,
             should_escalate,
             escalation_lock: Default::default(),
+            check_frequency,
             shutdown_indicator,
         }
     }
@@ -159,7 +160,7 @@ impl Actor for Processor {
             let shutdown_indicator = self.shutdown_indicator.clone();
 
             ctx.run_interval(
-                Duration::from_secs(CRON_JOB_INTERVAL),
+                Duration::from_secs(self.check_frequency),
                 move |_proc, _ctx| {
                     // Acquire new handles for async task.
                     let db = Arc::clone(&db);