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
mod config;
pub use config::{
LoggerConfig, LoggerConfigBuilder, LoggerOutputConfig, LoggerOutputConfigBuilder,
};
use fern::{
colors::{Color, ColoredLevelConfig},
Dispatch,
};
use thiserror::Error;
pub const LOGGER_STDOUT_NAME: &str = "stdout";
#[derive(Error, Debug)]
#[non_exhaustive]
pub enum Error {
#[error("Creating output file failed.")]
CreatingFileFailed,
#[error("Initializing the logger backend failed.")]
InitializationFailed,
}
macro_rules! log_format {
($target:expr, $level:expr, $message:expr, $target_width:expr, $level_width:expr) => {
format_args!(
"{} {:target_width$} {:level_width$} {}",
time_helper::format(&time_helper::now_utc()),
$target,
$level,
$message,
target_width = $target_width,
level_width = $level_width
)
};
}
pub fn logger_init(config: LoggerConfig) -> Result<(), Error> {
let target_width = config.target_width;
let level_width = config.level_width;
let mut logger = Dispatch::new();
for output in config.outputs {
let mut dispatch = if output.color_enabled {
let colors = ColoredLevelConfig::new()
.trace(Color::BrightMagenta)
.debug(Color::BrightBlue)
.info(Color::BrightGreen)
.warn(Color::BrightYellow)
.error(Color::BrightRed);
Dispatch::new().format(move |out, message, record| {
out.finish(log_format!(
record.target(),
colors.color(record.level()),
message,
target_width,
level_width
))
})
} else {
Dispatch::new().format(move |out, message, record| {
out.finish(log_format!(
record.target(),
record.level(),
message,
target_width,
level_width
))
})
}
.level(output.level_filter);
if !output.target_filters.is_empty() {
let target_filters = output.target_filters;
dispatch = dispatch.filter(move |metadata| {
let target = metadata.target().to_lowercase();
target_filters.iter().any(|f| target.contains(f))
});
}
if !output.target_exclusions.is_empty() {
let target_exclusions = output.target_exclusions;
dispatch = dispatch.filter(move |metadata| {
let target = metadata.target().to_lowercase();
!target_exclusions.iter().any(|f| target.contains(f))
});
}
dispatch = if output.name == LOGGER_STDOUT_NAME {
dispatch.chain(std::io::stdout())
} else {
dispatch.chain(fern::log_file(output.name).map_err(|_| Error::CreatingFileFailed)?)
};
logger = logger.chain(dispatch);
}
logger.apply().map_err(|_| Error::InitializationFailed)?;
Ok(())
}