jvmMain.aws.smithy.kotlin.runtime.telemetry.otel.AttributeUtils.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of telemetry-provider-otel-jvm Show documentation

Telemetry provider based on OpenTelemetry

There is a newer version: 1.3.25

Show newest version

/*
 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * SPDX-License-Identifier: Apache-2.0
 */

package aws.smithy.kotlin.runtime.telemetry.otel

import aws.smithy.kotlin.runtime.collections.AttributeKey
import aws.smithy.kotlin.runtime.collections.Attributes
import aws.smithy.kotlin.runtime.collections.get
import io.opentelemetry.api.common.AttributeKey as OtelAttributeKey
import io.opentelemetry.api.common.Attributes as OtelAttributes

internal fun Attributes.toOtelAttributes(): OtelAttributes {
    val keys = this.keys
    if (keys.isEmpty()) return OtelAttributes.empty()
    val attrs = OtelAttributes.builder()
    keys.forEach {
        @Suppress("UNCHECKED_CAST")
        val key = it as AttributeKey
        val value = get(key)
        it.otelAttrKeyOrNull(value)?.let { otelKey ->
            attrs.put(otelKey, value)
        }
    }

    return attrs.build()
}

internal fun  AttributeKey.otelAttrKeyOrNull(value: T): OtelAttributeKey? {
    val otelKey = when (value) {
        is String -> OtelAttributeKey.stringKey(name)
        is Long -> OtelAttributeKey.longKey(name)
        is Boolean -> OtelAttributeKey.booleanKey(name)
        is Double -> OtelAttributeKey.doubleKey(name)
        is List<*> -> {
            when (value.firstOrNull()) {
                is String -> OtelAttributeKey.stringArrayKey(name)
                is Long -> OtelAttributeKey.longArrayKey(name)
                is Boolean -> OtelAttributeKey.booleanArrayKey(name)
                is Double -> OtelAttributeKey.doubleArrayKey(name)
                else -> null
            }
        }
        else -> null
    }

    @Suppress("UNCHECKED_CAST")
    return otelKey as? OtelAttributeKey
}