From 49d9c0470a6c4663446e7faeaaf34eef356ce539 Mon Sep 17 00:00:00 2001
From: Garry O'Donnell <garry.o'donnell@diamond.ac.uk>
Date: Wed, 17 Apr 2024 11:30:41 +0000
Subject: [PATCH] Add appropriate metadata to traces

---
 pyproject.toml                     |  1 +
 src/graph_energy_scan/telemetry.py | 13 ++++++++++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 524018b..20df7c2 100644
--- 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
index c81fe22..319d59b 100644
--- 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))