Update OtelBase.kt

aws · Nov 5, 2024 · 8da36b5 · 8da36b5
1 parent 06181dd
commit 8da36b5
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/...tbrains-community/src/software/aws/toolkits/jetbrains/services/telemetry/otel/OtelBase.kt b/...tbrains-community/src/software/aws/toolkits/jetbrains/services/telemetry/otel/OtelBase.kt
@@ -211,6 +211,11 @@ abstract class AbstractBaseSpan<SpanType : AbstractBaseSpan<SpanType>>(internal
         return this as SpanType
     }
 
+    override fun recordException(exception: Throwable): SpanType {
+        delegate.recordException(exception)
+        return this as SpanType
+    }
+
     override fun end() {
         validateRequiredAttributes()
         delegate.end()
@@ -221,6 +226,22 @@ abstract class AbstractBaseSpan<SpanType : AbstractBaseSpan<SpanType>>(internal
         delegate.end()
     }
 
+    fun passive(passive: Boolean): SpanType {
+        this._passive = passive
+        return this as SpanType
+    }
+
+    fun unit(unit: MetricUnit): SpanType {
+        this._unit = unit
+        return this as SpanType
+
+    }
+
+    fun value(value: Number): SpanType {
+        this._value = value.toDouble()
+        return this as SpanType
+    }
+
     private fun validateRequiredAttributes() {
         val missingFields = requiredFields.filter { delegate.getAttribute(AttributeKey.stringKey(it)) == null }
         val message = { "${delegate.name} is missing required fields: ${missingFields.joinToString(", ")}" }