soph_database/config/
mod.rs

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
use serde::Deserialize;

#[derive(Debug, Default, Deserialize)]
pub struct Database {
    pub url: String,

    #[serde(default = "default_migration_path")]
    pub migration_path: String,

    #[serde(default = "default_model_path")]
    pub model_path: String,

    #[serde(default = "default_max_connections")]
    pub max_connections: u32,

    #[serde(default = "default_min_connections")]
    pub min_connections: u32,

    #[serde(default = "default_connect_timeout")]
    pub connect_timeout: u64,

    #[serde(default = "default_idle_timeout")]
    pub idle_timeout: u64,

    #[serde(default = "default_acquire_timeout")]
    pub acquire_timeout: u64,

    #[serde(default = "default_max_lifetime")]
    pub max_lifetime: u64,

    #[serde(default)]
    pub sqlx_logging: bool,

    #[serde(default = "default_sqlx_logging_level")]
    pub sqlx_logging_level: LevelFilter,

    #[serde(default)]
    pub sqlx_slow_statements_logging_level: LevelFilter,

    #[serde(default = "default_sqlx_slow_statements_logging_threshold")]
    pub sqlx_slow_statements_logging_threshold: u64,

    #[serde(default = "default_test_before_acquire")]
    pub test_before_acquire: bool,
}

#[derive(Clone, Debug, Default, Deserialize)]
pub enum LevelFilter {
    #[serde(rename = "off")]
    #[default]
    Off,
    #[serde(rename = "error")]
    Error,
    #[serde(rename = "warn")]
    Warn,
    #[serde(rename = "info")]
    Info,
    #[serde(rename = "debug")]
    Debug,
    #[serde(rename = "trace")]
    Trace,
}

impl From<LevelFilter> for tracing::log::LevelFilter {
    fn from(level: LevelFilter) -> Self {
        match level {
            LevelFilter::Off => tracing::log::LevelFilter::Off,
            LevelFilter::Error => tracing::log::LevelFilter::Error,
            LevelFilter::Warn => tracing::log::LevelFilter::Warn,
            LevelFilter::Info => tracing::log::LevelFilter::Info,
            LevelFilter::Debug => tracing::log::LevelFilter::Debug,
            LevelFilter::Trace => tracing::log::LevelFilter::Trace,
        }
    }
}

fn default_migration_path() -> String {
    "database/migrations".to_string()
}

fn default_model_path() -> String {
    "app/src/models".to_string()
}

fn default_max_connections() -> u32 {
    10
}

fn default_min_connections() -> u32 {
    1
}

fn default_connect_timeout() -> u64 {
    3 * 1000
}

fn default_idle_timeout() -> u64 {
    8 * 1000
}

fn default_acquire_timeout() -> u64 {
    3 * 1000
}

fn default_max_lifetime() -> u64 {
    8 * 1000
}

fn default_sqlx_logging_level() -> LevelFilter {
    LevelFilter::Info
}

fn default_sqlx_slow_statements_logging_threshold() -> u64 {
    1000
}

fn default_test_before_acquire() -> bool {
    true
}