mirror of
https://github.com/alibaba/druid.git
synced 2023-03-04 19:17:16 +00:00
SchemaStatVisitor support cast expr
This commit is contained in:
@@ -42,6 +42,7 @@ import com.alibaba.druid.sql.ast.expr.SQLAggregateExpr;
|
||||
import com.alibaba.druid.sql.ast.expr.SQLAllColumnExpr;
|
||||
import com.alibaba.druid.sql.ast.expr.SQLArrayExpr;
|
||||
import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
|
||||
import com.alibaba.druid.sql.ast.expr.SQLCastExpr;
|
||||
import com.alibaba.druid.sql.ast.expr.SQLCurrentOfCursorExpr;
|
||||
import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
|
||||
import com.alibaba.druid.sql.ast.expr.SQLInListExpr;
|
||||
@@ -449,11 +450,15 @@ public class SchemaStatVisitor extends SQLASTVisitorAdapter {
|
||||
}
|
||||
|
||||
protected void handleCondition(SQLExpr expr, String operator, SQLExpr... valueExprs) {
|
||||
if (expr instanceof SQLCastExpr) {
|
||||
expr = ((SQLCastExpr) expr).getExpr();
|
||||
}
|
||||
|
||||
Column column = getColumn(expr);
|
||||
if (column == null) {
|
||||
return;
|
||||
}
|
||||
|
||||
|
||||
Condition condition = null;
|
||||
for (Condition item : this.getConditions()) {
|
||||
if (item.getColumn().equals(column) && item.getOperator().equals(operator)) {
|
||||
|
||||
@@ -0,0 +1,32 @@
|
||||
package com.alibaba.druid.bvt.sql.schemaStat;
|
||||
|
||||
import org.junit.Assert;
|
||||
|
||||
import com.alibaba.druid.sql.SQLUtils;
|
||||
import com.alibaba.druid.sql.ast.SQLStatement;
|
||||
import com.alibaba.druid.sql.parser.SQLParserUtils;
|
||||
import com.alibaba.druid.sql.parser.SQLStatementParser;
|
||||
import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
|
||||
import com.alibaba.druid.util.JdbcConstants;
|
||||
|
||||
import junit.framework.TestCase;
|
||||
|
||||
public class SchemaStatTest7 extends TestCase {
|
||||
|
||||
public void test_schemaStat() throws Exception {
|
||||
String sql = "select t.a,t.b from t where cast(t.statdate as decimal) >=20160626";
|
||||
|
||||
String dbType = JdbcConstants.MYSQL;
|
||||
SQLStatementParser parser = SQLParserUtils.createSQLStatementParser(sql, dbType);
|
||||
SQLStatement stmt = parser.parseStatementList().get(0);
|
||||
|
||||
SchemaStatVisitor statVisitor = SQLUtils.createSchemaStatVisitor(dbType);
|
||||
stmt.accept(statVisitor);
|
||||
|
||||
// System.out.println(statVisitor.getColumns());
|
||||
// System.out.println(statVisitor.getGroupByColumns()); // group by
|
||||
|
||||
Assert.assertEquals(3, statVisitor.getColumns().size());
|
||||
Assert.assertEquals(1, statVisitor.getConditions().size());
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user