imbibe-telemetry 0.0.1

contains helper methods for open-telemetry support for imbibe
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

mod error;

pub use self::error::TelemetryError;

use core::time::Duration;

use std::borrow::Cow;

use opentelemetry::trace::TracerProvider;
use opentelemetry_otlp::{SpanExporter, WithExportConfig};
use opentelemetry_sdk::trace::SdkTracerProvider;
use tracing::{Subscriber, subscriber};
use tracing_subscriber::{EnvFilter, Registry, fmt::format::FmtSpan, layer::SubscriberExt};

use self::error::Result;

pub fn init_subscriber<S>(s: S) -> Result<()>
where
	S: Subscriber + Send + Sync,
{
	subscriber::set_global_default(s).map_err(From::from)
}

pub fn make_tracing_subscriber<N, U>(
	name: N,
	exporter_endpoint: U,
	timeout: Duration,
	env_filter: EnvFilter,
) -> Result<impl Subscriber + Send + Sync>
where
	Cow<'static, str>: From<N>,
	String: From<U>,
{
	let exporter = SpanExporter::builder()
		.with_tonic()
		.with_endpoint(exporter_endpoint)
		.with_timeout(timeout)
		.build()?;

	let tracer = SdkTracerProvider::builder().with_batch_exporter(exporter).build().tracer(name);

	let telemetry = tracing_opentelemetry::layer().with_tracer(tracer);

	let subscriber = Registry::default()
		.with(telemetry)
		.with(
			tracing_subscriber::fmt::layer()
				.with_target(false)
				.with_line_number(true)
				.with_span_events(FmtSpan::NEW | FmtSpan::CLOSE),
		)
		.with(env_filter);

	Ok(subscriber)
}