Update OtelBase.kt

Author: rli

plugins/core/jetbrains-community/src/software/aws/toolkits/jetbrains/services/telemetry/otel/OtelBase.kt

@@ -191,9 +191,9 @@ abstract class AbstractSpanBuilder<
 
 abstract class AbstractBaseSpan<SpanType : AbstractBaseSpan<SpanType>>(internal val context: Context?, private val delegate: ReadWriteSpan) : Span by delegate {
     protected open val requiredFields: Collection<String> = emptySet()
-    protected var _passive: Boolean = false
-    protected var _unit: MetricUnit = MetricUnit.NONE
-    protected var _value: Double = 0.0
+    private var passive: Boolean = false
+    private var unit: MetricUnit = MetricUnit.NONE
+    private var value: Double = 0.0
 
     /**
      * Same as [com.intellij.platform.diagnostic.telemetry.helpers.use] except downcasts to specific subclass of [BaseSpan]
@@ -220,6 +220,21 @@ 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(", ")}" }