mirror of
https://github.com/alibaba/druid.git
synced 2023-03-04 19:17:16 +00:00
sql parser support TimeStamp Literal.
This commit is contained in:
@@ -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);
|
||||
|
||||
Reference in New Issue
Block a user