bug fixed for SchemaStatVisitor. for issue #2306

This commit is contained in:
wenshao
2018-02-04 13:24:57 +08:00
parent 6672fc5cd9
commit 6d9c176450
3 changed files with 68 additions and 4 deletions
@@ -641,11 +641,18 @@ public class SchemaStatVisitor extends SQLASTVisitorAdapter {
if (valueColumn != null) {
continue;
}
Object value = SQLEvalVisitorUtils.eval(dbType, item, parameters, false);
if (value == SQLEvalVisitor.EVAL_VALUE_NULL) {
value = null;
Object value;
if (item instanceof SQLMethodInvokeExpr) {
value = item.toString();
} else {
value = SQLEvalVisitorUtils.eval(dbType, item, parameters, false);
if (value == SQLEvalVisitor.EVAL_VALUE_NULL) {
value = null;
}
}
condition.getValues().add(value);
condition.addValue(value);
}
}
@@ -323,6 +323,10 @@ public class TableStat {
return values;
}
public void addValue(Object value) {
this.values.add(value);
}
@Override
public int hashCode() {
final int prime = 31;
@@ -0,0 +1,53 @@
/*
* 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.mysql.select;
import com.alibaba.druid.sql.MysqlTest;
import com.alibaba.druid.sql.SQLUtils;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
import com.alibaba.druid.util.JdbcConstants;
import java.util.List;
public class MySqlSelectTest_107_between extends MysqlTest {
public void test_0() throws Exception {
String sql = "\n" +
"select * from travelrecord a where a.createtime between date_sub(now(),interval 1 day) and now();";
MySqlStatementParser parser = new MySqlStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
assertEquals(1, statementList.size());
SQLStatement stmt = statementList.get(0);
assertEquals("SELECT *\n" +
"FROM travelrecord a\n" +
"WHERE a.createtime BETWEEN date_sub(now(), INTERVAL 1 DAY) AND now();", stmt.toString());
SchemaStatVisitor visitor = SQLUtils.createSchemaStatVisitor(JdbcConstants.MYSQL);
stmt.accept(visitor);
// System.out.println("Tables : " + visitor.getTables());
// System.out.println("fields : " + visitor.getColumns());
System.out.println("coditions : " + visitor.getConditions());
System.out.println("orderBy : " + visitor.getOrderByColumns());
}
}