add businessmetrics feature ID for env-based bearer token (#3182)

Author: lucix-aws

.changelog/8aa450c47b4b4c08bd9e87041915eb82.json

@@ -0,0 +1,10 @@
+{
+ "id": "8aa450c4-7b4b-4c08-bd9e-87041915eb82",
+ "type": "feature",
+ "description": "Emit user-agent business metrics for use of env-based bearer tokens.",
+ "modules": [
+ ".",
+ "service/bedrock",
+ "service/bedrockruntime"
+ ]
+}

aws/middleware/user_agent.go

@@ -135,6 +135,8 @@ const (
 UserAgentFeatureCredentialsAwsSdkStore = "y" // n/a (this is used by .NET based sdk)
 UserAgentFeatureCredentialsHTTP = "z"
 UserAgentFeatureCredentialsIMDS = "0"
+
+UserAgentFeatureBearerServiceEnvVars = "3"
 )
 
 var credentialSourceToFeature = map[aws.CredentialSource]UserAgentFeature{

codegen/smithy-aws-go-codegen/src/main/java/software/amazon/smithy/aws/go/codegen/customization/service/BearerTokenEnvProvider.java

@@ -5,6 +5,7 @@
 
 import java.util.List;
 import java.util.Map;
+import software.amazon.smithy.aws.go.codegen.customization.AwsCustomGoDependency;
 import software.amazon.smithy.aws.traits.auth.SigV4Trait;
 import software.amazon.smithy.go.codegen.GoCodegenContext;
 import software.amazon.smithy.go.codegen.GoWriter;
@@ -59,7 +60,7 @@ private GoWriter.Writable bearerTokenResolver(ServiceShape service) {
  .replace(' ', '_')
  .replace('-', '_');
  return goTemplate("""
- $context:D $os:D $bearer:D
+ $context:D $os:D $bearer:D $awsmiddleware:D
  func resolveEnvBearerToken(options *Options) {
  token := os.Getenv("AWS_BEARER_TOKEN_$envSuffix:L")
  if len(token) == 0 { return }
@@ -68,12 +69,22 @@ func resolveEnvBearerToken(options *Options) {
  return bearer.Token{Value: token}, nil
  })
  options.AuthSchemePreference = []string{"httpBearerAuth"}
+ options.APIOptions = append(options.APIOptions, func(stack *middleware.Stack) error {
+ ua, err := getOrAddRequestUserAgent(stack)
+ if err != nil {
+ return err
+ }
+
+ ua.AddUserAgentFeature(awsmiddleware.UserAgentFeatureBearerServiceEnvVars)
+ return nil
+ })
  }
  """,
  Map.of(
  "context", SmithyGoDependency.CONTEXT,
  "os", SmithyGoDependency.OS,
  "bearer", SmithyGoDependency.SMITHY_AUTH_BEARER,
+ "awsmiddleware", AwsCustomGoDependency.AWS_MIDDLEWARE,
  "envSuffix", envSuffix
  ));
  }