azure_servicebus_async_batch_send_and_receive_queue_messages.py

1import asyncio
2from azure.servicebus.aio import ServiceBusClient
3from azure.servicebus import ServiceBusMessage
4from azure.identity.aio import DefaultAzureCredential
5
6# Substitution variables
7NAMESPACE_CONNECTION_STR = "YOUR_SERVICE_BUS_CONNECTION_STRING"
8QUEUE_NAME = "YOUR_QUEUE_NAME"
9
10async def send_single_message(sender):
11    # Create a ServiceBusMessage object
12    message = ServiceBusMessage("Single Message")
13    # Send the message to the queue
14    await sender.send_messages(message)
15    print("Sent a single message")
16
17async def send_a_list_of_messages(sender):
18    # Create a list of messages
19    messages = [ServiceBusMessage("Message in list") for _ in range(5)]
20    # Send the list of messages to the queue
21    await sender.send_messages(messages)
22    print("Sent a list of 5 messages")
23
24async def send_batch_message(sender):
25    # Create a batch object
26    async with sender:
27        batch_message = await sender.create_message_batch()
28        for _ in range(10):
29            try:
30                # Add a message to the batch
31                batch_message.add_message(ServiceBusMessage("Message inside a ServiceBusMessageBatch"))
32            except ValueError:
33                # ServiceBusMessageBatch object reaches max_size.
34                # New ServiceBusMessageBatch object can be created here to send more messages.
35                break
36        # Send the batch of messages to the queue
37        await sender.send_messages(batch_message)
38    print("Sent a batch of 10 messages")
39
40async def run():
41    # create a Service Bus client using the connection string
42    async with ServiceBusClient.from_connection_string(
43        conn_str=NAMESPACE_CONNECTION_STR,
44        logging_enable=True
45    ) as servicebus_client:
46        # get a Queue Sender object to send messages to the queue
47        sender = servicebus_client.get_queue_sender(queue_name=QUEUE_NAME)
48        async with sender:
49            # send one message
50            await send_single_message(sender)
51            # send a list of messages
52            await send_a_list_of_messages(sender)
53            # send a batch of messages
54            await send_batch_message(sender)
55
56        # get a Queue Receiver object to receive messages from the queue
57        receiver = servicebus_client.get_queue_receiver(queue_name=QUEUE_NAME, max_wait_time=5)
58        async with receiver:
59            received_msgs = await receiver.receive_messages(max_message_count=20, max_wait_time=5)
60            for msg in received_msgs:
61                print("Received: " + str(msg))
62                # complete the message so that it is not received again
63                await receiver.complete_message(msg)
64
65asyncio.run(run())