gateway: config open telemetry propagator

OpenRailAssociation · Feb 15, 2024 · e9399c4 · e9399c4
1 parent 5e1c2a3
commit e9399c4
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 1 deletion.
diff --git a/docker/gateway.dev.simple.toml b/docker/gateway.dev.simple.toml
@@ -8,6 +8,8 @@ listen_addr = "0.0.0.0"
 
 [telemetry.tracing]
 enable = true
+type = "Otlp"
+service_name = "gateway"
 endpoint = "http://osrd-jaeger:4317"
 
 [[targets]]

diff --git a/gateway/README.md b/gateway/README.md
@@ -89,3 +89,12 @@ user_id = "example-user-id"
 # It's used for testing.
 require_login = true
 ```
+
+# Enable open telemetry
+```toml
+[telemetry.tracing]
+endpoint = "http://localhost:4317"
+type = "Otlp"
+enable = true
+service_name = "gateway"
+```
diff --git a/gateway/gateway.toml b/gateway/gateway.toml
@@ -4,6 +4,12 @@
 # python3 -c "import secrets, base64; print(base64.standard_b64encode(secrets.token_bytes(64)).decode())"
 secret_key = "NOT+A+SECRET++NOT+A+SECRET++NOT+A+SECRET++NOT+A+SECRET++NOT+A+SECRET++NOT+A+SECRET++NOT+A+SECRET"
 
+[telemetry.tracing]
+type = "Otlp"
+enable = true
+service_name = "gateway"
+endpoint = "http://localhost::4317"
+
 [[targets]]
 prefix = "/api"
 upstream = "http://localhost:8090"

diff --git a/gateway/src/config.rs b/gateway/src/config.rs
@@ -7,7 +7,9 @@ use figment::{
 use log::info;
 use opentelemetry::global;
 use opentelemetry_otlp::WithExportConfig;
-use opentelemetry_sdk::{runtime::TokioCurrentThread, trace::TracerProvider};
+use opentelemetry_sdk::{
+    propagation::TraceContextPropagator, runtime::TokioCurrentThread, trace::TracerProvider,
+};
 use serde::{Deserialize, Serialize};
 
 #[derive(Deserialize, Serialize, Clone)]
@@ -63,6 +65,7 @@ impl TracingTelemetry {
             .with_batch_exporter(exporter, TokioCurrentThread)
             .build();
 
+        global::set_text_map_propagator(TraceContextPropagator::new());
         global::set_tracer_provider(provider);
     }