emit business metrics

Author: xinsong-cui

aws-runtime/aws-config/api/aws-config.api

@@ -307,6 +307,10 @@ public final class aws/sdk/kotlin/runtime/config/endpoints/AccountIdEndpointMode
 	public static fun values ()[Laws/sdk/kotlin/runtime/config/endpoints/AccountIdEndpointMode;
 }
 
+public final class aws/sdk/kotlin/runtime/config/endpoints/AccountIdEndpointModeKt {
+	public static final fun toBusinessMetric (Laws/sdk/kotlin/runtime/config/endpoints/AccountIdEndpointMode;)Laws/smithy/kotlin/runtime/businessmetrics/BusinessMetric;
+}
+
 public final class aws/sdk/kotlin/runtime/config/endpoints/ResolversKt {
 	public static final fun resolveAccountId (Laws/sdk/kotlin/runtime/config/endpoints/AccountIdEndpointMode;Laws/smithy/kotlin/runtime/collections/Attributes;)Ljava/lang/String;
 	public static final fun resolveAccountIdEndpointMode (Laws/smithy/kotlin/runtime/util/PlatformProvider;Laws/smithy/kotlin/runtime/util/LazyAsyncValue;Lkotlin/coroutines/Continuation;)Ljava/lang/Object;

aws-runtime/aws-config/common/src/aws/sdk/kotlin/runtime/config/endpoints/AccountIdEndpointMode.kt

@@ -4,6 +4,9 @@
  */
 package aws.sdk.kotlin.runtime.config.endpoints
 
+import aws.smithy.kotlin.runtime.businessmetrics.BusinessMetric
+import aws.smithy.kotlin.runtime.businessmetrics.SmithyBusinessMetric
+
 /**
  * Controls how the account ID endpoint parameter is bound for services that support routing
  * endpoints based on it.
@@ -25,3 +28,10 @@ public enum class AccountIdEndpointMode {
      */
     REQUIRED,
 }
+
+public fun AccountIdEndpointMode.toBusinessMetric(): BusinessMetric = when (this) {
+    AccountIdEndpointMode.PREFERRED -> SmithyBusinessMetric.ACCOUNT_ID_MODE_PREFERRED
+    AccountIdEndpointMode.DISABLED -> SmithyBusinessMetric.ACCOUNT_ID_MODE_DISABLED
+    AccountIdEndpointMode.REQUIRED -> SmithyBusinessMetric.ACCOUNT_ID_MODE_REQUIRED
+    else -> throw IllegalStateException("Unexpected AccountIdEndpointMode value: ${this::class.simpleName}")
+}

codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/AwsRuntimeTypes.kt

@@ -17,8 +17,11 @@ object AwsRuntimeTypes {
         val AwsServiceException = symbol("AwsServiceException")
         val ClientException = symbol("ClientException")
 
+        // FUTURE MERGE CONFLICG: This package will be removed when region provider PR merges: https://github.com/awslabs/aws-sdk-kotlin/pull/1488
+        // Prepare for merge conflicts
         object Client : RuntimeTypePackage(AwsKotlinDependency.AWS_CORE, "client") {
             val AwsSdkClientConfig = symbol("AwsSdkClientConfig")
+            val AwsClientOption = symbol("AwsClientOption")
         }
     }
 
@@ -41,6 +44,7 @@ object AwsRuntimeTypes {
             val resolveEndpointUrl = symbol("resolveEndpointUrl")
             val resolveAccountId = symbol("resolveAccountId")
             val resolveAccountIdEndpointMode = symbol("resolveAccountIdEndpointMode")
+            val toBusinessMetric = symbol("toBusinessMetric")
         }
 
         object Profile : RuntimeTypePackage(AwsKotlinDependency.AWS_CONFIG, "config.profile") {

codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/businessmetrics/EndpointBusinessMetricsIntegration.kt

@@ -38,6 +38,12 @@ class EndpointBusinessMetricsIntegration : KotlinIntegration {
             RuntimeTypes.Core.BusinessMetrics.emitBusinessMetric,
             AwsRuntimeTypes.Http.Interceptors.BusinessMetrics.AwsBusinessMetric,
         )
+        writer.write(
+            "if (request.identity.attributes.contains(#T.AccountId)) request.context.#T(#T.RESOLVED_ACCOUNT_ID)",
+            AwsRuntimeTypes.Core.Client.AwsClientOption,
+            RuntimeTypes.Core.BusinessMetrics.emitBusinessMetric,
+            RuntimeTypes.Core.BusinessMetrics.SmithyBusinessMetric,
+        )
         writer.write("")
     }
 }

codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/customization/AccountIdEndpointBuiltinCustomization.kt

@@ -9,14 +9,18 @@ import aws.sdk.kotlin.codegen.ServiceClientCompanionObjectWriter
 import aws.sdk.kotlin.codegen.endpoints.AwsBuiltins
 import software.amazon.smithy.kotlin.codegen.KotlinSettings
 import software.amazon.smithy.kotlin.codegen.core.CodegenContext
+import software.amazon.smithy.kotlin.codegen.core.KotlinWriter
 import software.amazon.smithy.kotlin.codegen.integration.AppendingSectionWriter
 import software.amazon.smithy.kotlin.codegen.integration.KotlinIntegration
 import software.amazon.smithy.kotlin.codegen.integration.SectionWriterBinding
 import software.amazon.smithy.kotlin.codegen.model.expectShape
 import software.amazon.smithy.kotlin.codegen.model.getEndpointRules
+import software.amazon.smithy.kotlin.codegen.rendering.protocol.ProtocolGenerator
+import software.amazon.smithy.kotlin.codegen.rendering.protocol.ProtocolMiddleware
 import software.amazon.smithy.kotlin.codegen.rendering.util.ConfigProperty
 import software.amazon.smithy.kotlin.codegen.rendering.util.ConfigPropertyType
 import software.amazon.smithy.model.Model
+import software.amazon.smithy.model.shapes.OperationShape
 import software.amazon.smithy.model.shapes.ServiceShape
 
 /**
@@ -59,4 +63,20 @@ class AccountIdEndpointBuiltinCustomization : KotlinIntegration {
 
     override fun additionalServiceConfigProps(ctx: CodegenContext): List<ConfigProperty> =
         listOf(AccountIdEndpointModeProp)
+
+    override fun customizeMiddleware(
+        ctx: ProtocolGenerator.GenerationContext,
+        resolved: List<ProtocolMiddleware>,
+    ): List<ProtocolMiddleware> = resolved + listOf(emitAccountIdEndpointModeMiddleware)
+}
+
+private val emitAccountIdEndpointModeMiddleware = object : ProtocolMiddleware {
+    override val name: String = "EmitAccountIdEndpointModeMiddleware"
+
+    override fun render(ctx: ProtocolGenerator.GenerationContext, op: OperationShape, writer: KotlinWriter) {
+        writer.write(
+            "op.context.emitBusinessMetric(config.accountIdEndpointMode.#T())",
+            AwsRuntimeTypes.Config.Endpoints.toBusinessMetric,
+        )
+    }
 }