1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

use crate::core::persistence::*;
use crate::prelude::*;
use crate::supervisor;
use serde::Deserialize;
use std::sync::{Arc, Mutex};
use std::thread;
use std::time::Duration;

#[derive(Deserialize, Debug, Clone)]
pub struct Settings {
    /// Interval in milliseconds.
    #[serde(default = "default_interval_ms")]
    pub interval_ms: u64,
}

fn default_interval_ms() -> u64 {
    60000
}

pub fn spawn(service_id: &str, settings: &Settings, db: &Arc<Mutex<Connection>>, bus: &mut Bus) -> Result<()> {
    let interval = Duration::from_millis(settings.interval_ms);
    let tx = bus.add_tx();
    let sensor = format!("{}::size", service_id);
    let db = db.clone();

    supervisor::spawn(
        format!("my-iot::db::{}", service_id),
        tx.clone(),
        move || -> Result<()> {
            loop {
                let size = select_size(&db.lock().unwrap())?;
                tx.send(
                    Composer::new(sensor.clone())
                        .value(Value::DataSize(size))
                        .title("Database Size".to_string())
                        .into(),
                )?;
                thread::sleep(interval);
            }
        },
    )?;

    Ok(())
}