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

lucix-aws · web-flow · commit 1d989f31b024 · 2024-11-07T10:40:33.000-05:00
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/codegen/smithy-aws-go-codegen/src/main/java/software/amazon/smithy/aws/go/codegen/AwsJsonRpc1_0.java b/codegen/smithy-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

Original file line number	Diff line number	Diff line change
`@@ -61,6 +61,14 @@ protected Set<StructureShape> generateErrorShapes(`
`61`	`61`	`}`
`62`	`62`	`}`
`63`	`63`
	`64`	`+ @Override`
	`65`	`+ protected void writeDefaultHeaders(GenerationContext context, OperationShape operation, GoWriter writer) {`
	`66`	`+ super.writeDefaultHeaders(context, operation, writer);`
	`67`	`+ if (isAwsQueryCompatibleTraitFound(context)) {`
	`68`	`+ writer.write("httpBindingEncoder.SetHeader(\"X-Amzn-Query-Mode\").Boolean(true)");`
	`69`	`+ }`
	`70`	`+ }`
	`71`	`+`
`64`	`72`	`@Override`
`65`	`73`	`protected void deserializeError(GenerationContext context, StructureShape shape) {`
`66`	`74`	`GoWriter writer = context.getWriter().get();`