simple merge implementation

Author: wumpz

src/main/java/net/sf/jsqlparser/statement/merge/Merge.java

@@ -88,6 +88,26 @@ public void setOnCondition(Expression onCondition) {
         this.onCondition = onCondition;
     }
 
+    private MergeInsert mergeInsert;
+
+    public MergeInsert getMergeInsert() {
+        return mergeInsert;
+    }
+
+    public void setMergeInsert(MergeInsert insert) {
+        this.mergeInsert = insert;
+    }
+
+    private MergeUpdate mergeUpdate;
+
+    public MergeUpdate getMergeUpdate() {
+        return mergeUpdate;
+    }
+
+    public void setMergeUpdate(MergeUpdate mergeUpdate) {
+        this.mergeUpdate = mergeUpdate;
+    }
+
     @Override
     public void accept(StatementVisitor statementVisitor) {
         statementVisitor.visit(this);
@@ -112,6 +132,14 @@ public String toString() {
         b.append(onCondition);
         b.append(")");
 
+        if (mergeUpdate != null) {
+            b.append(mergeUpdate.toString());
+        }
+
+        if (mergeInsert != null) {
+            b.append(mergeInsert.toString());
+        }
+
         return b.toString();
     }
 }

src/main/java/net/sf/jsqlparser/statement/merge/MergeInsert.java

@@ -0,0 +1,59 @@
+/*
+ * #%L
+ * JSQLParser library
+ * %%
+ * Copyright (C) 2004 - 2015 JSQLParser
+ * %%
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation, either version 2.1 of the
+ * License, or (at your option) any later version.
+ * 
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Lesser Public License for more details.
+ * 
+ * You should have received a copy of the GNU General Lesser Public
+ * License along with this program.  If not, see
+ * <http://www.gnu.org/licenses/lgpl-2.1.html>.
+ * #L%
+ */
+package net.sf.jsqlparser.statement.merge;
+
+import java.util.List;
+import net.sf.jsqlparser.expression.Expression;
+import net.sf.jsqlparser.schema.Column;
+import net.sf.jsqlparser.statement.select.PlainSelect;
+
+/**
+ *
+ * @author toben
+ */
+public class MergeInsert {
+
+    private List<Column> columns = null;
+    private List<Expression> values = null;
+
+    public List<Column> getColumns() {
+        return columns;
+    }
+
+    public void setColumns(List<Column> columns) {
+        this.columns = columns;
+    }
+
+    public List<Expression> getValues() {
+        return values;
+    }
+
+    public void setValues(List<Expression> values) {
+        this.values = values;
+    }
+
+    @Override
+    public String toString() {
+        return " WHEN NOT MATCHED THEN INSERT " + PlainSelect.getStringList(columns, true, true)
+                + " VALUES " + PlainSelect.getStringList(values, true, true);
+    }
+}

src/main/java/net/sf/jsqlparser/statement/merge/MergeUpdate.java

@@ -19,30 +19,70 @@
  * <http://www.gnu.org/licenses/lgpl-2.1.html>.
  * #L%
  */
-/*
- * Copyright (C) 2015 JSQLParser.
- *
- * This library is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation; either
- * version 2.1 of the License, or (at your option) any later version.
- *
- * This library is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public
- * License along with this library; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
- * MA 02110-1301  USA
- */
 package net.sf.jsqlparser.statement.merge;
 
+import java.util.List;
+import net.sf.jsqlparser.expression.Expression;
+import net.sf.jsqlparser.schema.Column;
+
 /**
  *
  * @author toben
  */
 public class MergeUpdate {
-    
+
+    private List<Column> columns = null;
+    private List<Expression> values = null;
+
+    public List<Column> getColumns() {
+        return columns;
+    }
+
+    public void setColumns(List<Column> columns) {
+        this.columns = columns;
+    }
+
+    public List<Expression> getValues() {
+        return values;
+    }
+
+    public void setValues(List<Expression> values) {
+        this.values = values;
+    }
+
+    private Expression whereCondition;
+
+    public Expression getWhereCondition() {
+        return whereCondition;
+    }
+
+    public void setWhereCondition(Expression whereCondition) {
+        this.whereCondition = whereCondition;
+    }
+
+    private Expression deleteWhereCondition;
+
+    public Expression getDeleteWhereCondition() {
+        return deleteWhereCondition;
+    }
+
+    public void setDeleteWhereCondition(Expression deleteWhereCondition) {
+        this.deleteWhereCondition = deleteWhereCondition;
+    }
+
+    @Override
+    public String toString() {
+        StringBuilder b = new StringBuilder();
+        b.append(" WHEN MATCHED THEN UPDATE SET ");
+        for (int i = 0; i < columns.size(); i++) {
+            b.append(columns.get(i).toString()).append(" = ").append(values.get(i).toString());
+        }
+        if (whereCondition != null) {
+            b.append(" WHERE ").append(whereCondition.toString());
+        }
+        if (deleteWhereCondition != null) {
+            b.append(" DELETE WHERE ").append(deleteWhereCondition.toString());
+        }
+        return b.toString();
+    }
 }

