set statemet with optional equals

Author: wumpz

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

@@ -49,12 +49,22 @@ public class SetStatement implements Statement {
 
     private String name;
     private Expression expression;
+    private boolean useEqual;
 
     public SetStatement(String name, Expression expression) {
         this.name = name;
         this.expression = expression;
     }
 
+    public boolean isUseEqual() {
+        return useEqual;
+    }
+
+    public SetStatement setUseEqual(boolean useEqual) {
+        this.useEqual = useEqual;
+        return this;
+    }
+
     public String getName() {
         return name;
     }
@@ -73,7 +83,7 @@ public void setExpression(Expression expression) {
 
     @Override
     public String toString() {
-        return "SET " + name + " = " + expression.toString();
+        return "SET " + name + (useEqual?" = ":" ") + expression.toString();
     }
 
     @Override

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

@@ -27,39 +27,42 @@
 
 public class SetStatementDeParser {
 
-	private StringBuilder buffer;
-	private ExpressionVisitor expressionVisitor;
+    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;
-	}
+    /**
+     * @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 StringBuilder getBuffer() {
+        return buffer;
+    }
 
-	public void setBuffer(StringBuilder buffer) {
-		this.buffer = buffer;
-	}
+    public void setBuffer(StringBuilder buffer) {
+        this.buffer = buffer;
+    }
 
-	public void deParse(SetStatement set) {
-		buffer.append("SET ").append(set.getName());
-		buffer.append(" = ");
+    public void deParse(SetStatement set) {
+        buffer.append("SET ").append(set.getName());
+        if (set.isUseEqual()) {
+            buffer.append(" =");
+        }
+        buffer.append(" ");
         set.getExpression().accept(expressionVisitor);
-	}
+    }
 
-	public ExpressionVisitor getExpressionVisitor() {
-		return expressionVisitor;
-	}
+    public ExpressionVisitor getExpressionVisitor() {
+        return expressionVisitor;
+    }
 
-	public void setExpressionVisitor(ExpressionVisitor visitor) {
-		expressionVisitor = visitor;
-	}
+    public void setExpressionVisitor(ExpressionVisitor visitor) {
+        expressionVisitor = visitor;
+    }
 }

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

@@ -411,11 +411,12 @@ void error_skipto(int kind) {
 SetStatement Set(): {
     String name;
     Expression value;
+    boolean useEqual = false;
 }
 {
-    <K_SET> name = RelObjectNameExt() "=" value=SimpleExpression()
+    <K_SET> name = RelObjectNameExt() ["=" { useEqual=true; } ] value=SimpleExpression()
     {
-        return new SetStatement(name,value);
+        return new SetStatement(name,value).setUseEqual(useEqual);
     }
 }
 

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

@@ -57,4 +57,13 @@ public void testSimpleSet() throws JSQLParserException {
         assertSqlCanBeParsedAndDeparsed("SET statement_timeout = 0");
     }
 
+    @Test
+    public void testIssue373() throws JSQLParserException {
+        assertSqlCanBeParsedAndDeparsed("SET deferred_name_resolution true");
+    }
+    
+    @Test
+    public void testIssue373_2() throws JSQLParserException {
+        assertSqlCanBeParsedAndDeparsed("SET tester 5");
+    }
 }