Spaces:
Build error
Build error
File size: 2,515 Bytes
d8435ba |
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 |
use std::collections::HashSet;
use serde::{Deserialize, Serialize};
use tracing_subscriber::fmt;
use super::*;
#[derive(Clone, Debug, Default, Eq, PartialEq, Deserialize, Serialize)]
#[serde(default)]
pub struct LoggerConfig {
#[serde(flatten)]
pub default: default::Config,
#[serde(default)]
pub on_disk: on_disk::Config,
}
impl LoggerConfig {
pub fn with_top_level_directive(&self, log_level: Option<String>) -> Self {
let mut logger_config = self.clone();
if logger_config.default.log_level.is_some() && log_level.is_some() {
eprintln!(
"Both top-level `log_level` and `logger.log_level` config directives are used. \
`logger.log_level` takes priority, so top-level `log_level` will be ignored."
);
}
if logger_config.default.log_level.is_none() {
logger_config.default.log_level = log_level;
}
logger_config
}
pub fn merge(&mut self, other: Self) {
self.default.merge(other.default);
self.on_disk.merge(other.on_disk);
}
}
#[derive(Copy, Clone, Debug, PartialEq, Eq, Deserialize, Serialize, Hash)]
#[serde(rename_all = "lowercase")]
pub enum SpanEvent {
New,
Enter,
Exit,
Close,
}
impl SpanEvent {
pub fn unwrap_or_default_config(events: &Option<HashSet<Self>>) -> fmt::format::FmtSpan {
Self::into_fmt_span(events.as_ref().unwrap_or(&HashSet::new()).iter().copied())
}
pub fn into_fmt_span(events: impl IntoIterator<Item = Self>) -> fmt::format::FmtSpan {
events
.into_iter()
.fold(fmt::format::FmtSpan::NONE, |events, event| {
events | event.into()
})
}
}
impl From<SpanEvent> for fmt::format::FmtSpan {
fn from(event: SpanEvent) -> Self {
match event {
SpanEvent::New => fmt::format::FmtSpan::NEW,
SpanEvent::Enter => fmt::format::FmtSpan::ENTER,
SpanEvent::Exit => fmt::format::FmtSpan::EXIT,
SpanEvent::Close => fmt::format::FmtSpan::CLOSE,
}
}
}
#[derive(Copy, Clone, Debug, Eq, PartialEq, Deserialize, Serialize, Default)]
#[serde(rename_all = "lowercase")]
pub enum Color {
#[default]
Auto,
#[serde(untagged)]
Explicit(bool),
}
impl Color {
pub fn to_bool(self) -> bool {
match self {
Self::Auto => colored::control::SHOULD_COLORIZE.should_colorize(),
Self::Explicit(bool) => bool,
}
}
}
|