add rollout switches for waiters (#527)

aws · Jul 19, 2024 · 0d6c804 · 0d6c804
1 parent fb114ca
commit 0d6c804
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/...mithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/integration/Waiters.java b/...mithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/integration/Waiters.java
@@ -58,8 +58,16 @@ public Set<Symbol> getAdditionalClientOptions() {
         return emptySet();
     }
 
+    public boolean enabledForService(Model model, ShapeId service) {
+        return true;
+    }
+
     @Override
     public void writeAdditionalFiles(GoCodegenContext ctx) {
+        if (!enabledForService(ctx.model(), ctx.settings().getService())) {
+            return;
+        }
+
         var service = ctx.settings().getService(ctx.model());
 
         TopDownIndex.of(ctx.model()).getContainedOperations(service).stream()

diff --git a/...ithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/integration/Waiters2.java b/...ithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/integration/Waiters2.java
@@ -57,8 +57,16 @@ public Set<Symbol> getAdditionalClientOptions() {
         return emptySet();
     }
 
+    public boolean enabledForService(Model model, ShapeId service) {
+        return true;
+    }
+
     @Override
     public void writeAdditionalFiles(GoCodegenContext ctx) {
+        if (!enabledForService(ctx.model(), ctx.settings().getService())) {
+            return;
+        }
+
         var service = ctx.settings().getService(ctx.model());
 
         TopDownIndex.of(ctx.model()).getContainedOperations(service).stream()