fix: request compression codegen was broken for operations specified …

…under resource (#489)
aws · Dec 11, 2023 · fa639ee · fa639ee
1 parent f982aed
commit fa639ee
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/...rc/main/java/software/amazon/smithy/go/codegen/requestcompression/RequestCompression.java b/...rc/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));
-        }
+        });
     }