tst

Author: rli

telemetry/jetbrains/src/main/kotlin/software/aws/toolkits/telemetry/generator/OTelTelemetryGenerator.kt

@@ -9,7 +9,6 @@ import com.squareup.kotlinpoet.CodeBlock
 import com.squareup.kotlinpoet.FileSpec
 import com.squareup.kotlinpoet.FunSpec
 import com.squareup.kotlinpoet.KModifier
-import com.squareup.kotlinpoet.NUMBER
 import com.squareup.kotlinpoet.ParameterizedTypeName.Companion.parameterizedBy
 import com.squareup.kotlinpoet.PropertySpec
 import com.squareup.kotlinpoet.STRING

telemetry/jetbrains/src/test/resources/generateOTelBaseSpan/output/software/aws/toolkits/telemetry/impl/BaseSpan.kt

@@ -9,24 +9,15 @@ import io.opentelemetry.api.trace.Span
 import io.opentelemetry.context.Context
 import io.opentelemetry.sdk.trace.ReadWriteSpan
 import kotlin.Boolean
-import kotlin.Number
 import kotlin.Suppress
-import software.amazon.awssdk.services.toolkittelemetry.model.MetricUnit
 import software.aws.toolkits.jetbrains.services.telemetry.otel.AbstractBaseSpan
 
 public open class BaseSpan<SpanType : BaseSpan<SpanType>>(
     context: Context?,
     `delegate`: Span,
 ) : AbstractBaseSpan<SpanType>(context, delegate as ReadWriteSpan) {
-    public fun passive(passive: Boolean) {
-        this._passive = passive
-    }
-
-    public fun unit(unit: MetricUnit) {
-        this._unit = unit
-    }
-
-    public fun `value`(`value`: Number) {
-        this._value = `value`.toDouble()
+    public fun success(success: Boolean): SpanType {
+        result(if(success) MetricResult.Succeeded else MetricResult.Failed)
+        return this as SpanType
     }
 }

telemetry/jetbrains/src/test/resources/generatesOTelWithNormalInput/output/software/aws/toolkits/telemetry/impl/BaseSpan.kt

@@ -11,9 +11,7 @@ import io.opentelemetry.sdk.trace.ReadWriteSpan
 import kotlin.Boolean
 import kotlin.Double
 import kotlin.Float
-import kotlin.Number
 import kotlin.Suppress
-import software.amazon.awssdk.services.toolkittelemetry.model.MetricUnit
 import software.aws.toolkits.jetbrains.services.telemetry.otel.AbstractBaseSpan
 
 public open class BaseSpan<SpanType : BaseSpan<SpanType>>(
@@ -23,26 +21,15 @@ public open class BaseSpan<SpanType : BaseSpan<SpanType>>(
     /**
      * The duration of the operation in miliseconds
      */
-    public fun duration(duration: Float?) {
-        metadata("duration", duration?.let { it.toString() })
-    }
+    public fun duration(duration: Float?): SpanType = metadata("duration", duration?.let { it.toString() })
 
     /**
      * The duration of the operation in miliseconds
      */
-    public fun duration(duration: Double?) {
-        metadata("duration", duration?.let { it.toString() })
-    }
-
-    public fun passive(passive: Boolean) {
-        this._passive = passive
-    }
-
-    public fun unit(unit: MetricUnit) {
-        this._unit = unit
-    }
+    public fun duration(duration: Double?): SpanType = metadata("duration", duration?.let { it.toString() })
 
-    public fun `value`(`value`: Number) {
-        this._value = `value`.toDouble()
+    public fun success(success: Boolean): SpanType {
+        result(if(success) MetricResult.Succeeded else MetricResult.Failed)
+        return this as SpanType
     }
 }

telemetry/jetbrains/src/test/resources/generatesOTelWithNormalInput/output/software/aws/toolkits/telemetry/impl/LambdaTracer.kt

@@ -35,9 +35,7 @@ public class LambdadeleteSpan internal constructor(
     /**
      * a test boolean type
      */
-    public fun booltype(booltype: Boolean) {
-        metadata("booltype", booltype.toString())
-    }
+    public fun booltype(booltype: Boolean): LambdadeleteSpan = metadata("booltype", booltype.toString())
 }
 
 public class LambdadeleteSpanBuilder internal constructor(
@@ -62,16 +60,12 @@ public class LambdacreateSpan internal constructor(
     /**
      * The lambda runtime
      */
-    public fun lambdaRuntime(lambdaRuntime: LambdaRuntime) {
-        metadata("lambdaRuntime", lambdaRuntime.toString())
-    }
+    public fun lambdaRuntime(lambdaRuntime: LambdaRuntime): LambdacreateSpan = metadata("lambdaRuntime", lambdaRuntime.toString())
 
     /**
      * untyped string type
      */
-    public fun arbitraryString(arbitraryString: String) {
-        metadata("arbitraryString", arbitraryString)
-    }
+    public fun arbitraryString(arbitraryString: String): LambdacreateSpan = metadata("arbitraryString", arbitraryString)
 }
 
 public class LambdacreateSpanBuilder internal constructor(
@@ -96,23 +90,17 @@ public class LambdaremoteinvokeSpan internal constructor(
     /**
      * The lambda runtime
      */
-    public fun lambdaRuntime(lambdaRuntime: LambdaRuntime?) {
-        metadata("lambdaRuntime", lambdaRuntime?.let { it.toString() })
-    }
+    public fun lambdaRuntime(lambdaRuntime: LambdaRuntime?): LambdaremoteinvokeSpan = metadata("lambdaRuntime", lambdaRuntime?.let { it.toString() })
 
     /**
      * a test int type
      */
-    public fun inttype(inttype: Long) {
-        metadata("inttype", inttype.toString())
-    }
+    public fun inttype(inttype: Long): LambdaremoteinvokeSpan = metadata("inttype", inttype.toString())
 
     /**
      * a test int type
      */
-    public fun inttype(inttype: Int) {
-        metadata("inttype", inttype.toString())
-    }
+    public fun inttype(inttype: Int): LambdaremoteinvokeSpan = metadata("inttype", inttype.toString())
 }
 
 public class LambdaremoteinvokeSpanBuilder internal constructor(