phil-core 0.4.0

MongoDB cluster initializer
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

#[cfg(test)]
mod test;

use std::{ffi::OsString, path::PathBuf};

use monger_core::Monger;
use mongodb::{
    options::{ClientOptions, Credential as DriverCredential, Tls, TlsOptions as DriverTlsOptions},
    sync::Client,
};
use typed_builder::TypedBuilder;

use crate::{
    error::Result,
    launch::{Launcher, Node},
};

#[derive(Debug, Clone)]
pub enum Topology {
    Single,
    ReplicaSet {
        set_name: String,
        db_paths: Vec<PathBuf>,
    },
    Sharded {
        num_mongos: u8,
        shard_db_paths: Vec<Vec<PathBuf>>,
        config_db_path: PathBuf,
    },
}

#[derive(Debug)]
pub struct Cluster {
    pub(crate) monger: Monger,
    pub(crate) client: Client,
    pub(crate) client_options: ClientOptions,
    pub(crate) topology: Topology,
    pub(crate) tls: Option<TlsOptions>,
    pub(crate) auth: Option<Credential>,
    pub(crate) nodes: Vec<Node>,
    pub(crate) cluster_id: String,
}

#[derive(Clone, Debug, TypedBuilder)]
pub struct ClusterOptions {
    pub topology: Topology,

    pub version_id: String,

    #[builder(default)]
    pub paths: Vec<PathBuf>,

    #[builder(default)]
    pub tls: Option<TlsOptions>,

    #[builder(default)]
    pub auth: Option<Credential>,

    #[builder(default)]
    extra_mongod_args: Vec<OsString>,

    #[builder(default)]
    verbose: bool,

    #[builder(default)]
    deprecated_tls_options: bool,

    #[builder(default)]
    save_logs: bool,
}

#[derive(Debug, Clone)]
pub struct TlsOptions {
    pub weak_tls: bool,
    pub allow_invalid_certificates: bool,
    pub ca_file_path: PathBuf,
    pub server_cert_file_path: PathBuf,
    pub client_cert_file_path: PathBuf,
}

impl From<TlsOptions> for Tls {
    fn from(opts: TlsOptions) -> Self {
        DriverTlsOptions::builder()
            .allow_invalid_certificates(opts.allow_invalid_certificates)
            .ca_file_path(opts.ca_file_path.to_string_lossy().into_owned())
            .cert_key_file_path(opts.server_cert_file_path.to_string_lossy().into_owned())
            .build()
            .into()
    }
}

#[derive(Debug, Clone)]
pub struct Credential {
    pub username: String,
    pub password: String,
    pub key_file: PathBuf,
}

impl From<Credential> for DriverCredential {
    fn from(credential: Credential) -> Self {
        Self::builder()
            .username(credential.username)
            .password(credential.password)
            .build()
    }
}

impl Cluster {
    pub fn new(options: ClusterOptions) -> Result<Self> {
        let launcher = Launcher::new(
            options.topology,
            options.version_id,
            options.tls,
            options.auth,
            options.verbose,
            options.deprecated_tls_options,
            options.save_logs,
            options.extra_mongod_args,
        )?;

        launcher.initialize_cluster()
    }

    pub fn client_options(&self) -> &ClientOptions {
        &self.client_options
    }
}