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

use crate::logger::{DataDogConfig, DataDogLogLevel, DataDogLogger, DataDogLoggerError};
use log::{LevelFilter, Log, Metadata, Record};

/// Requires `log` feature enabled
impl Log for DataDogLogger {
    fn enabled(&self, _metadata: &Metadata) -> bool {
        true
    }
    fn log(&self, record: &Record) {
        let level = match record.level() {
            log::Level::Error => DataDogLogLevel::Error,
            log::Level::Warn => DataDogLogLevel::Warning,
            log::Level::Info => DataDogLogLevel::Informational,
            log::Level::Debug | log::Level::Trace => DataDogLogLevel::Debug,
        };

        &self.log(format!("{}", record.args()), level);
    }
    fn flush(&self) {}
}

impl DataDogLogger {
    /// Initializes DataDogLogger with `log` crate
    ///
    /// Requires `log` feature enabled
    pub fn init_with_log(
        config: DataDogConfig,
        level: LevelFilter,
    ) -> Result<(), DataDogLoggerError> {
        let logger = DataDogLogger::new(config)?;
        log::set_boxed_logger(Box::new(logger))?;
        log::set_max_level(level);
        Ok(())
    }
}