-
Notifications
You must be signed in to change notification settings - Fork 3
/
logs.rs
99 lines (93 loc) · 3.54 KB
/
logs.rs
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
use anyhow::{Context, Result};
use tracing_appender::non_blocking::WorkerGuard;
use tracing_subscriber::{
fmt::{
self,
format::{Format, PrettyFields},
},
prelude::__tracing_subscriber_SubscriberExt,
util::SubscriberInitExt,
EnvFilter,
};
use crate::args::{Args, LogStructure};
// TODO: custom logging output format?
// struct Output;
// impl<'writer> FormatFields<'writer> for Output {
// fn format_fields<R: tracing_subscriber::prelude::__tracing_subscriber_field_RecordFields>(
// &self,
// writer: &'writer mut dyn std::fmt::Write,
// fields: R,
// ) -> std::fmt::Result {
// todo!()
// }
// }
pub fn setup(args: &Args) -> Result<Option<WorkerGuard>> {
let filter_layer = EnvFilter::try_from_default_env()
.or_else(|_| EnvFilter::try_new("info"))?
.add_directive("tokio_util=off".parse()?)
.add_directive("h2=off".parse()?);
let registry = tracing_subscriber::registry().with(filter_layer);
// this code duplication is unfortunate-- but it looks
// like tracing doesn't support dynamically configured subscribers
Ok(match &args.log_file {
Some(file) => {
let appender = tracing_appender::rolling::never(
file.parent()
.with_context(|| format!("failed to start log on file {:#?}", file))?,
file.file_name()
.with_context(|| format!("failed to start log on file {:#?}", file))?,
);
let (non_blocking_appender, _guard) = tracing_appender::non_blocking(appender);
// dual layers will get registry to log to both stdout and file
match args.output {
LogStructure::Pretty => {
registry
.with(
fmt::layer()
.event_format(
Format::default().pretty().with_source_location(false),
)
.fmt_fields(PrettyFields::new())
.with_target(false),
)
.with(fmt::layer().with_writer(non_blocking_appender))
.init();
}
LogStructure::Debug => {
registry
.with(fmt::layer())
.with(fmt::layer().with_writer(non_blocking_appender))
.init();
}
LogStructure::Json => {
registry
.with(fmt::layer().json())
.with(fmt::layer().json().with_writer(non_blocking_appender))
.init();
}
}
Some(_guard)
}
None => {
match args.output {
LogStructure::Pretty => {
registry
.with(
fmt::layer()
.event_format(Format::default().with_source_location(false))
.fmt_fields(PrettyFields::new())
.with_target(false),
)
.init();
}
LogStructure::Debug => {
registry.with(fmt::layer()).init();
}
LogStructure::Json => {
registry.with(fmt::layer().json()).init();
}
}
None
}
})
}