[wip] add AstRewriter; 266 plan tests failing

partiql · Oct 29, 2024 · dfc1db7 · dfc1db7
1 parent 46eb17f
commit dfc1db7
Show file tree

Hide file tree

Showing 9 changed files with 907 additions and 86 deletions.
diff --git a/partiql-ast/src/main/java/org/partiql/ast/v1/QueryBody.java b/partiql-ast/src/main/java/org/partiql/ast/v1/QueryBody.java
@@ -11,6 +11,17 @@
 import java.util.List;
 
 public abstract class QueryBody extends AstNode {
+    @Override
+    public <R, C> R accept(@NotNull AstVisitor<R, C> visitor, C ctx) {
+        if (this instanceof QueryBody.SFW) {
+            return visitor.visitQueryBodySFW((QueryBody.SFW) this, ctx);
+        } else if (this instanceof QueryBody.SetOp) {
+            return visitor.visitQueryBodySetOp((QueryBody.SetOp) this, ctx);
+        } else {
+            throw new IllegalStateException("Unknown QueryBody type: " + this.getClass().getName());
+        }
+    }
+
     @Builder(builderClassName = "Builder")
     @EqualsAndHashCode(callSuper = false)
     public static class SFW extends QueryBody {