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
use log::*;

pub fn str_to_logfilter_level(level: &str) -> LevelFilter {
    match level {
        "trace" => LevelFilter::Trace,
        "debug" => LevelFilter::Debug,
        "info" => LevelFilter::Info,
        "warn" => LevelFilter::Warn,
        "error" => LevelFilter::Error,
        "off" => LevelFilter::Off,
        _ => LevelFilter::Off,
    }
}

pub fn str_to_log_level(level: &str) -> Level {
    match level {
        "trace" => Level::Trace,
        "debug" => Level::Debug,
        "info" => Level::Info,
        "warn" => Level::Warn,
        "error" => Level::Error,
        _ => Level::Error,
    }
}

pub struct CombineLogger {
    loggers: Vec<Box<dyn Log>>
}

impl CombineLogger {
    pub fn new() -> CombineLogger
    {
        CombineLogger{loggers: vec![]}
    }

    pub fn append(mut self, logger: Box<dyn Log>) -> Self {
        self.loggers.push(logger);
        self
    }
    
    pub fn start(self) -> bool {
        // 处理多个模块都嵌有Logger的情况,如果已经设置过logger,这个函数不起作用,返回false,以第一次创建的为准
        // 成功注册,则返回true
        if let Err(e) = log::set_boxed_logger(Box::new(self)) {
            warn!("logger already initialized, use prev one. {}", e);
            false
        } else {
            true
        }
    }
}

impl Log for CombineLogger {
    fn enabled(&self, metadata: &Metadata) -> bool {
        let mut ret = true;
        for logger in &self.loggers {
            ret = ret | logger.enabled(metadata);
        }
        ret
    }

    fn log(&self, record: &Record) {
        for logger in &self.loggers {
            logger.log(record);
        }
    }

    fn flush(&self) {
        for logger in &self.loggers {
            logger.flush();
        }
    }
}