rmqttd 0.17.0

MQTT Server for v3.1, v3.1.1 and v5.0 protocols
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
126
127

use std::fs::{File, OpenOptions};
use std::io::{self, Write};

use anyhow::anyhow;
use slog::{o, Drain, Record};
use slog_scope::GlobalLoggerGuard;
use slog_term::{CountingWriter, RecordDecorator, ThreadSafeTimestampFn};

use rmqtt_conf::logging::{Level, Log, To};
use rmqtt_net::Result;

/// Initializes a logger using `slog` and `slog_scope`.
///
/// This function creates a `GlobalLoggerGuard` and sets the global logger to the `logger` passed
/// in the `Runtime` instance. It also initializes `slog_stdlog` with the log level specified in
/// the `Runtime` settings.
pub fn logger_init(cfg: &Log) -> Result<(GlobalLoggerGuard, slog::Logger)> {
    let logger = config_logger(cfg.filename(), cfg.to, cfg.level)?;
    let level = slog_log_to_level(cfg.level.inner());
    // Make sure to save the guard, see documentation for more information
    let guard = slog_scope::set_global_logger(logger.clone());
    // register slog_stdlog as the log handler with the log crate
    slog_stdlog::init_with_level(level)?;
    Ok((guard, logger))
}

fn slog_log_to_level(level: slog::Level) -> log::Level {
    match level {
        slog::Level::Trace => log::Level::Trace,
        slog::Level::Debug => log::Level::Debug,
        slog::Level::Info => log::Level::Info,
        slog::Level::Warning => log::Level::Warn,
        slog::Level::Error => log::Level::Error,
        slog::Level::Critical => log::Level::Error,
    }
}

/// Creates a new `slog::Logger` with two `Drain`s: one for printing to the console and another for
/// printing to a file.
///
/// This function takes three arguments: `filename`, which specifies the name of the file to print
/// to; `to`, which specifies where to print the logs (either the console or a file); and `level`,
/// which specifies the minimum log level to print. The function sets the format for the logs and
/// creates the two `Drain`s using the provided parameters. It then combines the two `Drain`s using a
/// `Tee` and returns the resulting `Logger`.
pub fn config_logger(filename: String, to: To, level: Level) -> Result<slog::Logger> {
    let custom_timestamp =
        |io: &mut dyn io::Write| write!(io, "{}", chrono::Local::now().format("%Y-%m-%d %H:%M:%S%.3f"));

    let print_msg_header = |fn_timestamp: &dyn ThreadSafeTimestampFn<Output = io::Result<()>>,

                            mut rd: &mut dyn RecordDecorator,
                            record: &Record,
                            _use_file_location: bool|

     -> io::Result<bool> {
        rd.start_timestamp()?;
        fn_timestamp(&mut rd)?;

        rd.start_whitespace()?;
        write!(rd, " ")?;

        rd.start_level()?;
        write!(rd, "{}", record.level().as_short_str())?;

        rd.start_location()?;
        if record.function().is_empty() {
            write!(rd, " {}.{} | ", record.module(), record.line())?;
        } else {
            write!(rd, " {}::{}.{} | ", record.module(), record.function(), record.line())?;
        }

        rd.start_msg()?;
        let mut count_rd = CountingWriter::new(&mut rd);
        write!(count_rd, "{}", record.msg())?;
        Ok(count_rd.count() != 0)
    };

    //Console
    let stdout_drain = if to.console() {
        let plain = slog_term::PlainSyncDecorator::new(std::io::stdout());
        let stdout_drain = slog_term::FullFormat::new(plain)
            .use_custom_timestamp(custom_timestamp)
            .use_custom_header_print(print_msg_header)
            .build()
            .fuse();
        Some(stdout_drain.filter_level(level.inner()).fuse())
    } else {
        None
    };

    //File
    let file_drain = if to.file() {
        let decorator = slog_term::PlainSyncDecorator::new(open_file(&filename)?);
        let file_drain = slog_term::FullFormat::new(decorator)
            .use_custom_timestamp(custom_timestamp)
            .use_custom_header_print(print_msg_header)
            .build()
            .fuse();

        //@TODO config ...
        let file_drain = slog_async::Async::new(file_drain)
            .chan_size(100_000)
            .overflow_strategy(slog_async::OverflowStrategy::DropAndReport)
            .build()
            .fuse();

        Some(file_drain.filter_level(level.inner()).fuse())
    } else {
        None
    };

    Ok(match (stdout_drain, file_drain) {
        (Some(stdout_drain), None) => slog::Logger::root(stdout_drain, o!()),
        (None, Some(file_drain)) => slog::Logger::root(file_drain, o!()),
        (Some(stdout_drain), Some(file_drain)) => {
            slog::Logger::root(slog::Duplicate::new(stdout_drain, file_drain).fuse(), o!())
        }
        (None, None) => slog::Logger::root(slog::Discard, o!()),
    })
}

fn open_file(filename: &str) -> Result<File> {
    OpenOptions::new()
        .create(true)
        .append(true)
        .open(filename)
        .map_err(|e| anyhow!(format!("logger file config error, filename: {}, {:?}", filename, e)))
}