send opt-in query-compatible header where applicable (#2891)

aws · Nov 7, 2024 · 1d989f3 · 1d989f3
1 parent 5d0eb23
commit 1d989f3
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 0 deletions.
diff --git a/.changelog/ccc0add5b568413fb8cf09c94a4c7fa7.json b/.changelog/ccc0add5b568413fb8cf09c94a4c7fa7.json
@@ -0,0 +1,8 @@
+{
+    "id": "ccc0add5-b568-413f-b8cf-09c94a4c7fa7",
+    "type": "feature",
+    "description": "Send opt-in query-compatible header for applicable services.",
+    "modules": [
+        "service/sqs"
+    ]
+}
diff --git a/...thy-aws-go-codegen/src/main/java/software/amazon/smithy/aws/go/codegen/AwsJsonRpc1_0.java b/...thy-aws-go-codegen/src/main/java/software/amazon/smithy/aws/go/codegen/AwsJsonRpc1_0.java
@@ -61,6 +61,14 @@ protected Set<StructureShape> generateErrorShapes(
         }
     }
 
+    @Override
+    protected void writeDefaultHeaders(GenerationContext context, OperationShape operation, GoWriter writer) {
+        super.writeDefaultHeaders(context, operation, writer);
+        if (isAwsQueryCompatibleTraitFound(context)) {
+            writer.write("httpBindingEncoder.SetHeader(\"X-Amzn-Query-Mode\").Boolean(true)");
+        }
+    }
+
     @Override
     protected void deserializeError(GenerationContext context, StructureShape shape) {
         GoWriter writer = context.getWriter().get();

diff --git a/service/sqs/serializers.go b/service/sqs/serializers.go