Fluent builder api #1004

pom.xml

@@ -49,6 +49,12 @@
             <version>3.16.1</version>
             <scope>test</scope>
         </dependency>
+        <dependency>
+		    <groupId>org.apache.commons</groupId>
+		    <artifactId>commons-lang3</artifactId>
+		    <version>3.10</version>
+		    <scope>test</scope>
+		</dependency>
     </dependencies>
 
     <developers>

src/main/java/net/sf/jsqlparser/Model.java

@@ -0,0 +1,18 @@
+/*-
+ * #%L
+ * JSQLParser library
+ * %%
+ * Copyright (C) 2004 - 2020 JSQLParser
+ * %%
+ * Dual licensed under GNU LGPL 2.1 or Apache License 2.0
+ * #L%
+ */
+package net.sf.jsqlparser;
+
+/**
+ * <p>A marker interface for jsqlparser-model-classes.</p>
+ * <p>The datastructure where the sql syntax is represented by a tree consists of {@link Model}'s</p>   
+ */
+public interface Model {
+
+}

src/main/java/net/sf/jsqlparser/expression/Alias.java

@@ -9,8 +9,12 @@
  */
 package net.sf.jsqlparser.expression;
 
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
 import java.util.List;
 import java.util.Objects;
+import java.util.Optional;
 import net.sf.jsqlparser.statement.create.table.ColDataType;
 
 public class Alias {
@@ -73,6 +77,33 @@ public String toString() {
         return alias;
     }
 
+    public Alias withName(String name) {
+        this.setName(name);
+        return this;
+    }
+
+    public Alias withUseAs(boolean useAs) {
+        this.setUseAs(useAs);
+        return this;
+    }
+
+    public Alias withAliasColumns(List<AliasColumn> aliasColumns) {
+        this.setAliasColumns(aliasColumns);
+        return this;
+    }
+
+    public Alias addAliasColumns(AliasColumn... aliasColumns) {
+        List<AliasColumn> collection = Optional.ofNullable(getAliasColumns()).orElseGet(ArrayList::new);
+        Collections.addAll(collection, aliasColumns);
+        return this.withAliasColumns(collection);
+    }
+
+    public Alias addAliasColumns(Collection<? extends AliasColumn> aliasColumns) {
+        List<AliasColumn> collection = Optional.ofNullable(getAliasColumns()).orElseGet(ArrayList::new);
+        collection.addAll(aliasColumns);
+        return this.withAliasColumns(collection);
+    }
+
     public static class AliasColumn {
 
         public final String name;

src/main/java/net/sf/jsqlparser/expression/AnalyticExpression.java

@@ -237,4 +237,75 @@ public Expression getFilterExpression() {
     public void setFilterExpression(Expression filterExpression) {
         this.filterExpression = filterExpression;
     }
+
+    public AnalyticExpression withName(String name) {
+        this.setName(name);
+        return this;
+    }
+
+    public AnalyticExpression withExpression(Expression expression) {
+        this.setExpression(expression);
+        return this;
+    }
+
+    public AnalyticExpression withOffset(Expression offset) {
+        this.setOffset(offset);
+        return this;
+    }
+
+    public AnalyticExpression withDefaultValue(Expression defaultValue) {
+        this.setDefaultValue(defaultValue);
+        return this;
+    }
+
+    public AnalyticExpression withAllColumns(boolean allColumns) {
+        this.setAllColumns(allColumns);
+        return this;
+    }
+
+    public AnalyticExpression withKeep(KeepExpression keep) {
+        this.setKeep(keep);
+        return this;
+    }
+
+    public AnalyticExpression withType(AnalyticType type) {
+        this.setType(type);
+        return this;
+    }
+
+    public AnalyticExpression withDistinct(boolean distinct) {
+        this.setDistinct(distinct);
+        return this;
+    }
+
+    public AnalyticExpression withIgnoreNulls(boolean ignoreNulls) {
+        this.setIgnoreNulls(ignoreNulls);
+        return this;
+    }
+
+    public AnalyticExpression withFilterExpression(Expression filterExpression) {
+        this.setFilterExpression(filterExpression);
+        return this;
+    }
+
+    public AnalyticExpression withWindowElement(WindowElement windowElement) {
+        this.setWindowElement(windowElement);
+        return this;
+    }
+
+    public <E extends Expression> E getExpression(Class<E> type) {
+        return type.cast(getExpression());
+    }
+
+    public <E extends Expression> E getOffset(Class<E> type) {
+        return type.cast(getOffset());
+    }
+
+    public <E extends Expression> E getDefaultValue(Class<E> type) {
+        return type.cast(getDefaultValue());
+    }
+
+    public <E extends Expression> E getFilterExpression(Class<E> type) {
+        return type.cast(getFilterExpression());
+    }
 }

src/main/java/net/sf/jsqlparser/expression/ArrayExpression.java

@@ -16,6 +16,10 @@ public class ArrayExpression extends ASTNodeAccessImpl implements Expression {
     private Expression objExpression;
     private Expression indexExpression;
 
+    public ArrayExpression() {
+        // empty constructor
+    }
+
     public ArrayExpression(Expression objExpression, Expression indexExpression) {
         this.objExpression = objExpression;
         this.indexExpression = indexExpression;
@@ -46,4 +50,22 @@ public void accept(ExpressionVisitor expressionVisitor) {
     public String toString() {
         return objExpression.toString() + "[" + indexExpression.toString() + "]";
     }
+
+    public ArrayExpression withObjExpression(Expression objExpression) {
+        this.setObjExpression(objExpression);
+        return this;
+    }
+
+    public ArrayExpression withIndexExpression(Expression indexExpression) {
+        this.setIndexExpression(indexExpression);
+        return this;
+    }
+
+    public <E extends Expression> E getObjExpression(Class<E> type) {
+        return type.cast(getObjExpression());
+    }
+
+    public <E extends Expression> E getIndexExpression(Class<E> type) {
+        return type.cast(getIndexExpression());
+    }
 }

src/main/java/net/sf/jsqlparser/expression/BinaryExpression.java

@@ -19,7 +19,7 @@ public abstract class BinaryExpression extends ASTNodeAccessImpl implements Expr
 
     private Expression leftExpression;
     private Expression rightExpression;
-//    private boolean not = false;
+    // private boolean not = false;
 
     public BinaryExpression() {
     }
@@ -32,31 +32,48 @@ public Expression getRightExpression() {
         return rightExpression;
     }
 
+    public BinaryExpression withLeftExpression(Expression expression) {
+        setLeftExpression(expression);
+        return this;
+    }
+
     public void setLeftExpression(Expression expression) {
         leftExpression = expression;
     }
 
+    public BinaryExpression withRightExpression(Expression expression) {
+        setRightExpression(expression);
+        return this;
+    }
+
     public void setRightExpression(Expression expression) {
         rightExpression = expression;
     }
 
-//    public void setNot() {
-//        not = true;
-//    }
-//    
-//    public void removeNot() {
-//        not = false;
-//    }
-// 
-//    public boolean isNot() {
-//        return not;
-//    }
+    // public void setNot() {
+    // not = true;
+    // }
+    //
+    // public void removeNot() {
+    // not = false;
+    // }
+    //
+    // public boolean isNot() {
+    // return not;
+    // }
     @Override
     public String toString() {
-        return //(not ? "NOT " : "") + 
+        return // (not ? "NOT " : "") +
                 getLeftExpression() + " " + getStringExpression() + " " + getRightExpression();
     }
 
     public abstract String getStringExpression();
 
+    public <E extends Expression> E getLeftExpression(Class<E> type) {
+        return type.cast(getLeftExpression());
+    }
+
+    public <E extends Expression> E getRightExpression(Class<E> type) {
+        return type.cast(getRightExpression());
+    }
 }

src/main/java/net/sf/jsqlparser/expression/CaseExpression.java

@@ -9,8 +9,11 @@
  */
 package net.sf.jsqlparser.expression;
 
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.Collections;
 import java.util.List;
-
+import java.util.Optional;
 import net.sf.jsqlparser.parser.ASTNodeAccessImpl;
 import net.sf.jsqlparser.statement.select.PlainSelect;
 
@@ -91,4 +94,39 @@ public String toString() {
                 + PlainSelect.getStringList(whenClauses, false, false) + " "
                 + ((elseExpression != null) ? "ELSE " + elseExpression + " " : "") + "END";
     }
+
+    public CaseExpression withSwitchExpression(Expression switchExpression) {
+        this.setSwitchExpression(switchExpression);
+        return this;
+    }
+
+    public CaseExpression withWhenClauses(List<WhenClause> whenClauses) {
+        this.setWhenClauses(whenClauses);
+        return this;
+    }
+
+    public CaseExpression withElseExpression(Expression elseExpression) {
+        this.setElseExpression(elseExpression);
+        return this;
+    }
+
+    public CaseExpression addWhenClauses(WhenClause... whenClauses) {
+        List<WhenClause> collection = Optional.ofNullable(getWhenClauses()).orElseGet(ArrayList::new);
+        Collections.addAll(collection, whenClauses);
+        return this.withWhenClauses(collection);
+    }
+
+    public CaseExpression addWhenClauses(Collection<? extends WhenClause> whenClauses) {
+        List<WhenClause> collection = Optional.ofNullable(getWhenClauses()).orElseGet(ArrayList::new);
+        collection.addAll(whenClauses);
+        return this.withWhenClauses(collection);
+    }
+
+    public <E extends Expression> E getSwitchExpression(Class<E> type) {
+        return type.cast(getSwitchExpression());
+    }
+
+    public <E extends Expression> E getElseExpression(Class<E> type) {
+        return type.cast(getElseExpression());
+    }
 }

src/main/java/net/sf/jsqlparser/expression/CastExpression.java

@@ -55,4 +55,23 @@ public String toString() {
             return leftExpression + "::" + type.toString();
         }
     }
+
+    public CastExpression withType(ColDataType type) {
+        this.setType(type);
+        return this;
+    }
+
+    public CastExpression withUseCastKeyword(boolean useCastKeyword) {
+        this.setUseCastKeyword(useCastKeyword);
+        return this;
+    }
+
+    public CastExpression withLeftExpression(Expression leftExpression) {
+        this.setLeftExpression(leftExpression);
+        return this;
+    }
+
+    public <E extends Expression> E getLeftExpression(Class<E> type) {
+        return type.cast(getLeftExpression());
+    }
 }

