opentelemetry_wsgi_middleware_tracing_with_console_exporter.py

1from opentelemetry.instrumentation.wsgi import OpenTelemetryMiddleware
2from opentelemetry import trace
3from opentelemetry.sdk.trace import TracerProvider
4from opentelemetry.sdk.trace.export import (
5    BatchSpanProcessor,
6    ConsoleSpanExporter,
7)
8
9# Initialize tracing
10trace.set_tracer_provider(TracerProvider())
11tracer_provider = trace.get_tracer_provider()
12tracer_provider.add_span_processor(
13    BatchSpanProcessor(ConsoleSpanExporter())
14)
15
16# A simple WSGI application
17def simple_wsgi_app(environ, start_response):
18    status = '200 OK'
19    headers = [('Content-type', 'text/plain; charset=utf-8')]
20    start_response(status, headers)
21    return [b"Hello World"]
22
23# Wrap the WSGI application with OpenTelemetry middleware
24app = OpenTelemetryMiddleware(simple_wsgi_app)
25
26# Now 'app' can be used with any WSGI server (e.g., Gunicorn, uWSGI, werkzeug)
27if __name__ == "__main__":
28    from werkzeug.serving import run_simple
29    run_simple('localhost', 5000, app)