richat-shared 8.0.1

Richat Shared code
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

use {
    serde::Deserialize,
    std::io::{self, IsTerminal},
    thiserror::Error,
    tracing::Subscriber,
    tracing_subscriber::{
        filter::{EnvFilter, FromEnvError, LevelFilter},
        fmt::layer,
        layer::{Layer, SubscriberExt},
        registry::LookupSpan,
        util::{SubscriberInitExt, TryInitError},
    },
};

#[derive(Debug, Default, Clone, Deserialize)]
#[serde(deny_unknown_fields, default)]
pub struct ConfigTracing {
    pub json: bool,
}

#[derive(Debug, Error)]
pub enum TracingSetupError {
    #[error(transparent)]
    FromEnv(#[from] FromEnvError),
    #[error(transparent)]
    Init(#[from] TryInitError),
}

pub fn setup(json: bool) -> Result<(), TracingSetupError> {
    let env = EnvFilter::builder()
        .with_default_directive(LevelFilter::INFO.into())
        .from_env()?;

    tracing_subscriber::registry()
        .with(env)
        .with(create_io_layer(json))
        .try_init()?;

    Ok(())
}

fn create_io_layer<S>(json: bool) -> Box<dyn Layer<S> + Send + Sync + 'static>
where
    S: Subscriber,
    for<'a> S: LookupSpan<'a>,
{
    let is_atty = io::stdout().is_terminal() && io::stderr().is_terminal();
    let io_layer = layer().with_ansi(is_atty).with_line_number(true);

    if json {
        Box::new(io_layer.json())
    } else {
        Box::new(io_layer)
    }
}