mirror of
https://github.com/alibaba/druid.git
synced 2023-03-04 19:17:16 +00:00
db2 sql parser support concat binary operator
This commit is contained in:
@@ -37,6 +37,7 @@ public class DB2Lexer extends Lexer {
|
||||
map.put("ONLY", Token.ONLY);
|
||||
map.put("OPTIMIZE", Token.OPTIMIZE);
|
||||
map.put("OF", Token.OF);
|
||||
map.put("CONCAT", Token.CONCAT);
|
||||
|
||||
DEFAULT_DB2_KEYWORDS = new Keywords(map);
|
||||
}
|
||||
|
||||
@@ -17,6 +17,7 @@ package com.alibaba.druid.sql.dialect.db2.visitor;
|
||||
|
||||
import com.alibaba.druid.sql.ast.SQLObject;
|
||||
import com.alibaba.druid.sql.ast.SQLOrderBy;
|
||||
import com.alibaba.druid.sql.ast.expr.SQLBinaryOperator;
|
||||
import com.alibaba.druid.sql.ast.statement.SQLSelect;
|
||||
import com.alibaba.druid.sql.ast.statement.SQLSelectQueryBlock;
|
||||
import com.alibaba.druid.sql.dialect.db2.ast.stmt.DB2SelectQueryBlock;
|
||||
@@ -91,4 +92,12 @@ public class DB2OutputVisitor extends SQLASTOutputVisitor implements DB2ASTVisit
|
||||
public void endVisit(DB2ValuesStatement x) {
|
||||
|
||||
}
|
||||
|
||||
protected void printOperator(SQLBinaryOperator operator) {
|
||||
if (operator == SQLBinaryOperator.Concat) {
|
||||
print0(ucase ? "CONCAT" : "concat");
|
||||
} else {
|
||||
print0(ucase ? operator.name : operator.name_lcase);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1223,7 +1223,7 @@ public class SQLExprParser extends SQLParser {
|
||||
|
||||
expr = new SQLBinaryOpExpr(expr, SQLBinaryOperator.Add, rightExp, getDbType());
|
||||
expr = additiveRest(expr);
|
||||
} else if (lexer.token() == Token.BARBAR) {
|
||||
} else if (lexer.token() == Token.BARBAR || lexer.token() == Token.CONCAT) {
|
||||
lexer.nextToken();
|
||||
SQLExpr rightExp = multiplicative();
|
||||
expr = new SQLBinaryOpExpr(expr, SQLBinaryOperator.Concat, rightExp, getDbType());
|
||||
|
||||
@@ -277,6 +277,8 @@ public enum Token {
|
||||
FORMAT("FORMAT"),
|
||||
QUALIFY("QUALIFY"),
|
||||
MOD("MOD"),
|
||||
|
||||
CONCAT("CONCAT"), // DB2
|
||||
|
||||
LPAREN("("),
|
||||
RPAREN(")"),
|
||||
|
||||
@@ -329,7 +329,7 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Printab
|
||||
} else {
|
||||
print0(" ");
|
||||
}
|
||||
print0(ucase ? x.getOperator().name : x.getOperator().name_lcase);
|
||||
printOperator(x.getOperator());
|
||||
print0(" ");
|
||||
}
|
||||
|
||||
@@ -342,6 +342,10 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Printab
|
||||
return false;
|
||||
}
|
||||
|
||||
protected void printOperator(SQLBinaryOperator operator) {
|
||||
print0(ucase ? operator.name : operator.name_lcase);
|
||||
}
|
||||
|
||||
private void visitorBinaryRight(SQLBinaryOpExpr x) {
|
||||
if (isPrettyFormat() && x.getRight().hasBeforeComment()) {
|
||||
printlnComments(x.getRight().getBeforeCommentsDirect());
|
||||
|
||||
Reference in New Issue
Block a user