diff --git a/impl/java/tracing/src/main/kotlin/br/com/guiabolso/tracing/engine/opentelemetry/OpenTelemetryTracer.kt b/impl/java/tracing/src/main/kotlin/br/com/guiabolso/tracing/engine/opentelemetry/OpenTelemetryTracer.kt index ca161e3a..491f5f32 100644 --- a/impl/java/tracing/src/main/kotlin/br/com/guiabolso/tracing/engine/opentelemetry/OpenTelemetryTracer.kt +++ b/impl/java/tracing/src/main/kotlin/br/com/guiabolso/tracing/engine/opentelemetry/OpenTelemetryTracer.kt @@ -28,7 +28,7 @@ class OpenTelemetryTracer : TracerEngine, ThreadContextManager { } override fun setOperationName(name: String) { - val span = getRootSpan() + val span = currentSpan() span?.updateName(name) } @@ -37,7 +37,7 @@ class OpenTelemetryTracer : TracerEngine, ThreadContextManager { } override fun addRootProperty(key: String, value: String?) { - getRootSpan()?.addProperty(key, value) + currentSpan()?.addProperty(key, value) } override fun addProperty(key: String, value: Number?) { @@ -45,7 +45,7 @@ class OpenTelemetryTracer : TracerEngine, ThreadContextManager { } override fun addRootProperty(key: String, value: Number?) { - getRootSpan()?.addProperty(key, value) + currentSpan()?.addProperty(key, value) } override fun addProperty(key: String, value: Boolean?) { @@ -53,7 +53,7 @@ class OpenTelemetryTracer : TracerEngine, ThreadContextManager { } override fun addRootProperty(key: String, value: Boolean?) { - getRootSpan()?.addProperty(key, value) + currentSpan()?.addProperty(key, value) } override fun addProperty(key: String, value: List<*>) { @@ -90,7 +90,7 @@ class OpenTelemetryTracer : TracerEngine, ThreadContextManager { } override fun notifyRootError(exception: Throwable, expected: Boolean) { - getRootSpan()?.let { span -> + currentSpan()?.let { span -> OpenTelemetryUtils.notifyError(span, exception, expected) } } @@ -102,7 +102,7 @@ class OpenTelemetryTracer : TracerEngine, ThreadContextManager { } override fun notifyRootError(message: String, params: Map, expected: Boolean) { - getRootSpan()?.let { span -> + currentSpan()?.let { span -> OpenTelemetryUtils.notifyError(span, message, params, expected) } } @@ -146,7 +146,7 @@ class OpenTelemetryTracer : TracerEngine, ThreadContextManager { } as AttributeKey } - private fun getRootSpan(): Span? = Context.root().get(SpanContextKey.KEY) + private fun currentSpan(): Span? = Span.current() companion object { const val TRACER_NAME = "events-tracing" diff --git a/impl/java/tracing/src/main/kotlin/br/com/guiabolso/tracing/engine/opentelemetry/SpanContextKey.kt b/impl/java/tracing/src/main/kotlin/br/com/guiabolso/tracing/engine/opentelemetry/SpanContextKey.kt deleted file mode 100644 index c68f86f4..00000000 --- a/impl/java/tracing/src/main/kotlin/br/com/guiabolso/tracing/engine/opentelemetry/SpanContextKey.kt +++ /dev/null @@ -1,8 +0,0 @@ -package br.com.guiabolso.tracing.engine.opentelemetry - -import io.opentelemetry.api.trace.Span -import io.opentelemetry.context.ContextKey - -internal object SpanContextKey { - internal val KEY = ContextKey.named("opentelemetry-trace-span-key") -}