fix codegen

aws · Jan 16, 2024 · 4a37e46 · 4a37e46
1 parent fa8094f
commit 4a37e46
Showing 1 changed file with 25 additions and 15 deletions.
diff --git a/...rc/main/java/software/amazon/smithy/aws/go/codegen/customization/CloudFrontKVSSigV4a.java b/...rc/main/java/software/amazon/smithy/aws/go/codegen/customization/CloudFrontKVSSigV4a.java
@@ -70,7 +70,7 @@ public Model preprocessModel(Model model, GoSettings settings) {
                 .addShape(
                         service.toBuilder()
                                 .addTrait(v4a)
-                                .addTrait(new AuthTrait(SetUtils.of(SigV4Trait.ID, SigV4ATrait.ID)))
+                                .addTrait(new AuthTrait(SetUtils.of(SigV4ATrait.ID, SigV4Trait.ID)))
                                 .build()
                 )
                 .build();
@@ -82,20 +82,30 @@ public void processFinalizedModel(GoSettings settings, Model model) {
             return;
         }
         runtimeClientPlugins.add(
-            RuntimeClientPlugin.builder()
-            .configFields(
-               ListUtils.of(
-                       ConfigField.builder()
-                               .name(AwsSignatureVersion4aUtils.V4A_SIGNER_INTERFACE_NAME)
-                               .type(SymbolUtils.createValueSymbolBuilder(
-                                               AwsSignatureVersion4aUtils.V4A_SIGNER_INTERFACE_NAME)
-                                       .build())
-                               .documentation("Signature Version 4a (SigV4a) Signer")
-                               .build()
-               )
-           )
-           .build()
-        );
+                RuntimeClientPlugin.builder()
+                    .configFields(
+                        ListUtils.of(
+                                ConfigField.builder()
+                                        .name(AwsSignatureVersion4aUtils.V4A_SIGNER_INTERFACE_NAME)
+                                        .type(SymbolUtils.createValueSymbolBuilder(
+                                                        AwsSignatureVersion4aUtils.V4A_SIGNER_INTERFACE_NAME)
+                                                .build())
+                                        .documentation("Signature Version 4a (SigV4a) Signer")
+                                        .build()
+                        )
+                    )
+                    .build());
+        runtimeClientPlugins.add(
+                RuntimeClientPlugin.builder()
+                    .servicePredicate(CloudFrontKVSSigV4a::isCFKVSService)
+                    .addConfigFieldResolver(
+                            ConfigFieldResolver.builder()
+                                    .location(ConfigFieldResolver.Location.CLIENT)
+                                    .target(ConfigFieldResolver.Target.INITIALIZATION)
+                                    .resolver(SymbolUtils.createValueSymbolBuilder(
+                                            AwsSignatureVersion4aUtils.SIGNER_RESOLVER).build())
+                                    .build())
+                    .build());
     }
 
     @Override