diff --git a/sdk/eventgrid/azure-eventgrid/samples/namespace/async_samples/sample_consume_process_events_async.py b/sdk/eventgrid/azure-eventgrid/samples/namespace/async_samples/sample_consume_process_events_async.py index a44dfe010460..9b3d42b051b5 100644 --- a/sdk/eventgrid/azure-eventgrid/samples/namespace/async_samples/sample_consume_process_events_async.py +++ b/sdk/eventgrid/azure-eventgrid/samples/namespace/async_samples/sample_consume_process_events_async.py @@ -47,89 +47,90 @@ async def run(): cloud_event_ack = CloudEvent(data="acknowledge", source="https://example.com", type="example") cloud_event_renew = CloudEvent(data="renew", source="https://example.com", type="example") - # Send Cloud Events - await publisher.send( - [ - cloud_event_reject, - cloud_event_release, - cloud_event_ack, - cloud_event_renew, - ] - ) - - # Receive Published Cloud Events - try: - receive_results = await client.receive( - max_events=10, - max_wait_time=10, + async with publisher, client: + # Send Cloud Events + await publisher.send( + [ + cloud_event_reject, + cloud_event_release, + cloud_event_ack, + cloud_event_renew, + ] ) - except HttpResponseError: - raise - - # Iterate through the results and collect the lock tokens for events we want to release/acknowledge/reject/renew: - - release_events = [] - acknowledge_events = [] - reject_events = [] - renew_events = [] - - for detail in receive_results: - data = detail.event.data - broker_properties = detail.broker_properties - if data == "release": - release_events.append(broker_properties.lock_token) - elif data == "acknowledge": - acknowledge_events.append(broker_properties.lock_token) - elif data == "renew": - renew_events.append(broker_properties.lock_token) - else: - reject_events.append(broker_properties.lock_token) - - # Release/Acknowledge/Reject/Renew events - - if len(release_events) > 0: - try: - release_result = await client.release( - lock_tokens=release_events, - ) - except HttpResponseError: - raise - - for succeeded_lock_token in release_result.succeeded_lock_tokens: - print(f"Succeeded Lock Token:{succeeded_lock_token}") - - if len(acknowledge_events) > 0: - try: - ack_result = await client.acknowledge( - lock_tokens=acknowledge_events, - ) - except HttpResponseError: - raise - - for succeeded_lock_token in ack_result.succeeded_lock_tokens: - print(f"Succeeded Lock Token:{succeeded_lock_token}") - - if len(reject_events) > 0: - try: - reject_result = await client.reject( - lock_tokens=reject_events, - ) - except HttpResponseError: - raise - - for succeeded_lock_token in reject_result.succeeded_lock_tokens: - print(f"Succeeded Lock Token:{succeeded_lock_token}") - if len(renew_events) > 0: + # Receive Published Cloud Events try: - renew_result = await client.renew_locks( - lock_tokens=renew_events, + receive_results = await client.receive( + max_events=10, + max_wait_time=10, ) except HttpResponseError: raise - for succeeded_lock_token in renew_result.succeeded_lock_tokens: - print(f"Succeeded Lock Token:{succeeded_lock_token}") + # Iterate through the results and collect the lock tokens for events we want to release/acknowledge/reject/renew: + + release_events = [] + acknowledge_events = [] + reject_events = [] + renew_events = [] + + for detail in receive_results: + data = detail.event.data + broker_properties = detail.broker_properties + if data == "release": + release_events.append(broker_properties.lock_token) + elif data == "acknowledge": + acknowledge_events.append(broker_properties.lock_token) + elif data == "renew": + renew_events.append(broker_properties.lock_token) + else: + reject_events.append(broker_properties.lock_token) + + # Release/Acknowledge/Reject/Renew events + + if len(release_events) > 0: + try: + release_result = await client.release( + lock_tokens=release_events, + ) + except HttpResponseError: + raise + + for succeeded_lock_token in release_result.succeeded_lock_tokens: + print(f"Succeeded Lock Token:{succeeded_lock_token}") + + if len(acknowledge_events) > 0: + try: + ack_result = await client.acknowledge( + lock_tokens=acknowledge_events, + ) + except HttpResponseError: + raise + + for succeeded_lock_token in ack_result.succeeded_lock_tokens: + print(f"Succeeded Lock Token:{succeeded_lock_token}") + + if len(reject_events) > 0: + try: + reject_result = await client.reject( + lock_tokens=reject_events, + ) + except HttpResponseError: + raise + + for succeeded_lock_token in reject_result.succeeded_lock_tokens: + print(f"Succeeded Lock Token:{succeeded_lock_token}") + + if len(renew_events) > 0: + try: + renew_result = await client.renew_locks( + lock_tokens=renew_events, + ) + except HttpResponseError: + raise + + for succeeded_lock_token in renew_result.succeeded_lock_tokens: + print(f"Succeeded Lock Token:{succeeded_lock_token}") if __name__ == "__main__": diff --git a/sdk/eventgrid/azure-eventgrid/samples/namespace/async_samples/sample_publish_cloud_event_async.py b/sdk/eventgrid/azure-eventgrid/samples/namespace/async_samples/sample_publish_cloud_event_async.py index bcb88e75a8a4..48f5b46ea7e6 100644 --- a/sdk/eventgrid/azure-eventgrid/samples/namespace/async_samples/sample_publish_cloud_event_async.py +++ b/sdk/eventgrid/azure-eventgrid/samples/namespace/async_samples/sample_publish_cloud_event_async.py @@ -30,18 +30,19 @@ async def publish(): - client.send( - [ - { - "type": "Contoso.Items.ItemReceived", - "source": "/contoso/items", - "data": {"itemSku": "Contoso Item SKU #1"}, - "subject": "Door1", - "specversion": "1.0", - "id": "randomclouduuid11", - } - ] - ) + async with client: + await client.send( + [ + { + "type": "Contoso.Items.ItemReceived", + "source": "/contoso/items", + "data": {"itemSku": "Contoso Item SKU #1"}, + "subject": "Door1", + "specversion": "1.0", + "id": "randomclouduuid11", + } + ] + ) if __name__ == "__main__": diff --git a/sdk/eventgrid/azure-eventgrid/samples/namespace/async_samples/sample_publish_cncf_events_async.py b/sdk/eventgrid/azure-eventgrid/samples/namespace/async_samples/sample_publish_cncf_events_async.py index 410d90eead20..d8b641ddf627 100644 --- a/sdk/eventgrid/azure-eventgrid/samples/namespace/async_samples/sample_publish_cncf_events_async.py +++ b/sdk/eventgrid/azure-eventgrid/samples/namespace/async_samples/sample_publish_cncf_events_async.py @@ -30,14 +30,15 @@ async def publish(): - await client.send( - [ - CloudEvent( - attributes={"type": "cloudevent", "source": "/cncf/cloud/event/1.0", "subject": "testing-cncf-event"}, - data=b"This is a cncf cloud event.", - ) - ] - ) + async with client: + await client.send( + [ + CloudEvent( + attributes={"type": "cloudevent", "source": "/cncf/cloud/event/1.0", "subject": "testing-cncf-event"}, + data=b"This is a cncf cloud event.", + ) + ] + ) if __name__ == "__main__":