|
| 1 | +import logging |
| 2 | +import os |
| 3 | +from urllib.parse import urlparse |
| 4 | + |
1 | 5 | from fastmcp.server.middleware import Middleware, MiddlewareContext |
2 | | -from opentelemetry import trace |
| 6 | +from opentelemetry import metrics, trace |
| 7 | +from opentelemetry._logs import set_logger_provider |
| 8 | +from opentelemetry.exporter.otlp.proto.grpc._log_exporter import OTLPLogExporter |
| 9 | +from opentelemetry.exporter.otlp.proto.grpc.metric_exporter import OTLPMetricExporter |
| 10 | +from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter |
| 11 | +from opentelemetry.sdk._logs import LoggerProvider, LoggingHandler |
| 12 | +from opentelemetry.sdk._logs.export import BatchLogRecordProcessor |
| 13 | +from opentelemetry.sdk.metrics import MeterProvider |
| 14 | +from opentelemetry.sdk.metrics.export import PeriodicExportingMetricReader |
| 15 | +from opentelemetry.sdk.resources import Resource |
| 16 | +from opentelemetry.sdk.trace import TracerProvider |
| 17 | +from opentelemetry.sdk.trace.export import SimpleSpanProcessor |
3 | 18 | from opentelemetry.trace import Status, StatusCode |
4 | 19 |
|
5 | 20 |
|
| 21 | +def configure_aspire_dashboard(service_name: str = "expenses-mcp"): |
| 22 | + """Configure OpenTelemetry to send telemetry to the Aspire standalone dashboard.""" |
| 23 | + otlp_endpoint = os.getenv("OTEL_EXPORTER_OTLP_ENDPOINT", "http://localhost:4317") |
| 24 | + parsed_endpoint = urlparse(otlp_endpoint) |
| 25 | + use_insecure = parsed_endpoint.scheme not in ("https", "grpcs") |
| 26 | + |
| 27 | + # Create resource with service name |
| 28 | + resource = Resource.create({"service.name": service_name}) |
| 29 | + |
| 30 | + # Configure Tracing |
| 31 | + tracer_provider = TracerProvider(resource=resource) |
| 32 | + tracer_provider.add_span_processor( |
| 33 | + SimpleSpanProcessor(OTLPSpanExporter(endpoint=otlp_endpoint, insecure=use_insecure)) |
| 34 | + ) |
| 35 | + trace.set_tracer_provider(tracer_provider) |
| 36 | + |
| 37 | + # Configure Metrics |
| 38 | + metric_reader = PeriodicExportingMetricReader( |
| 39 | + OTLPMetricExporter(endpoint=otlp_endpoint, insecure=use_insecure) |
| 40 | + ) |
| 41 | + meter_provider = MeterProvider(resource=resource, metric_readers=[metric_reader]) |
| 42 | + metrics.set_meter_provider(meter_provider) |
| 43 | + |
| 44 | + # Configure Logging |
| 45 | + logger_provider = LoggerProvider(resource=resource) |
| 46 | + logger_provider.add_log_record_processor( |
| 47 | + BatchLogRecordProcessor(OTLPLogExporter(endpoint=otlp_endpoint, insecure=use_insecure)) |
| 48 | + ) |
| 49 | + set_logger_provider(logger_provider) |
| 50 | + |
| 51 | + # Add logging handler to send Python logs to OTLP |
| 52 | + root_logger = logging.getLogger() |
| 53 | + handler_exists = any( |
| 54 | + isinstance(existing, LoggingHandler) |
| 55 | + and getattr(existing, "logger_provider", None) is logger_provider |
| 56 | + for existing in root_logger.handlers |
| 57 | + ) |
| 58 | + |
| 59 | + if not handler_exists: |
| 60 | + handler = LoggingHandler(level=logging.NOTSET, logger_provider=logger_provider) |
| 61 | + root_logger.addHandler(handler) |
| 62 | + |
| 63 | + |
6 | 64 | class OpenTelemetryMiddleware(Middleware): |
7 | 65 | """Middleware that creates OpenTelemetry spans for MCP operations.""" |
8 | 66 |
|
|
0 commit comments