mirror of
https://github.com/alibaba/druid.git
synced 2023-03-04 19:17:16 +00:00
增加mysql创建存储过程时参数可选类型IN|OUT|INOUT
This commit is contained in:
@@ -32,8 +32,17 @@ public class MySqlParameter extends MySqlObjectImpl {
|
||||
private SQLExpr name;
|
||||
private SQLDataType dataType;
|
||||
private SQLExpr defaultValue;
|
||||
private ParameterType paramType;
|
||||
|
||||
public SQLExpr getDefaultValue() {
|
||||
public ParameterType getParamType() {
|
||||
return paramType;
|
||||
}
|
||||
|
||||
public void setParamType(ParameterType paramType) {
|
||||
this.paramType = paramType;
|
||||
}
|
||||
|
||||
public SQLExpr getDefaultValue() {
|
||||
return defaultValue;
|
||||
}
|
||||
|
||||
@@ -66,5 +75,15 @@ public class MySqlParameter extends MySqlObjectImpl {
|
||||
}
|
||||
visitor.endVisit(this);
|
||||
}
|
||||
/**
|
||||
* mysql procedure parameter type
|
||||
*/
|
||||
public static enum ParameterType
|
||||
{
|
||||
DEFAULT,
|
||||
IN,//in
|
||||
OUT,//out
|
||||
INOUT//inout
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
@@ -73,6 +73,7 @@ import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlIterateStatement;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlLeaveStatement;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlLoopStatement;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlParameter;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlParameter.ParameterType;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlRepeatStatement;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlSelectIntoStatement;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlWhileStatement;
|
||||
@@ -2785,7 +2786,28 @@ public class MySqlStatementParser extends SQLStatementParser {
|
||||
parameter.setDataType(dataType);
|
||||
|
||||
parameter.setDefaultValue(new SQLQueryExpr(select));
|
||||
|
||||
} else if (lexer.token() == Token.IN || lexer.token() == Token.OUT || lexer.token() == Token.INOUT) {
|
||||
|
||||
if(lexer.token()==Token.IN)
|
||||
{
|
||||
parameter.setParamType(ParameterType.IN);
|
||||
}
|
||||
else if(lexer.token()==Token.OUT)
|
||||
{
|
||||
parameter.setParamType(ParameterType.OUT);
|
||||
}
|
||||
else if(lexer.token()==Token.INOUT)
|
||||
{
|
||||
parameter.setParamType(ParameterType.INOUT);
|
||||
}
|
||||
lexer.nextToken();
|
||||
|
||||
parameter.setName(this.exprParser.name());
|
||||
|
||||
parameter.setDataType(this.exprParser.parseDataType());
|
||||
} else {
|
||||
parameter.setParamType(ParameterType.DEFAULT);//default parameter type is in
|
||||
parameter.setName(this.exprParser.name());
|
||||
parameter.setDataType(this.exprParser.parseDataType());
|
||||
|
||||
|
||||
@@ -59,6 +59,7 @@ import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlIterateStatement;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlLeaveStatement;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlLoopStatement;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlParameter;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlParameter.ParameterType;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlRepeatStatement;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlSelectIntoStatement;
|
||||
import com.alibaba.druid.sql.dialect.mysql.ast.clause.MySqlWhileStatement;
|
||||
@@ -3277,6 +3278,19 @@ public class MySqlOutputVisitor extends SQLASTOutputVisitor implements MySqlASTV
|
||||
decrementIndent();
|
||||
|
||||
} else {
|
||||
|
||||
if(x.getParamType()==ParameterType.IN)
|
||||
{
|
||||
print("IN ");
|
||||
}
|
||||
else if(x.getParamType()==ParameterType.OUT)
|
||||
{
|
||||
print("OUT ");
|
||||
}
|
||||
else if(x.getParamType()==ParameterType.INOUT)
|
||||
{
|
||||
print("INOUT ");
|
||||
}
|
||||
x.getName().accept(this);
|
||||
print(" ");
|
||||
|
||||
|
||||
@@ -135,7 +135,7 @@ public class Keywords {
|
||||
map.put("GRANT", Token.GRANT);
|
||||
map.put("REVOKE", Token.REVOKE);
|
||||
|
||||
// MySql procedure by:zz
|
||||
// MySql procedure: add by zz
|
||||
map.put("WHILE", Token.WHILE);
|
||||
map.put("DO", Token.DO);
|
||||
map.put("DECLARE", Token.DECLARE);
|
||||
@@ -148,6 +148,8 @@ public class Keywords {
|
||||
map.put("CLOSE", Token.CLOSE);
|
||||
map.put("CURSOR", Token.CURSOR);
|
||||
map.put("FETCH", Token.FETCH);
|
||||
map.put("OUT", Token.OUT);
|
||||
map.put("INOUT", Token.INOUT);
|
||||
|
||||
DEFAULT_KEYWORDS = new Keywords(map);
|
||||
}
|
||||
|
||||
@@ -128,7 +128,7 @@ public enum Token {
|
||||
SHOW("SHOW"),
|
||||
REPLACE("REPLACE"),
|
||||
|
||||
// MySql procedure by:zz
|
||||
// MySql procedure add by zz
|
||||
WHILE("WHILE"),
|
||||
DO("DO"),
|
||||
LEAVE("LEAVE"),
|
||||
@@ -137,6 +137,8 @@ public enum Token {
|
||||
UNTIL("UNTIL"),
|
||||
OPEN("OPEN"),
|
||||
CLOSE("CLOSE"),
|
||||
OUT("OUT"),
|
||||
INOUT("INOUT"),
|
||||
|
||||
//postgresql
|
||||
WINDOW("WINDOW"),
|
||||
|
||||
Reference in New Issue
Block a user