sql parser support TimeStamp Literal.

This commit is contained in:
wenshao
2018-02-07 19:16:31 +08:00
parent acf0bbbc98
commit 307480fe48
6 changed files with 108 additions and 0 deletions
@@ -35,6 +35,11 @@ public class SQLTimestampExpr extends SQLExprImpl implements SQLValuableExpr {
}
public SQLTimestampExpr(String literal){
this.literal = literal;
}
public SQLTimestampExpr clone() {
SQLTimestampExpr x = new SQLTimestampExpr();
x.literal = literal;
@@ -323,6 +323,12 @@ public class SQLExprParser extends SQLParser {
SQLDateExpr dateExpr = new SQLDateExpr();
dateExpr.setLiteral(literal);
sqlExpr = dateExpr;
} else if (hash_lower == FnvHash.Constants.TIMESTAMP
&& (lexer.token == Token.LITERAL_CHARS || lexer.token == Token.VARIANT)
&& !JdbcConstants.ORACLE.equals(dbType)) {
SQLTimestampExpr dateExpr = new SQLTimestampExpr(lexer.stringVal());
lexer.nextToken();
sqlExpr = dateExpr;
} else if (JdbcConstants.MYSQL.equalsIgnoreCase(dbType) && ident.startsWith("0x") && (ident.length() % 2) == 0) {
sqlExpr = new SQLHexExpr(ident.substring(2));
} else {
@@ -97,6 +97,9 @@ public final class ExportParameterVisitorUtils {
} else if (param instanceof SQLHexExpr) {
value = ((SQLHexExpr) param).toBytes();
replace = true;
} else if (param instanceof SQLTimestampExpr || param instanceof SQLDateExpr) {
value = ((SQLTimestampExpr) param).getValue();
replace = true;
} else if (param instanceof SQLListExpr) {
SQLListExpr list = ((SQLListExpr) param);
@@ -17,6 +17,7 @@ package com.alibaba.druid.sql.visitor;
import java.util.List;
import com.alibaba.druid.sql.SQLUtils;
import com.alibaba.druid.sql.ast.*;
import com.alibaba.druid.sql.ast.statement.*;
import com.alibaba.druid.sql.dialect.db2.visitor.DB2OutputVisitor;
@@ -287,4 +288,18 @@ public class ParameterizedOutputVisitorUtils {
return new SQLASTOutputVisitor(out, true);
}
public static String restore(String sql, String dbType, List<Object> parameters) {
List<SQLStatement> stmtList = SQLUtils.parseStatements(sql, dbType);
StringBuilder out = new StringBuilder();
SQLASTOutputVisitor visitor = SQLUtils.createOutputVisitor(out, dbType);
visitor.setInputParameters(parameters);
for (SQLStatement stmt : stmtList) {
stmt.accept(visitor);
}
return out.toString();
}
}
@@ -4278,6 +4278,16 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Paramet
@Override
public boolean visit(SQLTimestampExpr x) {
if (this.parameterized) {
print('?');
incrementReplaceCunt();
if(this.parameters != null){
ExportParameterVisitorUtils.exportParameter(this.parameters, x);
}
return false;
}
print0(ucase ? "TIMESTAMP " : "timestamp ");
if (x.isWithTimeZone()) {
@@ -5537,6 +5547,16 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Paramet
}
public boolean visit(SQLDateExpr x) {
if (this.parameterized) {
print('?');
incrementReplaceCunt();
if(this.parameters != null){
ExportParameterVisitorUtils.exportParameter(this.parameters, x);
}
return false;
}
SQLExpr literal = x.getLiteral();
print0(ucase ? "DATE " : "date ");
printExpr(literal);