use metrics_tracing_context::MetricsLayer;

use tracing::{

dispatcher::{set_global_default, Dispatch},

span::Span,

};

use tracing_limit::Limit;

use tracing_log::LogTracer;

use tracing_subscriber::{layer::SubscriberExt, FmtSubscriber};

pub use tracing_futures::Instrument;

pub use tracing_tower::{InstrumentableService, InstrumentedService};

pub fn init(color: bool, json: bool, levels: &str) {

let dispatch = if json {

let subscriber = FmtSubscriber::builder()

.with_env_filter(levels)

.json()

.flatten_event(true)

.finish()

.with(Limit::default())

.with(MetricsLayer::new());

Dispatch::new(subscriber)

} else {

let subscriber = FmtSubscriber::builder()

.with_ansi(color)

.with_env_filter(levels)

.finish()

.with(Limit::default())

.with(MetricsLayer::new());

Dispatch::new(subscriber)

};

let _ = LogTracer::init();

let _ = set_global_default(dispatch);

}

pub fn current_span() -> Span {

Span::current()

}