cronoq-store-sqlite 0.2.0

SQLite store for cronoq: durable state in a local file, with standby processes and fencing
Documentation 
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

//! A fresh database directory per test, removed afterwards.
#![allow(dead_code)]

use std::path::PathBuf;
use std::sync::atomic::{AtomicU32, Ordering};
use std::time::Duration;

use cronoq_store_sqlite::{SqliteStore, SqliteStoreConfig};

pub struct TestDb {
    pub dir: PathBuf,
}

impl TestDb {
    pub fn new() -> TestDb {
        static N: AtomicU32 = AtomicU32::new(0);
        let dir = std::env::temp_dir().join(format!(
            "cronoq-sqlite-{}-{}",
            std::process::id(),
            N.fetch_add(1, Ordering::SeqCst)
        ));
        let _ = std::fs::remove_dir_all(&dir);
        std::fs::create_dir_all(&dir).unwrap();
        TestDb { dir }
    }

    pub fn path(&self) -> PathBuf {
        self.dir.join("state.db")
    }

    pub fn cfg() -> SqliteStoreConfig {
        SqliteStoreConfig {
            lease_ttl: Duration::from_millis(1500),
            lease_renew: Duration::from_millis(200),
            lock_poll: Duration::from_millis(50),
            ..Default::default()
        }
    }

    pub async fn open(&self) -> SqliteStore {
        SqliteStore::open_with(self.path(), Self::cfg())
            .await
            .unwrap()
    }
}

impl Drop for TestDb {
    fn drop(&mut self) {
        let _ = std::fs::remove_dir_all(&self.dir);
    }
}