Implement from_log_record for Trace type data in live metrics (#39922)

Author: lzchen

sdk/monitor/azure-monitor-opentelemetry-exporter/CHANGELOG.md

@@ -13,6 +13,9 @@
 
 ### Bugs Fixed
 
+- Implement `from_log_record` for `Trace` data types in live metrics
+  ([#39922](https://github.com/Azure/azure-sdk-for-python/pull/39922))
+
 ### Other Changes
 
 ## 1.0.0b34 (2025-02-26)

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

@@ -197,6 +197,14 @@ def _TraceData(log_record: LogRecord):
             custom_dimensions=log_record.attributes,
         )
 
+    @staticmethod
+    @no_type_check
+    def _from_log_record(log_record: LogRecord):
+        return _TraceData(
+            message=str(log_record.body),
+            custom_dimensions=log_record.attributes,
+        )
+
 
 def _get_field_names(data_type: type):
     field_map = {}

sdk/monitor/azure-monitor-opentelemetry-exporter/tests/quickpulse/test_types.py

@@ -20,9 +20,67 @@
     _DependencyData,
     _ExceptionData,
     _RequestData,
+    _TelemetryData,
 )
 
 
+
+class TestTelemetryData(unittest.TestCase):
+
+    @patch("azure.monitor.opentelemetry.exporter._quickpulse._types._RequestData")
+    def test_from_span_server_kind_server(self, fn_mock):
+        span = Mock()
+        span.kind = SpanKind.SERVER
+        data_mock = Mock()
+        fn_mock._from_span.return_value = data_mock
+        result = _TelemetryData._from_span(span)
+        self.assertEqual(result, data_mock)
+        fn_mock._from_span.assert_called_once_with(span)
+
+    @patch("azure.monitor.opentelemetry.exporter._quickpulse._types._RequestData")
+    def test_from_span_consumer_kind_(self, fn_mock):
+        span = Mock()
+        span.kind = SpanKind.CONSUMER
+        data_mock = Mock()
+        fn_mock._from_span.return_value = data_mock
+        result = _TelemetryData._from_span(span)
+        self.assertEqual(result, data_mock)
+        fn_mock._from_span.assert_called_once_with(span)
+
+    @patch("azure.monitor.opentelemetry.exporter._quickpulse._types._DependencyData")
+    def test_from_span_dependency_kind(self, fn_mock):
+        span = Mock()
+        span.kind = SpanKind.CLIENT
+        data_mock = Mock()
+        fn_mock._from_span.return_value = data_mock
+        result = _TelemetryData._from_span(span)
+        self.assertEqual(result, data_mock)
+        fn_mock._from_span.assert_called_once_with(span)
+
+    @patch("azure.monitor.opentelemetry.exporter._quickpulse._types._ExceptionData")
+    def test_from_log_record_with_exception(self, fn_mock):
+        log_record = Mock()
+        log_record.attributes = {
+            SpanAttributes.EXCEPTION_TYPE: "SomeException",
+            SpanAttributes.EXCEPTION_MESSAGE: "An error occurred"
+        }
+        data_mock = Mock()
+        fn_mock._from_log_record.return_value = data_mock
+        result = _TelemetryData._from_log_record(log_record)
+        self.assertEqual(result, data_mock)
+        fn_mock._from_log_record.assert_called_once_with(log_record)
+
+    @patch("azure.monitor.opentelemetry.exporter._quickpulse._types._TraceData")
+    def test_from_log_record_without_exception(self, fn_mock):
+        log_record = Mock()
+        log_record.attributes = {}
+        data_mock = Mock()
+        fn_mock._from_log_record.return_value = data_mock
+        result = _TelemetryData._from_log_record(log_record)
+        self.assertEqual(result, data_mock)
+        fn_mock._from_log_record.assert_called_once_with(log_record)
+
+
 class TestRequestData(unittest.TestCase):
 
     @patch("azure.monitor.opentelemetry.exporter.export.trace._utils._get_url_for_http_request")