1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
#!/usr/bin/env python
# --------------------------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
# --------------------------------------------------------------------------------------------
"""
Example to show streaming sending events with different options to an Event Hub asynchronously.
"""
import time
import asyncio
import os
from azure.eventhub.aio import EventHubProducerClient
from azure.eventhub import EventData
from azure.identity.aio import DefaultAzureCredential
FULLY_QUALIFIED_NAMESPACE = os.environ["EVENT_HUB_HOSTNAME"]
EVENTHUB_NAME = os.environ["EVENT_HUB_NAME"]
async def run():
producer = EventHubProducerClient(
fully_qualified_namespace=FULLY_QUALIFIED_NAMESPACE,
eventhub_name=EVENTHUB_NAME,
credential=DefaultAzureCredential(),
)
to_send_message_cnt = 500
bytes_per_message = 256
async with producer:
event_data_batch = await producer.create_batch()
for i in range(to_send_message_cnt):
event_data = EventData("D" * bytes_per_message)
try:
event_data_batch.add(event_data)
except ValueError:
await producer.send_batch(event_data_batch)
event_data_batch = await producer.create_batch()
event_data_batch.add(event_data)
if len(event_data_batch) > 0:
await producer.send_batch(event_data_batch)
start_time = time.time()
asyncio.run(run())
print("Send messages in {} seconds.".format(time.time() - start_time))
|
