refactor: reduce otel java usage

Author: fractalwrench

embrace-android-core/src/main/kotlin/io/embrace/android/embracesdk/internal/payload/EnvelopeExt.kt

@@ -2,7 +2,6 @@ package io.embrace.android.embracesdk.internal.payload
 
 import io.embrace.android.embracesdk.internal.otel.schema.EmbType
 import io.embrace.android.embracesdk.internal.otel.sdk.findAttributeValue
-import io.embrace.android.embracesdk.internal.otel.sdk.hasEmbraceAttribute
 import io.embrace.android.embracesdk.internal.otel.spans.hasEmbraceAttribute
 import io.opentelemetry.semconv.incubating.SessionIncubatingAttributes
 

embrace-android-core/src/test/java/io/embrace/android/embracesdk/internal/injection/OpenTelemetryModuleImplTest.kt

@@ -4,9 +4,11 @@ import io.embrace.android.embracesdk.internal.otel.logs.LogSinkImpl
 import io.embrace.android.embracesdk.internal.otel.spans.EmbraceSpanService
 import io.embrace.android.embracesdk.internal.otel.spans.SpanSinkImpl
 import io.embrace.android.embracesdk.internal.spans.CurrentSessionSpanImpl
+import io.embrace.opentelemetry.kotlin.ExperimentalApi
 import org.junit.Assert.assertTrue
 import org.junit.Test
 
