Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

refactor

Browse Source
This commit is contained in:
温少
2012-05-05 19:30:00 +08:00
parent 563ba6dd3b
commit 96993b5889
8 changed files with 222 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/test/java/com/alibaba/druid/benckmark/pool/Oracle_Case4.java
+3 -2
View File
@@ -30,7 +30,7 @@ public class Oracle_Case4 extends TestCase {
private int maxWait = 5000;
private String validationQuery = "SELECT 1 FROM DUAL";
private int threadCount = 1;
private int loopCount = 3;
private int loopCount = 5;
final int LOOP_COUNT = 1000 * 1;
private boolean testOnBorrow = false;
private boolean preparedStatementCache = true;
@@ -69,6 +69,7 @@ public class Oracle_Case4 extends TestCase {
dataSource.setTestOnBorrow(testOnBorrow);
dataSource.setConnectionProperties(properties);
dataSource.setUseOracleImplicitCache(true);
dataSource.init();
// printAV_INFO(dataSource);
// printTables(dataSource);
@@ -155,7 +156,7 @@ public class Oracle_Case4 extends TestCase {
int mod = i % 500;
String sql = SQL + " AND ROWNUM <= " + (mod + 1);
String sql = SQL; // + " AND ROWNUM <= " + (mod + 1);
PreparedStatement stmt = conn.prepareStatement(sql);
stmt.setInt(1, 61);
ResultSet rs = stmt.executeQuery();
src/test/java/com/alibaba/druid/bvt/sql/mysql/visitor/MySqlResourceTest.java
+15 -11
View File
@@ -19,17 +19,21 @@ import com.alibaba.druid.util.JdbcUtils;
public class MySqlResourceTest extends TestCase {
public void test_0() throws Exception {
exec_test("bvt/parser/mysql-0.txt");
exec_test("bvt/parser/mysql-1.txt");
exec_test("bvt/parser/mysql-2.txt");
exec_test("bvt/parser/mysql-3.txt");
exec_test("bvt/parser/mysql-4.txt");
exec_test("bvt/parser/mysql-5.txt");
exec_test("bvt/parser/mysql-6.txt");
exec_test("bvt/parser/mysql-7.txt");
exec_test("bvt/parser/mysql-8.txt");
exec_test("bvt/parser/mysql-9.txt");
exec_test("bvt/parser/mysql-10.txt");
// exec_test("bvt/parser/mysql-0.txt");
// exec_test("bvt/parser/mysql-1.txt");
// exec_test("bvt/parser/mysql-2.txt");
// exec_test("bvt/parser/mysql-3.txt");
// exec_test("bvt/parser/mysql-4.txt");
// exec_test("bvt/parser/mysql-5.txt");
// exec_test("bvt/parser/mysql-6.txt");
// exec_test("bvt/parser/mysql-7.txt");
// exec_test("bvt/parser/mysql-8.txt");
// exec_test("bvt/parser/mysql-9.txt");
// exec_test("bvt/parser/mysql-10.txt");
// exec_test("bvt/parser/mysql-11.txt");
// exec_test("bvt/parser/mysql-12.txt");
// exec_test("bvt/parser/mysql-13.txt");
exec_test("bvt/parser/mysql-14.txt");
}
public void exec_test(String resource) throws Exception {
src/test/java/com/alibaba/druid/sql/oracle/demo/Demo1.java
+147
View File
@@ -0,0 +1,147 @@
package com.alibaba.druid.sql.oracle.demo;
import java.util.ArrayList;
import java.util.List;
import junit.framework.TestCase;
import com.alibaba.druid.sql.SQLUtils;
import com.alibaba.druid.sql.ast.SQLExpr;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
import com.alibaba.druid.sql.ast.expr.SQLPropertyExpr;
import com.alibaba.druid.sql.ast.expr.SQLVariantRefExpr;
import com.alibaba.druid.sql.ast.statement.SQLExprTableSource;
import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleSelectTableReference;
import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
import com.alibaba.druid.sql.dialect.oracle.visitor.OracleASTVisitorAdapter;
import com.alibaba.druid.sql.parser.SQLStatementParser;
public class Demo1 extends TestCase {
public void test_0() throws Exception {
String sql = "select * from user where uid = ? and uname = ?";
List<Object> parameters = new ArrayList<Object>();
parameters.add(1);
parameters.add("wenshao");
SQLStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> stmtList = parser.parseStatementList(); //
SQLStatement first = (SQLStatement) stmtList.get(0);
GetVariantVisitor variantVisitor = new GetVariantVisitor();
first.accept(variantVisitor);
SQLVariantRefExpr firstVar = variantVisitor.getVariantList().get(0);
int varIndex = (Integer) firstVar.getAttribute("varIndex");
Integer param = (Integer) parameters.get(varIndex);
String tableName;
if (param.intValue() == 1) {
tableName = "user_1";
} else {
tableName = "user_x";
}
MyOracleVisitor visitor = new MyOracleVisitor(tableName);
first.accept(visitor);
String realSql = SQLUtils.toOracleString(first);
System.out.println(realSql);
}
private static class GetVariantVisitor extends OracleASTVisitorAdapter {
private int varIndex = 0;
private List<SQLVariantRefExpr> variantList = new ArrayList<SQLVariantRefExpr>();
public boolean visit(SQLVariantRefExpr x) {
x.getAttributes().put("varIndex", varIndex++);
return true;
}
public boolean visit(SQLBinaryOpExpr x) {
if (x.getLeft() instanceof SQLIdentifierExpr && x.getRight() instanceof SQLVariantRefExpr) {
SQLIdentifierExpr identExpr = (SQLIdentifierExpr) x.getLeft();
String ident = identExpr.getName();
if (ident.equals("uid")) {
variantList.add((SQLVariantRefExpr) x.getRight());
}
}
return true;
}
public int getVarIndex() {
return varIndex;
}
public void setVarIndex(int varIndex) {
this.varIndex = varIndex;
}
public List<SQLVariantRefExpr> getVariantList() {
return variantList;
}
public void setVariantList(List<SQLVariantRefExpr> variantList) {
this.variantList = variantList;
}
}
private static class MyOracleVisitor extends OracleASTVisitorAdapter {
private String tableName;
public MyOracleVisitor(String tableName){
this.tableName = tableName;
}
public boolean visit(OracleSelectTableReference x) {
SQLExpr expr = x.getExpr();
if (expr instanceof SQLIdentifierExpr) {
SQLIdentifierExpr identExpr = (SQLIdentifierExpr) expr;
String tableName = identExpr.getName();
if (tableName.equals("user")) {
identExpr.setName(this.tableName);
}
} else if (expr instanceof SQLPropertyExpr) {
SQLPropertyExpr proExpr = (SQLPropertyExpr) expr;
String tableName = proExpr.getName();
if (tableName.equals("user")) {
proExpr.setName(this.tableName);
}
}
return true;
}
public boolean visit(SQLExprTableSource x) {
SQLExpr expr = x.getExpr();
if (expr instanceof SQLIdentifierExpr) {
SQLIdentifierExpr identExpr = (SQLIdentifierExpr) expr;
String tableName = identExpr.getName();
if (tableName.equals("user")) {
identExpr.setName(this.tableName);
}
} else if (expr instanceof SQLPropertyExpr) {
SQLPropertyExpr proExpr = (SQLPropertyExpr) expr;
String tableName = proExpr.getName();
if (tableName.equals("user")) {
proExpr.setName(this.tableName);
}
}
return true;
}
}
}