81 lines
2.7 KiB
Rust
81 lines
2.7 KiB
Rust
use actix_web::http::header::HeaderMap;
|
|
use actix_web::HttpRequest;
|
|
use opentelemetry::global::BoxedTracer;
|
|
use opentelemetry::{global, Context, KeyValue};
|
|
use opentelemetry::propagation::TextMapPropagator;
|
|
use opentelemetry_appender_log::OpenTelemetryLogBridge;
|
|
use opentelemetry_otlp::WithExportConfig;
|
|
use opentelemetry_sdk::logs::{BatchLogProcessor, SdkLoggerProvider};
|
|
use opentelemetry_sdk::propagation::TraceContextPropagator;
|
|
use opentelemetry_sdk::Resource;
|
|
|
|
pub fn global_tracer() -> BoxedTracer {
|
|
global::tracer("image-server")
|
|
}
|
|
|
|
pub fn init_tracing() {
|
|
let resources = Resource::builder()
|
|
.with_attributes([
|
|
KeyValue::new("service.name", "image-server"),
|
|
KeyValue::new("service.version", env!("CARGO_PKG_VERSION")),
|
|
])
|
|
.build();
|
|
|
|
let span_exporter = opentelemetry_otlp::SpanExporter::builder()
|
|
.with_tonic()
|
|
.with_endpoint(std::env::var("OTLP_OTLS_ENDPOINT").unwrap())
|
|
.build()
|
|
.unwrap();
|
|
|
|
let tracer_provider = opentelemetry_sdk::trace::SdkTracerProvider::builder()
|
|
.with_batch_exporter(span_exporter)
|
|
.with_resource(resources)
|
|
.build();
|
|
|
|
global::set_tracer_provider(tracer_provider);
|
|
}
|
|
|
|
pub fn init_logs() {
|
|
let otlp_exporter = opentelemetry_otlp::LogExporter::builder()
|
|
.with_tonic()
|
|
.with_endpoint(std::env::var("OTLP_OTLS_ENDPOINT").unwrap())
|
|
.build()
|
|
.unwrap();
|
|
|
|
let exporter = opentelemetry_stdout::LogExporter::default();
|
|
|
|
let resources = Resource::builder()
|
|
.with_attributes([
|
|
KeyValue::new("service.name", "image-server"),
|
|
KeyValue::new("service.version", "1.0"),
|
|
])
|
|
.build();
|
|
|
|
let log_provider = SdkLoggerProvider::builder()
|
|
.with_log_processor(BatchLogProcessor::builder(exporter).build())
|
|
.with_log_processor(BatchLogProcessor::builder(otlp_exporter).build())
|
|
.with_resource(resources)
|
|
.build();
|
|
|
|
let otel_log_appender = OpenTelemetryLogBridge::new(&log_provider);
|
|
log::set_boxed_logger(Box::new(otel_log_appender)).expect("Unable to set boxed logger");
|
|
//TODO: Still set this with the env? Ideally we still have a clean/simple local logger for local dev
|
|
log::set_max_level(log::LevelFilter::Info);
|
|
}
|
|
|
|
struct HeaderExtractor<'a>(&'a HeaderMap);
|
|
|
|
impl<'a> opentelemetry::propagation::Extractor for HeaderExtractor<'a> {
|
|
fn get(&self, key: &str) -> Option<&str> {
|
|
self.0.get(key).and_then(|v| v.to_str().ok())
|
|
}
|
|
|
|
fn keys(&self) -> Vec<&str> {
|
|
self.0.keys().map(|k| k.as_str()).collect()
|
|
}
|
|
}
|
|
pub fn extract_context_from_request(req: &HttpRequest) -> Context {
|
|
let propagator = TraceContextPropagator::new();
|
|
propagator.extract(&HeaderExtractor(req.headers()))
|
|
}
|