fix codegen for orchestrator telemetry

smithy-lang · Aug 19, 2024 · 1c7e6e8 · 1c7e6e8
1 parent e7e71e9
commit 1c7e6e8
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/...in/kotlin/software/amazon/smithy/swift/codegen/middleware/MiddlewareExecutionGenerator.kt b/...in/kotlin/software/amazon/smithy/swift/codegen/middleware/MiddlewareExecutionGenerator.kt
@@ -84,7 +84,9 @@ class MiddlewareExecutionGenerator(
             let op = builder.attributes(context)
                 .telemetry(${"$"}N(
                     telemetryProvider: config.telemetryProvider,
-                    metricsAttributes: metricsAttributes
+                    metricsAttributes: metricsAttributes,
+                    meterScope: serviceName,
+                    tracerScope: serviceName
                 ))
                 .executeRequest(client)
                 .build()

diff --git a/smithy-swift-codegen/src/test/kotlin/EventStreamTests.kt b/smithy-swift-codegen/src/test/kotlin/EventStreamTests.kt
@@ -239,7 +239,9 @@ extension EventStreamTestClientTypes.TestStream {
         let op = builder.attributes(context)
             .telemetry(ClientRuntime.OrchestratorTelemetry(
                 telemetryProvider: config.telemetryProvider,
-                metricsAttributes: metricsAttributes
+                metricsAttributes: metricsAttributes,
+                meterScope: serviceName,
+                tracerScope: serviceName
             ))
             .executeRequest(client)
             .build()

diff --git a/smithy-swift-codegen/src/test/kotlin/HttpProtocolClientGeneratorTests.kt b/smithy-swift-codegen/src/test/kotlin/HttpProtocolClientGeneratorTests.kt
@@ -180,7 +180,9 @@ extension RestJsonProtocolClient {
         let op = builder.attributes(context)
             .telemetry(ClientRuntime.OrchestratorTelemetry(
                 telemetryProvider: config.telemetryProvider,
-                metricsAttributes: metricsAttributes
+                metricsAttributes: metricsAttributes,
+                meterScope: serviceName,
+                tracerScope: serviceName
             ))
             .executeRequest(client)
             .build()