Changes to accomodated breaking log changes from Otel

Author: rads-1996

sdk/monitor/azure-monitor-opentelemetry-exporter/azure/monitor/opentelemetry/exporter/_performance_counters/_manager.py

@@ -9,7 +9,7 @@
 
 from opentelemetry import metrics
 from opentelemetry.metrics import CallbackOptions, Observation
-from opentelemetry.sdk._logs import LogData
+from opentelemetry.sdk._logs import ReadableLogRecord
 from opentelemetry.sdk.trace import ReadableSpan
 from opentelemetry.semconv.attributes.exception_attributes import (
     EXCEPTION_MESSAGE,
@@ -632,13 +632,13 @@ def _record_span(self, span: ReadableSpan) -> None:
         except Exception:  # pylint: disable=broad-except
             _logger.exception("Exception occurred while recording span.")  # pylint: disable=C4769
 
-    def _record_log_record(self, log_data: LogData) -> None:
+    def _record_log_record(self, readable_log_record: ReadableLogRecord) -> None:
         try:
             # pylint: disable=global-statement
             global _EXCEPTIONS_COUNT
-            if log_data.log_record:
+            if readable_log_record.log_record:
                 exc_type = None
-                log_record = log_data.log_record
+                log_record = readable_log_record.log_record
                 if log_record.attributes:
                     exc_type = log_record.attributes.get(EXCEPTION_TYPE)
                     exc_message = log_record.attributes.get(EXCEPTION_MESSAGE)

sdk/monitor/azure-monitor-opentelemetry-exporter/azure/monitor/opentelemetry/exporter/_performance_counters/_processor.py

@@ -1,7 +1,7 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License.
 
-from opentelemetry.sdk._logs import LogData, LogRecordProcessor
+from opentelemetry.sdk._logs import ReadableLogRecord, LogRecordProcessor
 from opentelemetry.sdk.trace import ReadableSpan, SpanProcessor
 
 from azure.monitor.opentelemetry.exporter._performance_counters._manager import _PerformanceCountersManager
@@ -13,18 +13,18 @@ def __init__(self):
         super().__init__()
         self.call_on_emit = hasattr(super(), 'on_emit')
 
-    def on_emit(self, log_data: LogData) -> None:  # type: ignore
+    def on_emit(self, readable_log_record: ReadableLogRecord) -> None:  # type: ignore
         pcm = _PerformanceCountersManager()
         if pcm:
-            pcm._record_log_record(log_data)
+            pcm._record_log_record(readable_log_record)
         if self.call_on_emit:
-            super().on_emit(log_data)  # type: ignore[safe-super]
+            super().on_emit(readable_log_record)  # type: ignore[safe-super]
         else:
             # this method was removed in opentelemetry-sdk and replaced with on_emit
-            super().emit(log_data)  # type: ignore[safe-super,misc] # pylint: disable=no-member
+            super().emit(readable_log_record)  # type: ignore[safe-super,misc] # pylint: disable=no-member
 
-    def emit(self, log_data: LogData) -> None:
-        self.on_emit(log_data)
+    def emit(self, readable_log_record: ReadableLogRecord) -> None:
+        self.on_emit(readable_log_record)
 
     def shutdown(self):
         pass

sdk/monitor/azure-monitor-opentelemetry-exporter/azure/monitor/opentelemetry/exporter/_quickpulse/_manager.py

@@ -9,7 +9,7 @@
 
 import psutil
 
-from opentelemetry.sdk._logs import LogData
+from opentelemetry.sdk._logs import ReadableLogRecord
 from opentelemetry.sdk.metrics import MeterProvider, Meter
 from opentelemetry.sdk.resources import Resource
 from opentelemetry.sdk.trace import ReadableSpan
@@ -353,7 +353,7 @@ def _record_span(self, span: ReadableSpan) -> None:
         except Exception as e:  # pylint: disable=broad-except
             _logger.exception("Exception occurred while recording span: %s", e)  # pylint: disable=C4769
 
-    def _record_log_record(self, log_data: LogData) -> None:
+    def _record_log_record(self, readable_log_record: ReadableLogRecord) -> None:
         # Only record if in post state and manager is initialized
         if not (_is_post_state() and self.is_initialized()):
             return
@@ -364,9 +364,9 @@ def _record_log_record(self, log_data: LogData) -> None:
             return
 
         try:
-            if log_data.log_record:
+            if readable_log_record.log_record:
                 exc_type = None
-                log_record = log_data.log_record
+                log_record = readable_log_record.log_record
                 if log_record.attributes:
                     exc_type = log_record.attributes.get(SpanAttributes.EXCEPTION_TYPE)
                     exc_message = log_record.attributes.get(SpanAttributes.EXCEPTION_MESSAGE)

sdk/monitor/azure-monitor-opentelemetry-exporter/azure/monitor/opentelemetry/exporter/_quickpulse/_processor.py

@@ -1,7 +1,7 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License.
 
-from opentelemetry.sdk._logs import LogData, LogRecordProcessor
+from opentelemetry.sdk._logs import ReadableLogRecord, LogRecordProcessor
 from opentelemetry.sdk.trace import ReadableSpan, SpanProcessor
 
 from azure.monitor.opentelemetry.exporter._quickpulse._state import get_quickpulse_manager
@@ -13,18 +13,18 @@ def __init__(self):
         super().__init__()
         self.call_on_emit = hasattr(super(), 'on_emit')
 
-    def on_emit(self, log_data: LogData) -> None:  # type: ignore
+    def on_emit(self, readable_log_record: ReadableLogRecord) -> None:  # type: ignore
         qpm = get_quickpulse_manager()
         if qpm:
-            qpm._record_log_record(log_data)
+            qpm._record_log_record(readable_log_record)
         if self.call_on_emit:
-            super().on_emit(log_data)  # type: ignore[safe-super]
+            super().on_emit(readable_log_record)  # type: ignore[safe-super]
         else:
             # this method was removed in opentelemetry-sdk and replaced with on_emit
-            super().emit(log_data)  # type: ignore[safe-super,misc] # pylint: disable=no-member
+            super().emit(readable_log_record)  # type: ignore[safe-super,misc] # pylint: disable=no-member
 
-    def emit(self, log_data: LogData) -> None:
-        self.on_emit(log_data)
+    def emit(self, readable_log_record: ReadableLogRecord) -> None:
+        self.on_emit(readable_log_record)
 
     def shutdown(self):
         pass

sdk/monitor/azure-monitor-opentelemetry-exporter/azure/monitor/opentelemetry/exporter/_quickpulse/_types.py

@@ -177,7 +177,7 @@ def _from_log_record(log_record: LogRecord):
 
     @staticmethod
     @no_type_check
-    def _from_span_event(span_event: Event):
+    def _from_span_event(span_event: LogRecord.event_name):
         return _ExceptionData(
             message=str(span_event.attributes.get(SpanAttributes.EXCEPTION_MESSAGE, "")),
             stack_trace=str(span_event.attributes.get(SpanAttributes.EXCEPTION_STACKTRACE, "")),

sdk/monitor/azure-monitor-opentelemetry-exporter/azure/monitor/opentelemetry/exporter/export/logs/_exporter.py

@@ -11,8 +11,8 @@
     EXCEPTION_STACKTRACE,
     EXCEPTION_TYPE,
 )
-from opentelemetry.sdk._logs import LogData
-from opentelemetry.sdk._logs.export import LogExporter, LogExportResult
+from opentelemetry.sdk._logs import ReadableLogRecord
+from opentelemetry.sdk._logs.export import LogRecordExporter, LogRecordExportResult
 
 from azure.monitor.opentelemetry.exporter import _utils
 from azure.monitor.opentelemetry.exporter._constants import (
@@ -53,16 +53,16 @@
 __all__ = ["AzureMonitorLogExporter"]
 
 
-class AzureMonitorLogExporter(BaseExporter, LogExporter):
+class AzureMonitorLogExporter(BaseExporter, LogRecordExporter):
     """Azure Monitor Log exporter for OpenTelemetry."""
 
-    def export(self, batch: Sequence[LogData], **kwargs: Any) -> LogExportResult:  # pylint: disable=unused-argument
+    def export(self, batch: Sequence[ReadableLogRecord], **kwargs: Any) -> LogRecordExportResult:  # pylint: disable=unused-argument
         """Export log data.
 
-        :param batch: OpenTelemetry LogData(s) to export.
-        :type batch: ~typing.Sequence[~opentelemetry._logs.LogData]
+        :param batch: OpenTelemetry ReadableLogRecord(s) to export.
+        :type batch: ~typing.Sequence[~opentelemetry._logs.ReadableLogRecord]
         :return: The result of the export.
-        :rtype: ~opentelemetry.sdk._logs.export.LogData
+        :rtype: ~opentelemetry.sdk._logs.export.ReadableLogRecord
         """
         envelopes = [self._log_to_envelope(log) for log in batch]
         try:
@@ -81,8 +81,8 @@ def shutdown(self) -> None:
         if self.storage:
             self.storage.close()
 
-    def _log_to_envelope(self, log_data: LogData) -> TelemetryItem:
-        envelope = _convert_log_to_envelope(log_data)
+    def _log_to_envelope(self, readable_log_record: ReadableLogRecord) -> TelemetryItem:
+        envelope = _convert_log_to_envelope(readable_log_record)
         envelope.instrumentation_key = self._instrumentation_key
         return envelope
 
@@ -106,8 +106,8 @@ def from_connection_string(cls, conn_str: str, **kwargs: Any) -> "AzureMonitorLo
         return cls(connection_string=conn_str, **kwargs)
 
 
-def _log_data_is_event(log_data: LogData) -> bool:
-    log_record = log_data.log_record
+def _log_data_is_event(readable_log_record: ReadableLogRecord) -> bool:
+    log_record = readable_log_record.log_record
     is_event = None
     if log_record.attributes:
         is_event = log_record.attributes.get(_MICROSOFT_CUSTOM_EVENT_NAME) or \
@@ -116,8 +116,8 @@ def _log_data_is_event(log_data: LogData) -> bool:
 
 
 # pylint: disable=protected-access
-def _convert_log_to_envelope(log_data: LogData) -> TelemetryItem:
-    log_record = log_data.log_record
+def _convert_log_to_envelope(readable_log_record: ReadableLogRecord) -> TelemetryItem:
+    log_record = readable_log_record.log_record
     time_stamp = log_record.timestamp if log_record.timestamp is not None else log_record.observed_timestamp
     envelope = _utils._create_telemetry_item(time_stamp)
     envelope.tags.update(_utils._populate_part_a_fields(log_record.resource))  # type: ignore
@@ -168,7 +168,7 @@ def _convert_log_to_envelope(log_data: LogData) -> TelemetryItem:
             exceptions=[exc_details],
         )
         envelope.data = MonitorBase(base_data=data, base_type="ExceptionData")
-    elif _log_data_is_event(log_data):  # Event telemetry
+    elif _log_data_is_event(readable_log_record):  # Event telemetry
         _set_statsbeat_custom_events_feature()
         envelope.name = "Microsoft.ApplicationInsights.Event"
         event_name = ""
@@ -198,10 +198,10 @@ def _convert_log_to_envelope(log_data: LogData) -> TelemetryItem:
     return envelope
 
 
-def _get_log_export_result(result: ExportResult) -> LogExportResult:
+def _get_log_export_result(result: ExportResult) -> LogRecordExportResult:
     if result == ExportResult.SUCCESS:
-        return LogExportResult.SUCCESS
-    return LogExportResult.FAILURE
+        return LogRecordExportResult.SUCCESS
+    return LogRecordExportResult.FAILURE
 
 
 # pylint: disable=line-too-long