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
pub mod config;
use std::str::FromStr;
use thiserror::Error;
#[derive(Error, Debug)]
pub enum LoggerError {
#[error("LoggerConfigurationError: [{message}]")]
LoggerConfigurationError { message: String },
}
impl From<log::SetLoggerError> for LoggerError {
fn from(error: log::SetLoggerError) -> Self {
LoggerError::LoggerConfigurationError {
message: format!("{}", error),
}
}
}
impl From<std::io::Error> for LoggerError {
fn from(error: std::io::Error) -> Self {
LoggerError::LoggerConfigurationError {
message: format!("{}", error),
}
}
}
pub fn setup_logger(logger_config: &config::LoggerConfig) -> Result<(), LoggerError> {
let mut log_dispatcher = fern::Dispatch::new()
.format(|out, message, record| {
out.finish(format_args!(
"{}[{}][{}] {}",
chrono::Local::now().format("[%Y-%m-%d][%H:%M:%S]"),
record.target(),
record.level(),
message
))
})
.level(
log::LevelFilter::from_str(&logger_config.level).map_err(|err| {
LoggerError::LoggerConfigurationError {
message: format!(
"The specified logger level is not valid: [{}]. err: {}",
&logger_config.level, err
),
}
})?,
);
log_dispatcher = log_dispatcher
.level_for("lettre".to_owned(), log::LevelFilter::Warn)
.level_for("postgres".to_owned(), log::LevelFilter::Warn)
.level_for("hyper".to_owned(), log::LevelFilter::Warn)
.level_for("mio".to_owned(), log::LevelFilter::Warn)
.level_for("tokio_io".to_owned(), log::LevelFilter::Warn)
.level_for("tokio_reactor".to_owned(), log::LevelFilter::Warn)
.level_for("tokio_tcp".to_owned(), log::LevelFilter::Warn)
.level_for("tokio_uds".to_owned(), log::LevelFilter::Warn);
if logger_config.stdout_output {
log_dispatcher = log_dispatcher.chain(std::io::stdout());
}
if logger_config.stderr_output {
log_dispatcher = log_dispatcher.chain(std::io::stderr());
}
if let Some(path) = &logger_config.file_output_path {
log_dispatcher = log_dispatcher.chain(fern::log_file(&path)?)
}
log_dispatcher.apply()?;
Ok(())
}