use Span api instead (#114)

GuiaBolso · Aug 15, 2023 · a612c79 · a612c79
1 parent 1dd3c68
commit a612c79
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 15 deletions.
diff --git a/...cing/src/main/kotlin/br/com/guiabolso/tracing/engine/opentelemetry/OpenTelemetryTracer.kt b/...cing/src/main/kotlin/br/com/guiabolso/tracing/engine/opentelemetry/OpenTelemetryTracer.kt
@@ -28,7 +28,7 @@ class OpenTelemetryTracer : TracerEngine, ThreadContextManager<Span> {
     }
 
     override fun setOperationName(name: String) {
-        val span = getRootSpan()
+        val span = currentSpan()
         span?.updateName(name)
     }
 
@@ -37,23 +37,23 @@ class OpenTelemetryTracer : TracerEngine, ThreadContextManager<Span> {
     }
 
     override fun addRootProperty(key: String, value: String?) {
-        getRootSpan()?.addProperty(key, value)
+        currentSpan()?.addProperty(key, value)
     }
 
     override fun addProperty(key: String, value: Number?) {
         Span.current()?.addProperty(key, value)
     }
 
     override fun addRootProperty(key: String, value: Number?) {
-        getRootSpan()?.addProperty(key, value)
+        currentSpan()?.addProperty(key, value)
     }
 
     override fun addProperty(key: String, value: Boolean?) {
         Span.current()?.addProperty(key, value)
     }
 
     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<Span> {
     }
 
     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<Span> {
     }
 
     override fun notifyRootError(message: String, params: Map<String, String?>, expected: Boolean) {
-        getRootSpan()?.let { span ->
+        currentSpan()?.let { span ->
             OpenTelemetryUtils.notifyError(span, message, params, expected)
         }
     }
@@ -146,7 +146,7 @@ class OpenTelemetryTracer : TracerEngine, ThreadContextManager<Span> {
         } as AttributeKey<T>
     }
 
-    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/...a/tracing/src/main/kotlin/br/com/guiabolso/tracing/engine/opentelemetry/SpanContextKey.kt b/...a/tracing/src/main/kotlin/br/com/guiabolso/tracing/engine/opentelemetry/SpanContextKey.kt