From 556f8a76d4ca2c8edda75a29974d3413686d09cc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=AB=98=E9=93=81?= <shaojin.wensj@alibaba-inc.com>
Date: Sun, 26 Jun 2016 00:20:02 +0800
Subject: [PATCH] SchemaStatVisitor support cast expr

---
 .../druid/sql/visitor/SchemaStatVisitor.java  |  7 +++-
 .../bvt/sql/schemaStat/SchemaStatTest7.java   | 32 +++++++++++++++++++
 2 files changed, 38 insertions(+), 1 deletion(-)
 create mode 100644 src/test/java/com/alibaba/druid/bvt/sql/schemaStat/SchemaStatTest7.java

diff --git a/src/main/java/com/alibaba/druid/sql/visitor/SchemaStatVisitor.java b/src/main/java/com/alibaba/druid/sql/visitor/SchemaStatVisitor.java
index 7f3989f844..92a3d31978 100644
--- a/src/main/java/com/alibaba/druid/sql/visitor/SchemaStatVisitor.java
+++ b/src/main/java/com/alibaba/druid/sql/visitor/SchemaStatVisitor.java
@@ -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)) {
diff --git a/src/test/java/com/alibaba/druid/bvt/sql/schemaStat/SchemaStatTest7.java b/src/test/java/com/alibaba/druid/bvt/sql/schemaStat/SchemaStatTest7.java
new file mode 100644
index 0000000000..c8315384fd
--- /dev/null
+++ b/src/test/java/com/alibaba/druid/bvt/sql/schemaStat/SchemaStatTest7.java
@@ -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());
+    }
+}