+@OptIn(ExperimentalApi::class)
 internal class OpenTelemetryModuleImplTest {
     @Test
     fun testInitModuleImplDefaults() {

embrace-android-core/src/test/java/io/embrace/android/embracesdk/internal/logs/EmbraceLogRecordDataTest.kt

@@ -36,12 +36,12 @@ import io.embrace.android.embracesdk.internal.otel.payload.toEmbracePayload
 import io.embrace.android.embracesdk.internal.otel.schema.EmbType
 import io.embrace.android.embracesdk.internal.otel.sdk.findAttributeValue
 import io.embrace.android.embracesdk.internal.otel.sdk.id.OtelIds
-import io.embrace.android.embracesdk.internal.otel.sdk.toEmbraceSpanData
 import io.embrace.android.embracesdk.internal.otel.spans.EmbraceSpanData
 import io.embrace.android.embracesdk.internal.payload.Envelope
 import io.embrace.android.embracesdk.internal.payload.NativeCrashData
 import io.embrace.android.embracesdk.internal.payload.SessionPayload
 import io.embrace.android.embracesdk.internal.payload.getSessionSpan
+import io.embrace.android.embracesdk.internal.toEmbraceSpanData
 import io.embrace.android.embracesdk.spans.ErrorCode
 import io.embrace.opentelemetry.kotlin.ExperimentalApi
 import org.junit.Assert.assertEquals

embrace-android-core/src/test/java/io/embrace/android/embracesdk/internal/resurrection/PayloadResurrectionServiceImplTest.kt

@@ -1,9 +1,8 @@
 package io.embrace.android.embracesdk.internal.otel.impl
 
 import io.embrace.opentelemetry.kotlin.ExperimentalApi
-import io.embrace.opentelemetry.kotlin.aliases.OtelJavaSpanContext
 import io.embrace.opentelemetry.kotlin.attributes.AttributeContainer
-import io.embrace.opentelemetry.kotlin.k2j.tracing.SpanContextAdapter
+import io.embrace.opentelemetry.kotlin.k2j.tracing.model.invalid
 import io.embrace.opentelemetry.kotlin.tracing.StatusCode
 import io.embrace.opentelemetry.kotlin.tracing.model.Link
 import io.embrace.opentelemetry.kotlin.tracing.model.Span
@@ -21,7 +20,7 @@ internal class EmbInvalidSpan : Span {
 
     override var name: String = ""
     override val parent: SpanContext? = null
-    override val spanContext: SpanContext = SpanContextAdapter(OtelJavaSpanContext.getInvalid())
+    override val spanContext: SpanContext = SpanContext.invalid()
     override val spanKind: SpanKind = SpanKind.INTERNAL
     override val startTimestamp: Long = 0
     override var status: StatusCode = StatusCode.Unset

embrace-android-otel/src/main/kotlin/io/embrace/android/embracesdk/internal/otel/attrs/EmbraceAttributeKeyExt.kt

@@ -2,22 +2,23 @@ package io.embrace.android.embracesdk.internal.otel.impl
 
 import io.embrace.android.embracesdk.internal.clock.nanosToMillis
 import io.embrace.android.embracesdk.internal.otel.spans.EmbraceSdkSpan
-import io.embrace.android.embracesdk.internal.otel.toOtelKotlin
 import io.embrace.android.embracesdk.internal.payload.Attribute
 import io.embrace.opentelemetry.kotlin.Clock
 import io.embrace.opentelemetry.kotlin.ExperimentalApi
-import io.embrace.opentelemetry.kotlin.aliases.OtelJavaSpanContext
-import io.embrace.opentelemetry.kotlin.aliases.OtelJavaTraceFlags
-import io.embrace.opentelemetry.kotlin.aliases.OtelJavaTraceState
 import io.embrace.opentelemetry.kotlin.attributes.AttributeContainer
 import io.embrace.opentelemetry.kotlin.k2j.tracing.SpanContextAdapter
 import io.embrace.opentelemetry.kotlin.k2j.tracing.convertToOtelJava
+import io.embrace.opentelemetry.kotlin.k2j.tracing.model.create
+import io.embrace.opentelemetry.kotlin.k2j.tracing.model.default
+import io.embrace.opentelemetry.kotlin.k2j.tracing.model.invalid
 import io.embrace.opentelemetry.kotlin.tracing.StatusCode
 import io.embrace.opentelemetry.kotlin.tracing.model.Link
 import io.embrace.opentelemetry.kotlin.tracing.model.Span
 import io.embrace.opentelemetry.kotlin.tracing.model.SpanContext
 import io.embrace.opentelemetry.kotlin.tracing.model.SpanEvent
 import io.embrace.opentelemetry.kotlin.tracing.model.SpanKind
+import io.embrace.opentelemetry.kotlin.tracing.model.TraceFlags
+import io.embrace.opentelemetry.kotlin.tracing.model.TraceState
 
 @OptIn(ExperimentalApi::class)
 class EmbSpan(
@@ -66,7 +67,7 @@ class EmbSpan(
     }
 
     override val spanContext: SpanContext
-        get() = SpanContextAdapter(impl.spanContext ?: OtelJavaSpanContext.getInvalid())
+        get() = impl.spanContext?.let(::SpanContextAdapter) ?: SpanContext.invalid()
 
     override fun isRecording(): Boolean = impl.isRecording
 
@@ -129,11 +130,11 @@ class EmbSpan(
     }
 
     private fun io.embrace.android.embracesdk.internal.payload.Link.retrieveSpanContext(): SpanContext {
-        return OtelJavaSpanContext.create(
-            checkNotNull(traceId),
-            checkNotNull(spanId),
-            OtelJavaTraceFlags.getDefault(),
-            OtelJavaTraceState.getDefault()
-        ).toOtelKotlin()
+        return SpanContext.create(
+            traceId = checkNotNull(traceId),
+            spanId = checkNotNull(spanId),
+            traceFlags = TraceFlags.default(),
+            traceState = TraceState.default(),
+        )
     }
 }

embrace-android-otel/src/main/kotlin/io/embrace/android/embracesdk/internal/otel/impl/EmbInvalidSpan.kt

@@ -3,24 +3,8 @@ package io.embrace.android.embracesdk.internal.otel.payload
 import io.embrace.android.embracesdk.internal.payload.Attribute
 import io.embrace.android.embracesdk.internal.payload.Log
 import io.embrace.opentelemetry.kotlin.ExperimentalApi
-import io.embrace.opentelemetry.kotlin.aliases.OtelJavaAttributes
-import io.embrace.opentelemetry.kotlin.aliases.OtelJavaLogRecordData
 import io.embrace.opentelemetry.kotlin.logging.model.ReadableLogRecord
 
-@Suppress("DEPRECATION") // suppress for backwards compat
-fun OtelJavaLogRecordData.toEmbracePayload(): Log {
-    val isSpanContextValid = spanContext.isValid
-    return Log(
-        traceId = if (isSpanContextValid) spanContext.traceId else null,
-        spanId = if (isSpanContextValid) spanContext.spanId else null,
-        timeUnixNano = timestampEpochNanos,
-        severityNumber = severity.severityNumber,
-        severityText = severityText,
-        body = body.asString(),
-        attributes = attributes.toEmbracePayload()
-    )
-}
-
 @OptIn(ExperimentalApi::class)
 fun ReadableLogRecord.toEmbracePayload(): Log {
     val isSpanContextValid = spanContext.isValid
@@ -34,6 +18,3 @@ fun ReadableLogRecord.toEmbracePayload(): Log {
         attributes = attributes.map { (key, value) -> Attribute(key, value.toString()) }
     )
 }
-
-fun OtelJavaAttributes.toEmbracePayload(): List<Attribute> =
-    this.asMap().entries.map { Attribute(it.key.key, it.value.toString()) }

embrace-android-otel/src/main/kotlin/io/embrace/android/embracesdk/internal/otel/impl/EmbSpan.kt

@@ -1,3 +1,5 @@
+@file:OptIn(ExperimentalApi::class)
+
 package io.embrace.android.embracesdk.internal.otel.payload
 
 import io.embrace.android.embracesdk.internal.clock.nanosToMillis

embrace-android-otel/src/main/kotlin/io/embrace/android/embracesdk/internal/otel/payload/LogMapper.kt

@@ -2,96 +2,32 @@ package io.embrace.android.embracesdk.internal.otel.sdk
 
 import io.embrace.android.embracesdk.internal.clock.nanosToMillis
 import io.embrace.android.embracesdk.internal.otel.attrs.EmbraceAttribute
-import io.embrace.android.embracesdk.internal.otel.attrs.EmbraceAttributeKey
-import io.embrace.android.embracesdk.internal.otel.attrs.asOtelAttributeKey
-import io.embrace.android.embracesdk.internal.otel.payload.toEmbracePayload
 import io.embrace.android.embracesdk.internal.otel.spans.EmbraceSpanData
 import io.embrace.android.embracesdk.internal.payload.Attribute
 import io.embrace.android.embracesdk.internal.payload.Link
 import io.embrace.android.embracesdk.spans.EmbraceSpanEvent
 import io.embrace.opentelemetry.kotlin.ExperimentalApi
-import io.embrace.opentelemetry.kotlin.aliases.OtelJavaAttributes
-import io.embrace.opentelemetry.kotlin.aliases.OtelJavaAttributesBuilder
-import io.embrace.opentelemetry.kotlin.aliases.OtelJavaEventData
-import io.embrace.opentelemetry.kotlin.aliases.OtelJavaLinkData
-import io.embrace.opentelemetry.kotlin.aliases.OtelJavaLogRecordBuilder
-import io.embrace.opentelemetry.kotlin.aliases.OtelJavaSpan
-import io.embrace.opentelemetry.kotlin.aliases.OtelJavaSpanData
-import io.embrace.opentelemetry.kotlin.k2j.tracing.convertToOtelKotlin
 import io.embrace.opentelemetry.kotlin.tracing.model.ReadableLink
 import io.embrace.opentelemetry.kotlin.tracing.model.ReadableSpan
 import io.embrace.opentelemetry.kotlin.tracing.model.ReadableSpanEvent
 import io.embrace.opentelemetry.kotlin.tracing.model.Span
 
 /**
- * Populate an [AttributesBuilder] with String key-value pairs from a [Map]
+ * Creates a sanitized map of attributes.
  */
-fun OtelJavaAttributesBuilder.fromMap(
-    attributes: Map<String, String>,
+fun Map<String, String>.sanitizeAttributesMap(
     internal: Boolean,
     limitsValidator: DataValidator,
-): OtelJavaAttributesBuilder {
-    attributes.filter {
+): Map<String, String> {
+    return filter {
         limitsValidator.isAttributeValid(it.key, it.value, internal) || it.key.isValidLongValueAttribute()
-    }.forEach {
-        put(it.key, it.value)
-    }
-    return this
+    }.toMap()
 }
 
-/**
- * Returns the attributes as a new Map<String, String>
- */
-fun OtelJavaAttributes.toStringMap(): Map<String, String> = asMap().entries.associate {
-    it.key.key.toString() to it.value.toString()
-}
-
-fun OtelJavaLinkData.toEmbracePayload() = Link(
-    spanId = spanContext.spanId,
-    traceId = spanContext.traceId,
-    attributes = attributes.toEmbracePayload(),
-    isRemote = spanContext.isRemote
-)
-
-fun OtelJavaLogRecordBuilder.setEmbraceAttribute(embraceAttribute: EmbraceAttribute): OtelJavaLogRecordBuilder {
-    setAttribute(embraceAttribute.key.asOtelAttributeKey(), embraceAttribute.value)
-    return this
-}
-
-fun OtelJavaSpan.setEmbraceAttribute(key: EmbraceAttributeKey, value: String): OtelJavaSpan {
-    setAttribute(key.name, value)
-    return this
-}
-
-fun OtelJavaSpan.setEmbraceAttribute(embraceAttribute: EmbraceAttribute): OtelJavaSpan =
-    this@setEmbraceAttribute.setEmbraceAttribute(embraceAttribute.key, embraceAttribute.value)
-
 @OptIn(ExperimentalApi::class)
 fun Span.setEmbraceAttribute(embraceAttribute: EmbraceAttribute) =
     setStringAttribute(embraceAttribute.key.name, embraceAttribute.value)
 
-@OptIn(ExperimentalApi::class)
-fun OtelJavaSpanData.toEmbraceSpanData(): EmbraceSpanData = EmbraceSpanData(
-    traceId = spanContext.traceId,
-    spanId = spanContext.spanId,
-    parentSpanId = parentSpanId,
-    name = name,
-    startTimeNanos = startEpochNanos,
-    endTimeNanos = endEpochNanos,
-    status = status.statusCode.convertToOtelKotlin(),
-    events = events?.mapNotNull { it.toEmbracePayload() } ?: emptyList(),
-    attributes = attributes.toStringMap(),
-    links = links.map { it.toEmbracePayload() }
-)
-
-private fun OtelJavaEventData.toEmbracePayload(): EmbraceSpanEvent? {
-    return EmbraceSpanEvent.create(
-        name = name,
-        timestampMs = epochNanos.nanosToMillis(),
-        attributes = attributes.toStringMap(),
-    )
-}
-
 @OptIn(ExperimentalApi::class)
 fun ReadableSpan.toEmbracePayload(): EmbraceSpanData = EmbraceSpanData(
     traceId = spanContext.traceId,
@@ -122,6 +58,3 @@ fun ReadableSpanEvent.toEmbracePayload(): EmbraceSpanEvent? {
         attributes = attributes.mapValues { it.value.toString() },
     )
 }
-
-fun OtelJavaAttributes.hasEmbraceAttribute(embraceAttribute: EmbraceAttribute): Boolean =
-    asMap()[embraceAttribute.key.asOtelAttributeKey()] == embraceAttribute.value