atomr-persistence-mongodb 0.3.1

MongoDB journal and snapshot store for atomr — indexed collections, atomic multi-document inserts, BSON payloads.
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
52
53
54
55
56
57
58

//! Connection configuration for the MongoDB provider.

use std::env;

#[derive(Debug, Clone)]
pub struct MongoConfig {
    pub url: String,
    pub database: String,
    pub journal_collection: String,
    pub snapshot_collection: String,
}

impl MongoConfig {
    pub fn new(url: impl Into<String>, database: impl Into<String>) -> Self {
        Self {
            url: url.into(),
            database: database.into(),
            journal_collection: "event_journal".into(),
            snapshot_collection: "snapshot_store".into(),
        }
    }

    pub fn with_collections(mut self, journal: impl Into<String>, snapshot: impl Into<String>) -> Self {
        self.journal_collection = journal.into();
        self.snapshot_collection = snapshot.into();
        self
    }

    /// Env lookup: `ATOMR_PERSISTENCE_MONGO_URL`, `ATOMR_IT_MONGO_URL`,
    /// `MONGODB_URL`, dev fallback `mongodb://127.0.0.1:27017`.
    pub fn from_env() -> Self {
        let url = env::var("ATOMR_PERSISTENCE_MONGO_URL")
            .or_else(|_| env::var("ATOMR_IT_MONGO_URL"))
            .or_else(|_| env::var("MONGODB_URL"))
            .unwrap_or_else(|_| "mongodb://127.0.0.1:27017".to_string());
        let db = env::var("ATOMR_PERSISTENCE_MONGO_DB").unwrap_or_else(|_| "atomr".into());
        Self::new(url, db)
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn defaults() {
        let cfg = MongoConfig::new("mongodb://x", "atomr");
        assert_eq!(cfg.journal_collection, "event_journal");
        assert_eq!(cfg.snapshot_collection, "snapshot_store");
    }

    #[test]
    fn custom_collections() {
        let cfg = MongoConfig::new("mongodb://x", "db").with_collections("j", "s");
        assert_eq!(cfg.journal_collection, "j");
        assert_eq!(cfg.snapshot_collection, "s");
    }
}