Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

improved sql parser. for issue #1933

Browse Source
This commit is contained in:
wenshao
2017-09-04 10:45:28 +08:00
parent 5badd35b7f
commit e8799e61b1
11 changed files with 322 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/sql/parser/SQLStatementParser.java
+27 -19
View File
@@ -15,7 +15,6 @@
*/
package com.alibaba.druid.sql.parser;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;
@@ -27,11 +26,9 @@ import com.alibaba.druid.sql.ast.expr.SQLVariantRefExpr;
import com.alibaba.druid.sql.ast.statement.*;
import com.alibaba.druid.sql.ast.statement.SQLCreateTriggerStatement.TriggerEvent;
import com.alibaba.druid.sql.ast.statement.SQLCreateTriggerStatement.TriggerType;
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlRepeatStatement;
import com.alibaba.druid.sql.dialect.oracle.parser.OracleExprParser;
import com.alibaba.druid.sql.dialect.postgresql.ast.stmt.*;
import com.alibaba.druid.util.FnvHash;
import com.alibaba.druid.util.JdbcConstants;
import com.alibaba.druid.util.JdbcUtils;
public class SQLStatementParser extends SQLParser {
@@ -2012,13 +2009,13 @@ public class SQLStatementParser extends SQLParser {
stmt.setName(this.exprParser.name());
if (lexer.identifierEquals("BEFORE")) {
if (lexer.identifierEquals(FnvHash.Constants.BEFORE)) {
stmt.setTriggerType(TriggerType.BEFORE);
lexer.nextToken();
} else if (lexer.identifierEquals("AFTER")) {
} else if (lexer.identifierEquals(FnvHash.Constants.AFTER)) {
stmt.setTriggerType(TriggerType.AFTER);
lexer.nextToken();
} else if (lexer.identifierEquals("INSTEAD")) {
} else if (lexer.identifierEquals(FnvHash.Constants.INSTEAD)) {
lexer.nextToken();
accept(Token.OF);
stmt.setTriggerType(TriggerType.INSTEAD_OF);
@@ -2027,21 +2024,26 @@ public class SQLStatementParser extends SQLParser {
for (;;) {
if (lexer.token == Token.INSERT) {
lexer.nextToken();
stmt.getTriggerEvents().add(TriggerEvent.INSERT);
stmt.setInsert(true);
} else if (lexer.token == Token.UPDATE) {
lexer.nextToken();
stmt.setUpdate(true);
if (lexer.token == Token.OF) {
lexer.nextToken();
this.exprParser.names(stmt.getUpdateOfColumns(), stmt);
}
} else if (lexer.token == Token.DELETE) {
lexer.nextToken();
stmt.setDelete(true);
}
if (lexer.token == Token.COMMA
|| lexer.token == Token.OR) {
lexer.nextToken();
continue;
}
if (lexer.token == Token.UPDATE) {
lexer.nextToken();
stmt.getTriggerEvents().add(TriggerEvent.UPDATE);
continue;
}
if (lexer.token == Token.DELETE) {
lexer.nextToken();
stmt.getTriggerEvents().add(TriggerEvent.DELETE);
continue;
}
break;
}
@@ -2055,6 +2057,12 @@ public class SQLStatementParser extends SQLParser {
stmt.setForEachRow(true);
}
if (lexer.token == Token.WHEN) {
lexer.nextToken();
SQLExpr condition = this.exprParser.expr();
stmt.setWhen(condition);
}
List<SQLStatement> body = this.parseStatementList();
if (body == null || body.isEmpty()) {
throw new ParserException("syntax error");