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 support mysql create procedure.

Browse Source
This commit is contained in:
wenshao
2017-04-25 10:35:19 +08:00
parent e3a868ec4f
commit 1d99c00f2c
2 changed files with 69 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/sql/dialect/mysql/parser/MySqlStatementParser.java
+19 -11
View File
@@ -3195,17 +3195,19 @@ public class MySqlStatementParser extends SQLStatementParser {
*/
SQLCreateProcedureStatement stmt = new SQLCreateProcedureStatement();
if (identifierEquals("DEFINER")) {
lexer.nextToken();
accept(Token.EQ);
SQLName definer = this.exprParser.name();
stmt.setDefiner(definer);
} else {
accept(Token.CREATE);
if (lexer.token() == Token.OR) {
if (lexer.token() != Token.PROCEDURE) {
if (identifierEquals("DEFINER")) {
lexer.nextToken();
accept(Token.REPLACE);
stmt.setOrReplace(true);
accept(Token.EQ);
SQLName definer = this.exprParser.name();
stmt.setDefiner(definer);
} else {
accept(Token.CREATE);
if (lexer.token() == Token.OR) {
lexer.nextToken();
accept(Token.REPLACE);
stmt.setOrReplace(true);
}
}
}
@@ -3218,7 +3220,13 @@ public class MySqlStatementParser extends SQLStatementParser {
parserParameters(stmt.getParameters());
accept(Token.RPAREN);// match ")"
}
SQLBlockStatement block = this.parseBlock();
SQLStatement block;
if (lexer.token() == Token.BEGIN) {
block = this.parseBlock();
} else {
block = this.parseStatement();
}
stmt.setBlock(block);