From 1c7e6e8aef5aef2f5cf5dfe1622da2438a2091a8 Mon Sep 17 00:00:00 2001
From: David Yaffe <dayaffe@amazon.com>
Date: Mon, 19 Aug 2024 17:35:23 -0400
Subject: [PATCH] fix codegen for orchestrator telemetry

---
 .../swift/codegen/middleware/MiddlewareExecutionGenerator.kt  | 4 +++-
 smithy-swift-codegen/src/test/kotlin/EventStreamTests.kt      | 4 +++-
 .../src/test/kotlin/HttpProtocolClientGeneratorTests.kt       | 4 +++-
 3 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/smithy-swift-codegen/src/main/kotlin/software/amazon/smithy/swift/codegen/middleware/MiddlewareExecutionGenerator.kt b/smithy-swift-codegen/src/main/kotlin/software/amazon/smithy/swift/codegen/middleware/MiddlewareExecutionGenerator.kt
index 4cf8c341b..bdfe4556b 100644
--- a/smithy-swift-codegen/src/main/kotlin/software/amazon/smithy/swift/codegen/middleware/MiddlewareExecutionGenerator.kt
+++ b/smithy-swift-codegen/src/main/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
index 77dd24f6a..bb2b1a59b 100644
--- 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
index 2d8fe5cee..74c6113fe 100644
--- 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()