opentelemetry_pika_rabbitmq_instrumentation_publish_consume_tracing.py

1import pika
2from opentelemetry.instrumentation.pika import PikaInstrumentor
3
4# Instrument pika
5PikaInstrumentor().instrument()
6
7# Standard Pika connection and channel setup
8connection = pika.BlockingConnection(pika.ConnectionParameters('localhost'))
9channel = connection.channel()
10
11channel.queue_declare(queue='hello')
12
13# Tracing is automatically captured during basic_publish
14channel.basic_publish(exchange='',
15                      routing_key='hello',
16                      body='Hello World!')
17
18print(" [x] Sent 'Hello World!'")
19
20# Tracing is also captured during consumption
21def callback(ch, method, properties, body):
22    print(f" [x] Received {body}")
23
24channel.basic_consume(queue='hello',
25                      on_message_callback=callback,
26                      auto_ack=True)
27
28print(' [*] Waiting for messages. To exit press CTRL+C')
29# channel.start_consuming() # Uncomment to start the consumer loop