infrarust 1.1.2

A Rust adaptation of the Infrared Minecraft proxy
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
88
89
90
91
92
93
94

use opentelemetry::global;
use opentelemetry::trace::TracerProvider;
use opentelemetry_otlp::WithExportConfig;
use opentelemetry_sdk::trace::{SdkTracerProvider, TracerProviderBuilder};
use opentelemetry_sdk::Resource;
use std::str::FromStr;
use tracing::Level;
use tracing_subscriber::Layer;
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};

pub struct TracerProviderGuard(pub SdkTracerProvider);

impl Drop for TracerProviderGuard {
    fn drop(&mut self) {
        if let Err(exporter) = self.0.shutdown() {
            println!("Failed to shutdown exporter: {:?}", exporter);
        };
    }
}

pub fn init_tracer_provider(resource: Resource, export_url: Option<String>) -> TracerProviderGuard {
    let mut provider = TracerProviderBuilder::default();
    if export_url.clone().is_some() {
        let exporter = opentelemetry_otlp::SpanExporter::builder()
            .with_tonic()
            .with_endpoint(export_url.clone().unwrap().to_string())
            .build()
            .unwrap();

        provider = provider.with_batch_exporter(exporter);
    }

    provider = provider.with_resource(resource);

    let provider = provider.build();

    if export_url.is_some() {
        init_subscriber_with_optl(&provider);
    } else {
        init_subscriber();
    }
    global::set_tracer_provider(provider.clone());
    TracerProviderGuard(provider)
}

pub fn init_subscriber_with_optl(provider: &SdkTracerProvider) {
    let tracer = provider.tracer("proxy");
    let layer = tracing_opentelemetry::layer().with_tracer(tracer);

    let log_level = if cfg!(debug_assertions) {
        Level::DEBUG
    } else {
        Level::INFO
    };

    let fmt_layer = tracing_subscriber::fmt::layer()
        .compact()
        .with_level(true) // Garder le niveau de log
        .with_filter(tracing_subscriber::filter::LevelFilter::from_level(
            log_level,
        ));

    tracing_subscriber::registry()
        .with(
            tracing_subscriber::EnvFilter::from_str(&format!("infrarust={}", log_level))
                .unwrap_or_else(|_| tracing_subscriber::EnvFilter::from_default_env()),
        )
        .with(layer)
        .with(fmt_layer)
        .init();
}

pub fn init_subscriber() {
    let log_level = if cfg!(debug_assertions) {
        Level::DEBUG
    } else {
        Level::INFO
    };

    let fmt_layer = tracing_subscriber::fmt::layer()
        .compact()
        .with_level(true)
        .with_filter(tracing_subscriber::filter::LevelFilter::from_level(
            log_level,
        ));

    tracing_subscriber::registry()
        .with(
            tracing_subscriber::EnvFilter::from_str(&format!("infrarust={}", log_level))
                .unwrap_or_else(|_| tracing_subscriber::EnvFilter::from_default_env()),
        )
        .with(fmt_layer)
        .init();
}