improved sql parser.

This commit is contained in:
wenshao
2017-10-27 20:31:42 +08:00
parent 1f0c49ec2c
commit 635413abd8
4 changed files with 103 additions and 13 deletions
@@ -16,6 +16,7 @@
package com.alibaba.druid.sql.ast.expr;
import com.alibaba.druid.sql.ast.SQLDataType;
import com.alibaba.druid.sql.ast.SQLExpr;
import com.alibaba.druid.sql.ast.SQLExprImpl;
import com.alibaba.druid.sql.ast.SQLObject;
import com.alibaba.druid.sql.ast.statement.SQLCharacterDataType;
@@ -29,26 +30,36 @@ import java.util.List;
public class SQLDateExpr extends SQLExprImpl implements SQLLiteralExpr, SQLValuableExpr {
public static final SQLDataType DEFAULT_DATA_TYPE = new SQLCharacterDataType("date");
private String literal;
private SQLExpr literal;
public SQLDateExpr(){
}
public SQLDateExpr(String literal) {
this.literal = literal;
this.setLiteral(literal);
}
public String getLiteral() {
public SQLExpr getLiteral() {
return literal;
}
public void setLiteral(String literal) {
this.literal = literal;
setLiteral(new SQLCharExpr(literal));
}
public void setLiteral(SQLExpr x) {
if (x != null) {
x.setParent(this);
}
this.literal = x;
}
public String getValue() {
return literal;
if (literal instanceof SQLCharExpr) {
return ((SQLCharExpr) literal).getText();
}
return null;
}
@Override
@@ -88,7 +99,13 @@ public class SQLDateExpr extends SQLExprImpl implements SQLLiteralExpr, SQLValua
}
public SQLDateExpr clone() {
return new SQLDateExpr(literal);
SQLDateExpr x = new SQLDateExpr();
if (this.literal != null) {
x.setLiteral(literal.clone());
}
return x;
}
@Override
@@ -314,13 +314,11 @@ public class SQLExprParser extends SQLParser {
lexer.nextToken();
if (hash_lower == FnvHash.Constants.DATE
&& lexer.token == Token.LITERAL_CHARS
&& (lexer.token == Token.LITERAL_CHARS || lexer.token == Token.VARIANT)
&& (JdbcConstants.ORACLE.equals(dbType)
|| JdbcConstants.POSTGRESQL.equals(dbType)
|| JdbcConstants.MYSQL.equals(dbType))) {
String literal = lexer.stringVal();
lexer.nextToken();
SQLExpr literal = this.primary();
SQLDateExpr dateExpr = new SQLDateExpr();
dateExpr.setLiteral(literal);
sqlExpr = dateExpr;
@@ -5329,9 +5329,10 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Paramet
}
public boolean visit(SQLDateExpr x) {
print0(ucase ? "DATE '" : "date '");
print0(x.getLiteral());
print('\'');
SQLExpr literal = x.getLiteral();
print0(ucase ? "DATE " : "date ");
printExpr(literal);
return false;
}
@@ -0,0 +1,74 @@
/*
* Copyright 1999-2017 Alibaba Group Holding Ltd.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.alibaba.druid.bvt.sql.oracle.select;
import com.alibaba.druid.sql.OracleTest;
import com.alibaba.druid.sql.SQLUtils;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.ast.statement.SQLSelectStatement;
import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
import com.alibaba.druid.sql.dialect.oracle.visitor.OracleSchemaStatVisitor;
import org.junit.Assert;
import java.util.List;
public class OracleSelectTest100_var extends OracleTest {
public void test_0() throws Exception {
String sql = //
"select * from zj_sb_zzs_fb3_mx where fphm=:\"SYS_B_0\" and sssq_q=date:\"SYS_B_1\"\n";
System.out.println(sql);
OracleStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
SQLSelectStatement stmt = (SQLSelectStatement) statementList.get(0);
System.out.println(stmt.toString());
Assert.assertEquals(1, statementList.size());
// SQLMethodInvokeExpr expr = (SQLMethodInvokeExpr) stmt.getSelect().getQueryBlock().getSelectList().get(0).getExpr();
// SQLMethodInvokeExpr param0 = (SQLMethodInvokeExpr) expr.getParameters().get(0);
// assertTrue(param0.getParameters().get(0)
// instanceof SQLAggregateExpr);
OracleSchemaStatVisitor visitor = new OracleSchemaStatVisitor();
stmt.accept(visitor);
{
String text = SQLUtils.toOracleString(stmt);
assertEquals("SELECT *\n" +
"FROM zj_sb_zzs_fb3_mx\n" +
"WHERE fphm = :\"SYS_B_0\"\n" +
"\tAND sssq_q = DATE :\"SYS_B_1\"", text);
}
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getColumns());
System.out.println("coditions : " + visitor.getConditions());
System.out.println("relationships : " + visitor.getRelationships());
System.out.println("orderBy : " + visitor.getOrderByColumns());
assertEquals(1, visitor.getTables().size());
assertEquals(3, visitor.getColumns().size());
assertEquals(2, visitor.getConditions().size());
assertEquals(0, visitor.getRelationships().size());
assertEquals(0, visitor.getOrderByColumns().size());
}
}