tin-logger 0.1.0

A logging library based on flexi_logger with file rotation, console output, and custom formatting support
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

use std::path::PathBuf;

use log::LevelFilter;

// 自定义类型
pub enum Duplicate {
    /// No messages are duplicated.
    None = 0,
    /// Only error messages are duplicated.
    Error = 1,
    /// Error and warn messages are duplicated.
    Warn = 2,
    /// Error, warn, and info messages are duplicated.
    Info = 3,
    /// Error, warn, info, and debug messages are duplicated.
    Debug = 4,
    /// All messages are duplicated.
    Trace = 5,
    /// All messages are duplicated.
    All = 6,
}

impl From<Duplicate> for flexi_logger::Duplicate {
    fn from(val: Duplicate) -> Self {
        match val {
            Duplicate::None => flexi_logger::Duplicate::None,
            Duplicate::Error => flexi_logger::Duplicate::Error,
            Duplicate::Warn => flexi_logger::Duplicate::Warn,
            Duplicate::Info => flexi_logger::Duplicate::Info,
            Duplicate::Debug => flexi_logger::Duplicate::Debug,
            Duplicate::Trace => flexi_logger::Duplicate::Trace,
            Duplicate::All => flexi_logger::Duplicate::All,
        }
    }
}

impl From<u8> for Duplicate {
    fn from(val: u8) -> Self {
        match val {
            0 => Duplicate::None,
            1 => Duplicate::Error,
            2 => Duplicate::Warn,
            3 => Duplicate::Info,
            4 => Duplicate::Debug,
            5 => Duplicate::Trace,
            6 => Duplicate::All,
            _ => unreachable!(),
        }
    }
}

pub struct LoggerConfig {
    pub level: LevelFilter,
    pub directory: PathBuf,
    pub basename: String,
    pub suffix: String,
    pub rotate_size: u64,
    pub duplicate_to_stdout: Duplicate,
}

impl Default for LoggerConfig {
    fn default() -> Self {
        Self {
            level: LevelFilter::Info,
            directory: PathBuf::from("logs"),
            basename: "app".to_string(),
            suffix: "log".to_string(),
            rotate_size: 100 * 1024 * 1024,
            duplicate_to_stdout: Duplicate::None,
        }
    }
}

impl LoggerConfig {
    pub fn builder() -> LoggerConfigBuilder {
        LoggerConfigBuilder::default()
    }
}

#[derive(Default)]
pub struct LoggerConfigBuilder {
    level: Option<LevelFilter>,
    directory: Option<PathBuf>,
    basename: Option<String>,
    suffix: Option<String>,
    rotate_size: Option<u64>,
    duplicate_to_stdout: Option<Duplicate>,
}

impl LoggerConfigBuilder {
    pub fn level(mut self, level: LevelFilter) -> Self {
        self.level = Some(level);
        self
    }
    pub fn directory<P: Into<PathBuf>>(mut self, directory: P) -> Self {
        self.directory = Some(directory.into());
        self
    }
    pub fn basename<S: Into<String>>(mut self, basename: S) -> Self {
        self.basename = Some(basename.into());
        self
    }
    pub fn suffix<S: Into<String>>(mut self, suffix: S) -> Self {
        self.suffix = Some(suffix.into());
        self
    }
    pub fn rotate_size(mut self, rotate_size: u64) -> Self {
        self.rotate_size = Some(rotate_size);
        self
    }
    pub fn duplicate_to_stdout(mut self, duplicate: Duplicate) -> Self {
        self.duplicate_to_stdout = Some(duplicate);
        self
    }
    pub fn build(self) -> LoggerConfig {
        LoggerConfig {
            level: self.level.unwrap_or(LevelFilter::Info),
            directory: self.directory.unwrap_or_else(|| PathBuf::from("logs")),
            basename: self.basename.unwrap_or_else(|| "app".to_string()),
            suffix: self.suffix.unwrap_or_else(|| "log".to_string()),
            rotate_size: self.rotate_size.unwrap_or(100 * 1024 * 1024),
            duplicate_to_stdout: self.duplicate_to_stdout.unwrap_or(Duplicate::All),
        }
    }
}