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

git-svn-id: http://code.alibabatech.com/svn/druid/trunk@139 b9813039-fb51-4c41-a8b9-e21c2acb5095

Browse Source
This commit is contained in:
wenshao
2011-05-19 12:42:17 +00:00
parent 5b5afbab78
commit 50bf093e41
4 changed files with 97 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/test/java/com/alibaba/druid/sql/oracle/bvt/OracleSQLParserResourceTest.java
-76
View File
@@ -1,76 +0,0 @@
package com.alibaba.druid.sql.oracle.bvt;
import java.util.List;
import junit.framework.Assert;
import junit.framework.TestCase;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
import com.alibaba.druid.sql.ast.expr.SQLBinaryOperator;
import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleUpdateSetListClause;
import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleUpdateSetListSingleColumnItem;
import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleUpdateStatement;
import com.alibaba.druid.sql.dialect.oracle.ast.visitor.OracleOutputVisitor;
import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
public class OracleSQLParserResourceTest extends TestCase {
public void test_0() throws Exception {
String sql = "UPDATE employees SET salary = salary * 1.1 WHERE employee_id IN (SELECT employee_id FROM job_history);";
OracleStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
Assert.assertEquals(1, statementList.size());
OracleUpdateStatement statement = (OracleUpdateStatement) statementList.get(0);
Assert.assertEquals(0, statement.getHints().size());
OracleUpdateSetListClause setList = (OracleUpdateSetListClause) statement.getSetClause();
Assert.assertEquals(1, setList.getItems().size());
OracleUpdateSetListSingleColumnItem item = (OracleUpdateSetListSingleColumnItem) setList.getItems().get(0);
SQLIdentifierExpr column = (SQLIdentifierExpr) item.getColumn();
Assert.assertEquals("salary", column.getName());
SQLBinaryOpExpr value = (SQLBinaryOpExpr) item.getValue();
Assert.assertEquals(SQLBinaryOperator.Multiply, value.getOperator());
String text = output(statementList);
System.out.println(text);
}
public void test_1() throws Exception {
String sql = "SELECT employees_seq.nextval FROM DUAL;";
OracleStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
Assert.assertEquals(1, statementList.size());
String text = output(statementList);
System.out.println(text);
}
public void test_2() throws Exception {
String sql = "SELECT LPAD(' ',2*(LEVEL-1)) || last_name org_chart, employee_id, manager_id, job_id FROM employees WHERE job_id != 'FI_MGR' START WITH job_id = 'AD_VP' CONNECT BY PRIOR employee_id = manager_id; ";
OracleStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
Assert.assertEquals(1, statementList.size());
String text = output(statementList);
System.out.println(text);
}
private String output(List<SQLStatement> stmtList) {
StringBuilder out = new StringBuilder();
OracleOutputVisitor visitor = new OracleOutputVisitor(out);
for (SQLStatement stmt : stmtList) {
stmt.accept(visitor);
}
return out.toString();
}
}
src/test/java/com/alibaba/druid/sql/oracle/bvt/OracleSQLParserResourceTest2.java
+62
View File
@@ -0,0 +1,62 @@
package com.alibaba.druid.sql.oracle.bvt;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.util.List;
import junit.framework.Assert;
import junit.framework.TestCase;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.dialect.oracle.ast.visitor.OracleOutputVisitor;
import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
import com.alibaba.druid.util.JdbcUtils;
public class OracleSQLParserResourceTest2 extends TestCase {
public void f_test_1() throws Exception {
InputStream is = null;
is = Thread.currentThread().getContextClassLoader().getResourceAsStream("bvt/parser/oracle-0.txt");
Reader reader = new InputStreamReader(is);
String sql = JdbcUtils.read(reader);
JdbcUtils.close(reader);
OracleStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
Assert.assertEquals(1, statementList.size());
String text = output(statementList);
System.out.println(text);
}
public void test_2() throws Exception {
InputStream is = null;
is = Thread.currentThread().getContextClassLoader().getResourceAsStream("bvt/parser/oracle-1.txt");
Reader reader = new InputStreamReader(is);
String sql = JdbcUtils.read(reader);
JdbcUtils.close(reader);
OracleStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
Assert.assertEquals(1, statementList.size());
String text = output(statementList);
System.out.println(text);
}
private String output(List<SQLStatement> stmtList) {
StringBuilder out = new StringBuilder();
OracleOutputVisitor visitor = new OracleOutputVisitor(out);
for (SQLStatement stmt : stmtList) {
stmt.accept(visitor);
}
return out.toString();
}
}
src/test/java/com/alibaba/druid/sql/oracle/bvt/OracleSQLParserTest.java
+33 -19
View File
@@ -1,28 +1,47 @@
package com.alibaba.druid.sql.oracle.bvt;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.util.List;
import junit.framework.Assert;
import junit.framework.TestCase;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.ast.expr.SQLBinaryOpExpr;
import com.alibaba.druid.sql.ast.expr.SQLBinaryOperator;
import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleUpdateSetListClause;
import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleUpdateSetListSingleColumnItem;
import com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleUpdateStatement;
import com.alibaba.druid.sql.dialect.oracle.ast.visitor.OracleOutputVisitor;
import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
import com.alibaba.druid.util.JdbcUtils;
public class OracleSQLParserTest extends TestCase {
public void f_test_1() throws Exception {
InputStream is = null;
public void test_0() throws Exception {
String sql = "UPDATE employees SET salary = salary * 1.1 WHERE employee_id IN (SELECT employee_id FROM job_history);";
OracleStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
Assert.assertEquals(1, statementList.size());
OracleUpdateStatement statement = (OracleUpdateStatement) statementList.get(0);
Assert.assertEquals(0, statement.getHints().size());
OracleUpdateSetListClause setList = (OracleUpdateSetListClause) statement.getSetClause();
Assert.assertEquals(1, setList.getItems().size());
OracleUpdateSetListSingleColumnItem item = (OracleUpdateSetListSingleColumnItem) setList.getItems().get(0);
SQLIdentifierExpr column = (SQLIdentifierExpr) item.getColumn();
Assert.assertEquals("salary", column.getName());
SQLBinaryOpExpr value = (SQLBinaryOpExpr) item.getValue();
Assert.assertEquals(SQLBinaryOperator.Multiply, value.getOperator());
String text = output(statementList);
System.out.println(text);
}
public void test_1() throws Exception {
String sql = "SELECT employees_seq.nextval FROM DUAL;";
is = Thread.currentThread().getContextClassLoader().getResourceAsStream("bvt/parser/oracle-0.txt");
Reader reader = new InputStreamReader(is);
String sql = JdbcUtils.read(reader);
JdbcUtils.close(reader);
OracleStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
@@ -31,15 +50,10 @@ public class OracleSQLParserTest extends TestCase {
String text = output(statementList);
System.out.println(text);
}
public void test_2() throws Exception {
InputStream is = null;
is = Thread.currentThread().getContextClassLoader().getResourceAsStream("bvt/parser/oracle-1.txt");
Reader reader = new InputStreamReader(is);
String sql = JdbcUtils.read(reader);
JdbcUtils.close(reader);
public void test_2() throws Exception {
String sql = "SELECT LPAD(' ',2*(LEVEL-1)) || last_name org_chart, employee_id, manager_id, job_id FROM employees WHERE job_id != 'FI_MGR' START WITH job_id = 'AD_VP' CONNECT BY PRIOR employee_id = manager_id; ";
OracleStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
src/test/resources/bvt/parser/oracle-2.txt
+2
View File
@@ -0,0 +1,2 @@
SELECT employee_id FROM (SELECT * FROM employees)
FOR UPDATE OF employee_id;