fix: request compression codegen was broken for operations specified under resource (#489)

lucix-aws · web-flow · commit fa639ee42bb8 · 2023-12-11T12:23:07.000-05:00
diff --git a/codegen/smithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/requestcompression/RequestCompression.java b/codegen/smithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/requestcompression/RequestCompression.java
@@ -35,7 +35,6 @@
 import software.amazon.smithy.model.knowledge.TopDownIndex;
 import software.amazon.smithy.model.shapes.OperationShape;
 import software.amazon.smithy.model.shapes.ServiceShape;
-import software.amazon.smithy.model.shapes.ShapeId;
 import software.amazon.smithy.model.traits.RequestCompressionTrait;
 import software.amazon.smithy.utils.ListUtils;
 import software.amazon.smithy.utils.MapUtils;
@@ -81,13 +80,12 @@ public void writeAdditionalFiles(
             GoDelegator goDelegator
     ) {
         ServiceShape service = settings.getService(model);
-        for (ShapeId operationID : service.getAllOperations()) {
-            OperationShape operation = model.expectShape(operationID, OperationShape.class);
+        TopDownIndex.of(model).getContainedOperations(service).forEach(operation -> {
             if (!operation.hasTrait(RequestCompressionTrait.class)) {
-                continue;
+                return;
             }
             goDelegator.useShapeWriter(operation, writeMiddlewareHelper(symbolProvider, operation));
-        }
+        });
     }
 
 
