fixes #110 - first implementation

Author: wumpz

pom.xml

@@ -19,18 +19,18 @@
 	</licenses>
 
 	<dependencies>
-		<dependency>
-			<groupId>junit</groupId>
-			<artifactId>junit</artifactId>
-			<version>4.11</version>
-            <scope>test</scope>
-		</dependency>
 		<dependency>
 			<groupId>commons-io</groupId>
 			<artifactId>commons-io</artifactId>
 			<version>2.4</version>
 			<scope>test</scope>
 		</dependency>
+	 <dependency>
+	  <groupId>junit</groupId>
+	  <artifactId>junit</artifactId>
+	  <version>4.12</version>
+	  <scope>test</scope>
+	 </dependency>
 	</dependencies>
 
 	<developers>

src/main/java/net/sf/jsqlparser/statement/SetStatement.java

@@ -0,0 +1,83 @@
+/*
+ * #%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%
+ */
+/*
+ * 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;
+
+import net.sf.jsqlparser.expression.Expression;
+
+/**
+ *
+ * @author toben
+ */
+public class SetStatement implements Statement {
+
+    private String name;
+    private Expression expression;
+
+    public SetStatement(String name, Expression expression) {
+        this.name = name;
+        this.expression = expression;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
+
+    public Expression getExpression() {
+        return expression;
+    }
+
+    public void setExpression(Expression expression) {
+        this.expression = expression;
+    }
+
+    @Override
+    public String toString() {
+        return "SET " + name + " = " + expression.toString();
+    }
+
+    @Override
+    public void accept(StatementVisitor statementVisitor) {
+        statementVisitor.visit(this);
+    }
+}

src/main/java/net/sf/jsqlparser/statement/StatementVisitor.java

@@ -61,4 +61,6 @@ public interface StatementVisitor {
     void visit(Statements stmts);
 
     void visit(Execute execute);
+    
+    void visit(SetStatement set);
 }

src/main/java/net/sf/jsqlparser/statement/StatementVisitorAdapter.java

@@ -99,4 +99,9 @@ public void visit(Statements stmts) {
     public void visit(Execute execute) {
 
     }
+
+    @Override
+    public void visit(SetStatement set) {
+        
+    }
 }

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

@@ -0,0 +1,65 @@
+/*
+ * #%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.util.deparser;
+
+import net.sf.jsqlparser.expression.ExpressionVisitor;
+import net.sf.jsqlparser.statement.SetStatement;
+import net.sf.jsqlparser.statement.select.PlainSelect;
+
+public class SetStatementDeParser {
+
+	private StringBuilder buffer;
+	private ExpressionVisitor expressionVisitor;
+
+	/**
+	 * @param expressionVisitor a {@link ExpressionVisitor} to de-parse
+	 * expressions. It has to share the same<br>
+	 * StringBuilder (buffer parameter) as this object in order to work
+	 * @param buffer the buffer that will be filled with the select
+	 */
+	public SetStatementDeParser(ExpressionVisitor expressionVisitor, StringBuilder buffer) {
+		this.buffer = buffer;
+		this.expressionVisitor = expressionVisitor;
+	}
+
+	public StringBuilder getBuffer() {
+		return buffer;
+	}
+
+	public void setBuffer(StringBuilder buffer) {
+		this.buffer = buffer;
+	}
+
+	public void deParse(SetStatement set) {
+		buffer.append("SET ").append(set.getName());
+		buffer.append(" = ");
+        set.getExpression().accept(expressionVisitor);
+	}
+
+	public ExpressionVisitor getExpressionVisitor() {
+		return expressionVisitor;
+	}
+
+	public void setExpressionVisitor(ExpressionVisitor visitor) {
+		expressionVisitor = visitor;
+	}
+}

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

@@ -22,6 +22,7 @@
 package net.sf.jsqlparser.util.deparser;
 
 import java.util.Iterator;
+import net.sf.jsqlparser.statement.SetStatement;
 
 import net.sf.jsqlparser.statement.StatementVisitor;
 import net.sf.jsqlparser.statement.Statements;
@@ -164,4 +165,14 @@ public void visit(Execute execute) {
         selectDeParser.setExpressionVisitor(expressionDeParser);
         executeDeParser.deParse(execute);
     }
+
+    @Override
+    public void visit(SetStatement set) {
+        SelectDeParser selectDeParser = new SelectDeParser();
+        selectDeParser.setBuffer(buffer);
+        ExpressionDeParser expressionDeParser = new ExpressionDeParser(selectDeParser, buffer);
+        SetStatementDeParser setStatementDeparser = new SetStatementDeParser(expressionDeParser, buffer);
+        selectDeParser.setExpressionVisitor(expressionDeParser);
+        setStatementDeparser.deParse(set);
+    }
 }

src/main/javacc/net/sf/jsqlparser/parser/JSqlParserCC.jj

@@ -288,6 +288,8 @@ Statement SingleStatement() :
     stm = Truncate()
     |
     stm = Execute()
+    |
+    stm = Set()
     )
     { return stm; }
 }
@@ -309,6 +311,17 @@ Statements Statements() :
     }
 }
 
+SetStatement Set(): {
+    String name;
+    Expression value;
+}
+{
+    <K_SET> name = RelObjectNameExt() "=" value=SimpleExpression()
+    {
+        return new SetStatement(name,value);
+    }
+}
+
 Update Update():
 {
 	Update update = new Update();

src/test/java/net/sf/jsqlparser/statement/SetStatementTest.java

@@ -0,0 +1,60 @@
+/*
+ * 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;
+
+import net.sf.jsqlparser.JSQLParserException;
+import static net.sf.jsqlparser.test.TestUtils.assertSqlCanBeParsedAndDeparsed;
+import org.junit.After;
+import org.junit.AfterClass;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+
+/**
+ *
+ * @author toben
+ */
+public class SetStatementTest {
+    
+    public SetStatementTest() {
+    }
+    
+    @BeforeClass
+    public static void setUpClass() {
+    }
+    
+    @AfterClass
+    public static void tearDownClass() {
+    }
+    
+    @Before
+    public void setUp() {
+    }
+    
+    @After
+    public void tearDown() {
+    }
+
+   
+    @Test
+    public void testSimpleSet() throws JSQLParserException {
+        assertSqlCanBeParsedAndDeparsed("SET statement_timeout = 0");
+    }
+    
+}