tracer

.env.template

@@ -12,3 +12,4 @@ NACHET_MAX_CONTENT_LENGTH=
 NACHET_VALID_EXTENSION=
 NACHET_VALID_DIMENSION=
 DEV_USER_EMAIL=
+OTEL_COLLECTOR_ENDPOINT=

app.py

@@ -16,6 +16,10 @@
 from collections import namedtuple
 from cryptography.fernet import Fernet
 from opentelemetry.instrumentation.asgi import OpenTelemetryMiddleware
+from opentelemetry import trace
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
 
 load_dotenv() # noqa: E402
 
@@ -86,6 +90,8 @@ class MaxContentLengthWarning(APIWarnings):
 
 NACHET_DATA = os.getenv("NACHET_DATA")
 
+OTEL_COLLECTOR_URL = os.getenv("OTEL_COLLECTOR_ENDPOINT")
+
 try:
     VALID_EXTENSION = json.loads(os.getenv("NACHET_VALID_EXTENSION"))
     VALID_DIMENSION = json.loads(os.getenv("NACHET_VALID_DIMENSION"))
@@ -131,6 +137,12 @@ class MaxContentLengthWarning(APIWarnings):
     "validators": []
 }
 
+
+trace.set_tracer_provider(TracerProvider())
+tracer = trace.get_tracer(__name__)
+span_processor = BatchSpanProcessor(OTLPSpanExporter(endpoint=OTEL_COLLECTOR_URL))
+trace.get_tracer_provider().add_span_processor(span_processor)
+
 app = Quart(__name__)
 app = cors(app, allow_origin="*", allow_methods=["GET", "POST", "OPTIONS"])
 app.config["MAX_CONTENT_LENGTH"] = MAX_CONTENT_LENGTH_MEGABYTES * 1024 * 1024

requirements.txt

@@ -11,4 +11,8 @@ pyyaml
 pydantic
 python-magic
 opentelemetry-instrumentation-asgi
+opentelemetry-distro
+opentelemetry-exporter-otlp
+opentelemetry-sdk
+opentelemetry-api
 nachet-datastore @git+https://github.com/ai-cfia/ailab-datastore.git@main