Add signoz

services/demo_session/app/main.py

@@ -3,16 +3,51 @@ Demo Session Service - Main Application
 Manages isolated demo sessions with ephemeral data
 """
 
-from fastapi import FastAPI, Request
+from fastapi import FastAPI, Request, Response
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 import structlog
 from contextlib import asynccontextmanager
+import os
 
 from app.core import settings, DatabaseManager
 from app.api import demo_sessions, demo_accounts, demo_operations, internal
 from shared.redis_utils import initialize_redis, close_redis
+from shared.monitoring.logging import setup_logging
+from shared.monitoring.metrics import MetricsCollector, add_metrics_middleware
 
+# OpenTelemetry imports
+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
+from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
+from opentelemetry.instrumentation.httpx import HTTPXClientInstrumentor
+from opentelemetry.instrumentation.redis import RedisInstrumentor
+from opentelemetry.sdk.resources import Resource
+
+# Configure OpenTelemetry tracing
+def setup_tracing(service_name: str = "demo-session"):
+    """Initialize OpenTelemetry tracing with OTLP exporter for Jaeger"""
+    resource = Resource.create({"service.name": service_name})
+
+    otlp_exporter = OTLPSpanExporter(
+        endpoint=os.getenv("OTEL_EXPORTER_OTLP_ENDPOINT", "http://otel-collector.monitoring.svc.cluster.local:4317"),
+        insecure=True
+    )
+
+    provider = TracerProvider(resource=resource)
+    processor = BatchSpanProcessor(otlp_exporter)
+    provider.add_span_processor(processor)
+    trace.set_tracer_provider(provider)
+
+    return provider
+
+# Initialize tracing
+tracer_provider = setup_tracing("demo-session")
+
+# Setup logging
+setup_logging("demo-session", getattr(settings, 'LOG_LEVEL', 'INFO'))
 logger = structlog.get_logger()
 
 # Initialize database
@@ -34,6 +69,10 @@ async def lifespan(app: FastAPI):
         max_connections=50
     )
 
+    # Start metrics server
+    metrics_collector.start_metrics_server(8080)
+    logger.info("Metrics server started on port 8080")
+
     logger.info("Demo Session Service started successfully")
 
     yield
@@ -52,6 +91,21 @@ app = FastAPI(
     lifespan=lifespan
 )
 
+# Instrument FastAPI with OpenTelemetry
+FastAPIInstrumentor.instrument_app(app)
+
+# Instrument httpx for outgoing requests
+HTTPXClientInstrumentor().instrument()
+
+# Instrument Redis
+RedisInstrumentor().instrument()
+
+# Initialize metrics collector
+metrics_collector = MetricsCollector("demo-session")
+
+# Add metrics middleware to track HTTP requests
+add_metrics_middleware(app, metrics_collector)
+
 # CORS middleware
 app.add_middleware(
     CORSMiddleware,
@@ -110,6 +164,15 @@ async def health():
     }
 
 
+@app.get("/metrics")
+async def metrics():
+    """Prometheus metrics endpoint"""
+    return Response(
+        content=metrics_collector.get_metrics(),
+        media_type="text/plain; version=0.0.4; charset=utf-8"
+    )
+
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(