src/main/java/net/sf/jsqlparser/expression/CollateExpression.java

@@ -16,6 +16,10 @@ public class CollateExpression extends ASTNodeAccessImpl implements Expression {
     private Expression leftExpression;
     private String collate;
 
+    public CollateExpression() {
+        // empty constructor
+    }
+
     public CollateExpression(Expression leftExpression, String collate) {
         this.leftExpression = leftExpression;
         this.collate = collate;
@@ -46,4 +50,18 @@ public void setCollate(String collate) {
     public String toString() {
         return leftExpression.toString() + " COLLATE " + collate;
     }
+
+    public CollateExpression withLeftExpression(Expression leftExpression) {
+        this.setLeftExpression(leftExpression);
+        return this;
+    }
+
+    public CollateExpression withCollate(String collate) {
+        this.setCollate(collate);
+        return this;
+    }
+
+    public <E extends Expression> E getLeftExpression(Class<E> type) {
+        return type.cast(getLeftExpression());
+    }
 }

src/main/java/net/sf/jsqlparser/expression/DateTimeLiteralExpression.java

@@ -42,6 +42,16 @@ public String toString() {
         return type.name() + " " + value;
     }
 
+    public DateTimeLiteralExpression withValue(String value) {
+        this.setValue(value);
+        return this;
+    }
+
+    public DateTimeLiteralExpression withType(DateTime type) {
+        this.setType(type);
+        return this;
+    }
+
     public static enum DateTime {
         DATE, TIME, TIMESTAMP;
     }