src/main/java/net/sf/jsqlparser/util/deparser/StatementDeParser.java

@@ -179,6 +179,7 @@ public void visit(SetStatement set) {
 
     @Override
     public void visit(Merge merge) {
-        throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates.
+        //TODO implementation of a deparser
+        buffer.append(merge.toString());
     }
 }

src/main/jjtree/net/sf/jsqlparser/parser/JSqlParserCC.jjt

@@ -527,6 +527,8 @@ Statement Merge() : {
     SubSelect select;
     Alias alias;
     Expression condition;
+    MergeUpdate update;
+    MergeInsert insert;
 }
 {
     <K_MERGE> <K_INTO> table=TableWithAlias() { merge.setTable(table); }
@@ -536,9 +538,9 @@ Statement Merge() : {
     ")" [ alias = Alias() { merge.setUsingAlias(alias); } ] <K_ON> 
        "(" condition = Condition() { merge.setOnCondition(condition); } ")"
 
-    [ LOOKAHEAD(2) MergeUpdateClause() ]
+    [ LOOKAHEAD(2) update = MergeUpdateClause() { merge.setMergeUpdate(update); } ]
 
-    [ MergeInsertClause() ]
+    [ insert = MergeInsertClause() { merge.setMergeInsert(insert); } ]
 
     { return merge; }
 }
@@ -547,7 +549,7 @@ MergeUpdate MergeUpdateClause() : {
     MergeUpdate mu = new MergeUpdate();
     List<Column> columns = new ArrayList<Column>();
     List<Expression> expList = new ArrayList<Expression>();
-    Columns col;
+    Column col;
     Expression exp;
     Expression condition;
 }
@@ -558,22 +560,30 @@ MergeUpdate MergeUpdateClause() : {
             { columns.add(col); expList.add(exp); } 
         ("," col = Column() "=" exp = SimpleExpression() { columns.add(col); expList.add(exp); } )* 
 
-        { mu.setColumns(columns); mu.setExpressions(expList); }
+        { mu.setColumns(columns); mu.setValues(expList); }
 
         [ <K_WHERE> condition = Condition() { mu.setWhereCondition(condition); }]
         [ <K_DELETE> <K_WHERE> condition = Condition() { mu.setDeleteWhereCondition(condition); } ]
 
         { return mu; }
 }
 
-void MergeInsertClause() : {
+MergeInsert MergeInsertClause() : {
+    MergeInsert mi = new MergeInsert();
     List<Column> columns = new ArrayList<Column>();
     List<Expression> expList = new ArrayList<Expression>();
+    Column col;
+    Expression exp;
 }
 {
     <K_WHEN> <K_NOT> <K_MATCHED> <K_THEN> 
-        <K_INSERT> "(" Column() ("," Column() )* ")"  <K_VALUES> 
-           "(" SimpleExpression() ("," SimpleExpression() )* ")" 
+        <K_INSERT> "(" col=Column() { columns.add(col); } ("," col=Column() { columns.add(col); } )* ")"  <K_VALUES> 
+           "(" exp=SimpleExpression() { expList.add(exp); } ("," exp=SimpleExpression() { expList.add(exp); } )* ")" 
+    {
+        mi.setColumns(columns);
+        mi.setValues(expList);
+        return mi;
+    }
 }
 
 // See: http://technet.microsoft.com/en-us/library/ms187879%28v=sql.105%29.aspx

src/test/java/net/sf/jsqlparser/test/merge/MergeTest.java

@@ -22,6 +22,7 @@
 import net.sf.jsqlparser.parser.CCJSqlParserUtil;
 import net.sf.jsqlparser.statement.Statement;
 import org.junit.Test;
+import static net.sf.jsqlparser.test.TestUtils.*;
 
 /**
  *
@@ -46,5 +47,7 @@ public void testOracleMergeIntoStatement() throws JSQLParserException {
         Statement statement = CCJSqlParserUtil.parse(sql);
 
         System.out.println(statement.toString());
+        
+        assertSqlCanBeParsedAndDeparsed(sql, true);
     }
 }