|
8 | 8 | //! - `Trace`
|
9 | 9 | use std::sync::Once;
|
10 | 10 |
|
11 |
| -use log::{info, LevelFilter}; |
| 11 | +use tracing::info; |
| 12 | +use tracing::level_filters::LevelFilter; |
12 | 13 |
|
13 | 14 | use crate::config::v1::LogLevel;
|
14 | 15 |
|
15 | 16 | static INIT: Once = Once::new();
|
16 | 17 |
|
17 | 18 | pub fn setup(log_level: &Option<LogLevel>) {
|
18 |
| - let level = config_level_or_default(log_level); |
| 19 | + let tracing_level = config_level_or_default(log_level); |
19 | 20 |
|
20 |
| - if level == log::LevelFilter::Off { |
| 21 | + if tracing_level == LevelFilter::OFF { |
21 | 22 | return;
|
22 | 23 | }
|
23 | 24 |
|
24 | 25 | INIT.call_once(|| {
|
25 |
| - stdout_config(level); |
| 26 | + tracing_stdout_init(tracing_level, &TraceStyle::Default); |
26 | 27 | });
|
27 | 28 | }
|
28 | 29 |
|
29 | 30 | fn config_level_or_default(log_level: &Option<LogLevel>) -> LevelFilter {
|
30 | 31 | match log_level {
|
31 |
| - None => log::LevelFilter::Info, |
| 32 | + None => LevelFilter::INFO, |
32 | 33 | Some(level) => match level {
|
33 |
| - LogLevel::Off => LevelFilter::Off, |
34 |
| - LogLevel::Error => LevelFilter::Error, |
35 |
| - LogLevel::Warn => LevelFilter::Warn, |
36 |
| - LogLevel::Info => LevelFilter::Info, |
37 |
| - LogLevel::Debug => LevelFilter::Debug, |
38 |
| - LogLevel::Trace => LevelFilter::Trace, |
| 34 | + LogLevel::Off => LevelFilter::OFF, |
| 35 | + LogLevel::Error => LevelFilter::ERROR, |
| 36 | + LogLevel::Warn => LevelFilter::WARN, |
| 37 | + LogLevel::Info => LevelFilter::INFO, |
| 38 | + LogLevel::Debug => LevelFilter::DEBUG, |
| 39 | + LogLevel::Trace => LevelFilter::TRACE, |
39 | 40 | },
|
40 | 41 | }
|
41 | 42 | }
|
42 | 43 |
|
43 |
| -fn stdout_config(level: LevelFilter) { |
44 |
| - if let Err(_err) = fern::Dispatch::new() |
45 |
| - .format(|out, message, record| { |
46 |
| - out.finish(format_args!( |
47 |
| - "{} [{}][{}] {}", |
48 |
| - chrono::Local::now().format("%+"), |
49 |
| - record.target(), |
50 |
| - record.level(), |
51 |
| - message |
52 |
| - )); |
53 |
| - }) |
54 |
| - .level(level) |
55 |
| - .chain(std::io::stdout()) |
56 |
| - .apply() |
57 |
| - { |
58 |
| - panic!("Failed to initialize logging.") |
59 |
| - } |
| 44 | +fn tracing_stdout_init(filter: LevelFilter, style: &TraceStyle) { |
| 45 | + let builder = tracing_subscriber::fmt().with_max_level(filter); |
| 46 | + |
| 47 | + let () = match style { |
| 48 | + TraceStyle::Default => builder.init(), |
| 49 | + TraceStyle::Pretty(display_filename) => builder.pretty().with_file(*display_filename).init(), |
| 50 | + TraceStyle::Compact => builder.compact().init(), |
| 51 | + TraceStyle::Json => builder.json().init(), |
| 52 | + }; |
60 | 53 |
|
61 | 54 | info!("logging initialized.");
|
62 | 55 | }
|
| 56 | + |
| 57 | +#[derive(Debug)] |
| 58 | +pub enum TraceStyle { |
| 59 | + Default, |
| 60 | + Pretty(bool), |
| 61 | + Compact, |
| 62 | + Json, |
| 63 | +} |
| 64 | + |
| 65 | +impl std::fmt::Display for TraceStyle { |
| 66 | + fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { |
| 67 | + let style = match self { |
| 68 | + TraceStyle::Default => "Default Style", |
| 69 | + TraceStyle::Pretty(path) => match path { |
| 70 | + true => "Pretty Style with File Paths", |
| 71 | + false => "Pretty Style without File Paths", |
| 72 | + }, |
| 73 | + |
| 74 | + TraceStyle::Compact => "Compact Style", |
| 75 | + TraceStyle::Json => "Json Format", |
| 76 | + }; |
| 77 | + |
| 78 | + f.write_str(style) |
| 79 | + } |
| 80 | +} |
0 commit comments