Create eventhub samples (Azure#24133)

lzchen · web-flow · commit ec7675208ad8 · 2022-04-22T14:59:50.000-07:00
diff --git a/sdk/monitor/azure-monitor-opentelemetry-exporter/azure/monitor/opentelemetry/exporter/_utils.py b/sdk/monitor/azure-monitor-opentelemetry-exporter/azure/monitor/opentelemetry/exporter/_utils.py
@@ -58,7 +58,7 @@ class PeriodicTask(threading.Thread):
     """
 
     def __init__(self, interval, function, *args, **kwargs):
-        super().__init__(name=kwargs.get('name'))
+        super().__init__(name=kwargs.pop('name', None))
         self.interval = interval
         self.function = function
         self.args = args or []
diff --git a/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/README.md b/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/README.md
@@ -14,6 +14,8 @@ These code samples show common champion scenario operations with the AzureMonito
 * Azure Service Bus Receive: [sample_servicebus_receive.py](https://github.com/Azure/azure-sdk-for-python/blob/main/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/sample_servicebus_receive.py)
 * Azure Storage Blob Create Container: [sample_storage_blob.py](https://github.com/Azure/azure-sdk-for-python/blob/main/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/sample_storage_blob.py)
 * Azure CosmosDb Create Db/Container: [sample_cosmos.py](https://github.com/Azure/azure-sdk-for-python/blob/main/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/sample_cosmos.py)
+* Azure EventHub Send EventData: [sample_event_hub.py](https://github.com/Azure/azure-sdk-for-python/blob/main/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/sample_event_hub.py)
+* Azure EventHub Blob Storage Checkpoint Store: [sample_blob_checkpoint.py](https://github.com/Azure/azure-sdk-for-python/blob/main/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/sample_blob_checkpoint.py)
 * Client: [sample_client.py](https://github.com/Azure/azure-sdk-for-python/blob/main/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/sample_client.py)
 * Event: [sample_span_event.py](https://github.com/Azure/azure-sdk-for-python/blob/main/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/sample_span_event.py)
 * Jaeger: [sample_jaeger.py](https://github.com/Azure/azure-sdk-for-python/blob/main/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/sample_jaeger.py)
@@ -199,14 +201,53 @@ The following sample assumes that you have setup Azure CosmosDb [account](https:
 * Run the sample
 
 ```sh
-$ # azure-storage-blob library
+$ # azure-cosmos library
 $ pip install azure-cosmos
 $ # azure sdk core tracing library for opentelemetry
 $ pip install azure-core-tracing-opentelemetry
 $ # from this directory
 $ python sample_cosmos.py
 ```
 
+### Azure EventHub Send EventData
+
+The following sample assumes that you have setup an Azure EventHubs namespace and [EventHub](https://docs.microsoft.com/azure/event-hubs/event-hubs-create).
+
+* Update `APPLICATIONINSIGHTS_CONNECTION_STRING` environment variable
+* Update `EVENT_HUB_CONN_STR` environment variable
+* Update `EVENT_HUB_NAME` environment variable
+
+* Run the sample
+
+```sh
+$ # azure-eventhub library
+$ pip install azure-eventhub
+$ # azure sdk core tracing library for opentelemetry
+$ pip install azure-core-tracing-opentelemetry
+$ # from this directory
+$ python sample_event_hub.py
+```
+
+### Azure EventHub Blob Storage Checkpoint Store
+
+The following sample assumes that you have setup an Azure EventHubs namespace, [EventHub](https://docs.microsoft.com/azure/event-hubs/event-hubs-create) and Azure Blob [storage](https://docs.microsoft.com/azure/storage/blobs/storage-quickstart-blobs-portal).
+
+* Update `APPLICATIONINSIGHTS_CONNECTION_STRING` environment variable
+* Update `EVENT_HUB_CONN_STR` environment variable
+* Update `EVENT_HUB_NAME` environment variable
+* Update `AZURE_STORAGE_CONN_STR` environment variable
+
+* Run the sample
+
+```sh
+$ # azure-eventhub-checkpointstoreblob library
+$ pip install azure-eventhub-checkpointstoreblob
+$ # azure sdk core tracing library for opentelemetry
+$ pip install azure-core-tracing-opentelemetry
+$ # from this directory
+$ python sample_blob_checkpoint.py
+```
+
 ## Explore the data
 
 After running the applications, data would be available in [Azure](
diff --git a/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/sample_blob_checkpoint.py b/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/sample_blob_checkpoint.py
@@ -0,0 +1,66 @@
+"""
+Examples to show usage of the azure-core-tracing-opentelemetry
+with the eventhub checkpoint storage blob SDK and exporting to
+Azure monitor backend. This example traces calls for sending
+checkpoints via the checkpoint storage blob sdk. The telemetry
+will be collected automatically and sent to Application Insights
+via the AzureMonitorTraceExporter
+"""
+
+import os
+
+# Declare OpenTelemetry as enabled tracing plugin for Azure SDKs
+from azure.core.settings import settings
+from azure.core.tracing.ext.opentelemetry_span import OpenTelemetrySpan
+
+settings.tracing_implementation = OpenTelemetrySpan
+
+# Regular open telemetry usage from here, see https://github.com/open-telemetry/opentelemetry-python
+# for details
+from opentelemetry import trace
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+
+trace.set_tracer_provider(TracerProvider())
+tracer = trace.get_tracer(__name__)
+
+# azure monitor trace exporter to send telemetry to appinsights
+from azure.monitor.opentelemetry.exporter import AzureMonitorTraceExporter
+span_processor = BatchSpanProcessor(
+    AzureMonitorTraceExporter.from_connection_string(
+        os.environ["APPLICATIONINSIGHTS_CONNECTION_STRING"]
+    )
+)
+trace.get_tracer_provider().add_span_processor(span_processor)
+
+# Example with EventHub SDKs
+from azure.eventhub import EventHubConsumerClient
+from azure.eventhub.extensions.checkpointstoreblob import BlobCheckpointStore
+
+CONNECTION_STR = os.environ["EVENT_HUB_CONN_STR"]
+EVENTHUB_NAME = os.environ['EVENT_HUB_NAME']
+STORAGE_CONNECTION_STR = os.environ["AZURE_STORAGE_CONN_STR"]
+BLOB_CONTAINER_NAME = "your-blob-container-name"  # Please make sure the blob container resource exists.
+
+def on_event(partition_context, event):
+    # Put your code here.
+    # Avoid time-consuming operations.
+    print(event)
+    partition_context.update_checkpoint(event)
+
+checkpoint_store = BlobCheckpointStore.from_connection_string(
+    STORAGE_CONNECTION_STR,
+    container_name=BLOB_CONTAINER_NAME,
+)
+client = EventHubConsumerClient.from_connection_string(
+    CONNECTION_STR,
+    consumer_group='$Default',
+    eventhub_name=EVENTHUB_NAME,
+    checkpoint_store=checkpoint_store
+)
+
+with tracer.start_as_current_span(name="MyEventHub"):
+    try:
+        client.receive(on_event)
+    except KeyboardInterrupt:
+        client.close()
diff --git a/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/sample_cosmos.py b/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/sample_cosmos.py
@@ -1,9 +1,9 @@
 """
 Examples to show usage of the azure-core-tracing-opentelemetry
-with the storage SDK and exporting to Azure monitor backend.
-This example traces calls for creating a container using storage SDK.
-The telemetry will be collected automatically and sent to Application
-Insights via the AzureMonitorTraceExporter
+with the CosmosDb SDK and exporting to Azure monitor backend.
+This example traces calls for creating a database and container using
+CosmosDb SDK. The telemetry will be collected automatically and sent
+to Application Insights via the AzureMonitorTraceExporter
 """
 
 import os
@@ -40,15 +40,18 @@
 client = CosmosClient(url, key)
 
 database_name = "testDatabase"
-try:
-    database = client.create_database(id=database_name)  # Call will be traced
-except exceptions.CosmosResourceExistsError:
-    database = client.get_database_client(database=database_name)
+
+with tracer.start_as_current_span(name="CreateDatabase"):
+    try:
+        database = client.create_database(id=database_name)  # Call will be traced
+    except exceptions.CosmosResourceExistsError:
+        database = client.get_database_client(database=database_name)
 
 container_name = "products"
-try:
-    container = database.create_container(
-        id=container_name, partition_key=PartitionKey(path="/productName")  # Call will be traced
-    )
-except exceptions.CosmosResourceExistsError:
-    container = database.get_container_client(container_name)
+with tracer.start_as_current_span(name="CreateContainer"):
+    try:
+        container = database.create_container(
+            id=container_name, partition_key=PartitionKey(path="/productName")  # Call will be traced
+        )
+    except exceptions.CosmosResourceExistsError:
+        container = database.get_container_client(container_name)
diff --git a/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/sample_event_hub.py b/sdk/monitor/azure-monitor-opentelemetry-exporter/samples/traces/sample_event_hub.py
@@ -0,0 +1,50 @@
+"""
+Examples to show usage of the azure-core-tracing-opentelemetry
+with the event hub SDK and exporting to Azure monitor backend.
+This example traces calls for sending event data using event hub SDK.
+The telemetry will be collected automatically and sent to Application
+Insights via the AzureMonitorTraceExporter
+"""
+
+import os
+
+# Declare OpenTelemetry as enabled tracing plugin for Azure SDKs
+from azure.core.settings import settings
+from azure.core.tracing.ext.opentelemetry_span import OpenTelemetrySpan
+
+settings.tracing_implementation = OpenTelemetrySpan
+
+# Regular open telemetry usage from here, see https://github.com/open-telemetry/opentelemetry-python
+# for details
+from opentelemetry import trace
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor
+
+trace.set_tracer_provider(TracerProvider())
+tracer = trace.get_tracer(__name__)
+
+# azure monitor trace exporter to send telemetry to appinsights
+from azure.monitor.opentelemetry.exporter import AzureMonitorTraceExporter
+span_processor = BatchSpanProcessor(
+    AzureMonitorTraceExporter.from_connection_string(
+        os.environ["APPLICATIONINSIGHTS_CONNECTION_STRING"]
+    )
+)
+trace.get_tracer_provider().add_span_processor(span_processor)
+
+# Example with EventHub SDKs
+from azure.eventhub import EventHubProducerClient, EventData
+
+CONNECTION_STR = os.environ['EVENT_HUB_CONN_STR']
+EVENTHUB_NAME = os.environ['EVENT_HUB_NAME']
+
+producer = EventHubProducerClient.from_connection_string(
+    conn_str=CONNECTION_STR,
+    eventhub_name=EVENTHUB_NAME
+)
+
+with tracer.start_as_current_span(name="MyEventHub"):
+    with producer:
+        event_data_batch = producer.create_batch()
+        event_data_batch.add(EventData('Single message'))
+        producer.send_batch(event_data_batch)
