add structlog

src/fastapi_dynamic_response/tracing.py

@@ -0,0 +1,35 @@
+# tracing.py
+
+from fastapi_dynamic_response.settings import Settings
+from opentelemetry import trace
+from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
+
+# from opentelemetry.exporter.richconsole import RichConsoleSpanExporter
+from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+
+
+def configure_tracing(app):
+    settings = Settings()
+    trace.set_tracer_provider(TracerProvider())
+    tracer_provider = trace.get_tracer_provider()
+
+    if settings.ENV == "local":
+        # Use console exporter for local development
+        # span_exporter = RichConsoleSpanExporter()
+        # span_processor = SimpleSpanProcessor(span_exporter)
+        # span_exporter = OTLPSpanExporter()
+        span_exporter = OTLPSpanExporter(
+            endpoint="http://localhost:4317", insecure=True
+        )
+        span_processor = BatchSpanProcessor(span_exporter)
+    else:
+        # Use OTLP exporter for production
+        span_exporter = OTLPSpanExporter()
+        span_processor = BatchSpanProcessor(span_exporter)
+
+    tracer_provider.add_span_processor(span_processor)
+
+    # Instrument FastAPI
+    FastAPIInstrumentor.instrument_app(app)