copy old impl as close as possible

aws · Sep 25, 2023 · dae337a · dae337a
1 parent 4637909
commit dae337a
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/.../src/main/java/software/amazon/smithy/go/codegen/endpoints/EndpointResolverGenerator.java b/.../src/main/java/software/amazon/smithy/go/codegen/endpoints/EndpointResolverGenerator.java
@@ -37,15 +37,18 @@
 import software.amazon.smithy.go.codegen.GoWriter;
 import software.amazon.smithy.go.codegen.SmithyGoDependency;
 import software.amazon.smithy.go.codegen.SymbolUtils;
+import software.amazon.smithy.model.SourceException;
 import software.amazon.smithy.model.SourceLocation;
 import software.amazon.smithy.rulesengine.language.Endpoint;
 import software.amazon.smithy.rulesengine.language.EndpointRuleSet;
+import software.amazon.smithy.rulesengine.language.error.RuleError;
 import software.amazon.smithy.rulesengine.language.evaluation.type.OptionalType;
 import software.amazon.smithy.rulesengine.language.syntax.Identifier;
 import software.amazon.smithy.rulesengine.language.syntax.expressions.Expression;
 import software.amazon.smithy.rulesengine.language.syntax.expressions.literal.Literal;
 import software.amazon.smithy.rulesengine.language.syntax.expressions.Reference;
 import software.amazon.smithy.rulesengine.language.syntax.expressions.functions.FunctionDefinition;
+import software.amazon.smithy.rulesengine.language.syntax.expressions.functions.IsSet;
 import software.amazon.smithy.rulesengine.language.syntax.parameters.Parameter;
 import software.amazon.smithy.rulesengine.language.syntax.parameters.Parameters;
 import software.amazon.smithy.rulesengine.language.syntax.rule.Condition;
@@ -329,7 +332,16 @@ private GoWriter.Writable generateRule(Rule rule, List<Condition> conditions, Sc
     }
 
     private static Expression conditionalFunc(Condition condition) {
-        return condition.getFunction();
+        var fn = condition.getFunction();
+        if (fn instanceof IsSet) {
+            var setFn = ((IsSet) fn);
+            List<Expression> argv = setFn.getArguments();
+            if (argv.size() == 1) {
+                return argv.get(0);
+            }
+            throw new RuleError(new SourceException("expected 1 argument but found " + argv.size(), setFn));
+        }
+        return fn;
     }
 
     private String nameForExpression(Expression expr) {