Add appropriate metadata to traces

DiamondLightSource · Apr 17, 2024 · 49d9c04 · 49d9c04
1 parent 785e9c1
commit 49d9c04
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -29,6 +29,7 @@ dependencies = [
     "pymysql",
     "opentelemetry-api",
     "opentelemetry-sdk",
+    "opentelemetry-semantic-conventions",
     "opentelemetry-instrumentation-asgi",
     "opentelemetry-exporter-otlp",
 ]

diff --git a/src/graph_energy_scan/telemetry.py b/src/graph_energy_scan/telemetry.py
@@ -1,11 +1,22 @@
 from opentelemetry import trace
 from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
+from opentelemetry.sdk.resources import Resource
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import BatchSpanProcessor, ConsoleSpanExporter
+from opentelemetry.semconv.resource import ResourceAttributes
+
+from graph_energy_scan import __version__
 
 
 def setup_telemetry(otel_collector_url: str):
-    tracer_provider = TracerProvider()
+    tracer_provider = TracerProvider(
+        resource=Resource.create(
+            {
+                ResourceAttributes.SERVICE_NAME: "graph-energy-scan",
+                ResourceAttributes.SERVICE_VERSION: __version__,
+            }
+        )
+    )
     tracer_provider.add_span_processor(BatchSpanProcessor(ConsoleSpanExporter()))
     tracer_provider.add_span_processor(
         BatchSpanProcessor(OTLPSpanExporter(otel_collector_url))