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
45
46
47
48
49
50
51
52
53
54
55
56

use crate::reading::Reading;
use crate::services::Service;
use crate::threading;
use crate::value::Value;
use crate::Result;
use chrono::Local;
use crossbeam_channel::{Receiver, Sender};
use serde::Deserialize;
use std::sync::{Arc, Mutex};
use std::thread;
use std::time::Duration;

pub struct Db {
    service_id: String,
    interval: Duration,
}

#[derive(Deserialize, Debug, Clone)]
pub struct Settings {
    /// Interval in milliseconds.
    pub interval_ms: Option<u64>,
}

impl Db {
    pub fn new(service_id: &str, settings: &Settings) -> Db {
        Db {
            service_id: service_id.into(),
            interval: Duration::from_millis(settings.interval_ms.unwrap_or(1000)),
        }
    }
}

impl Service for Db {
    fn spawn(
        self: Box<Self>,
        db: Arc<Mutex<crate::db::Db>>,
        tx: Sender<Reading>,
        _rx: Receiver<Reading>,
    ) -> Result<()> {
        let sensor = format!("{}:size", &self.service_id);
        threading::spawn(self.service_id.clone(), move || loop {
            let size = { db.lock().unwrap().select_size().unwrap() };

            tx.send(Reading {
                sensor: sensor.clone(),
                value: Value::Size(size),
                timestamp: Local::now(),
                is_persisted: true,
            })
            .unwrap();

            thread::sleep(self.interval);
        })?;
        Ok(())
    }
}