pi_logger 0.9.4

pi_logger
Documentation 
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

use crate::observability::config::{LogFormat, LogLocalConfig};
use crate::observability::filter::SharedOrderedFilter;
use chrono::Utc;
use chrono_tz::Tz;
use std::fmt;
use tracing_subscriber::{
    fmt::{format::Writer, time::FormatTime},
    Layer, Registry,
};

#[derive(Clone, Copy)]
struct TimezoneTimer(Tz);

impl FormatTime for TimezoneTimer {
    fn format_time(&self, writer: &mut Writer<'_>) -> fmt::Result {
        write!(writer, "{}", Utc::now().with_timezone(&self.0).to_rfc3339())
    }
}

/// Builds the stdout JSON layer.
///
/// This uses the same shared ordered filter as local file and remote outputs.
pub fn build_console_layer(
    format: LogFormat,
    timezone: Tz,
    filter: SharedOrderedFilter,
) -> Box<dyn Layer<Registry> + Send + Sync> {
    let base = tracing_subscriber::fmt::layer()
        .with_timer(TimezoneTimer(timezone))
        .with_target(true)
        .with_file(true)
        .with_line_number(true)
        .with_thread_ids(true)
        .with_thread_names(true)
        .with_writer(std::io::stdout);

    match format {
        LogFormat::Json => base
            .json()
            .with_current_span(true)
            .with_span_list(true)
            .with_filter(filter)
            .boxed(),
        LogFormat::Compact => base.compact().with_filter(filter).boxed(),
        LogFormat::Full => base.with_filter(filter).boxed(),
        LogFormat::Pretty => base.pretty().with_filter(filter).boxed(),
    }
}

/// Builds the local rolling JSON file layer.
///
/// The supplied `filter` is the shared ordered filter used by all log outputs.
pub fn build_local_layer(
    config: &LogLocalConfig,
    format: LogFormat,
    timezone: Tz,
    filter: SharedOrderedFilter,
) -> (
    Box<dyn Layer<Registry> + Send + Sync>,
    tracing_appender::non_blocking::WorkerGuard,
) {
    let file_appender = tracing_appender::rolling::daily(&config.file_dir, &config.file_name);
    let (writer, guard) = tracing_appender::non_blocking(file_appender);

    let base = tracing_subscriber::fmt::layer()
        .with_timer(TimezoneTimer(timezone))
        .with_target(true)
        .with_file(true)
        .with_line_number(true)
        .with_thread_ids(true)
        .with_thread_names(true)
        .with_writer(writer);

    let layer = match format {
        LogFormat::Json => base
            .json()
            .with_current_span(true)
            .with_span_list(true)
            .with_filter(filter)
            .boxed(),
        LogFormat::Compact => base.compact().with_filter(filter).boxed(),
        LogFormat::Full => base.with_filter(filter).boxed(),
        LogFormat::Pretty => base.pretty().with_filter(filter).boxed(),
    };

    (layer, guard)
}