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

修正输出格式

Browse Source 
git-svn-id: http://code.alibabatech.com/svn/druid/trunk@159 b9813039-fb51-4c41-a8b9-e21c2acb5095
This commit is contained in:
wenshao
2011-05-20 11:56:32 +00:00
parent edbaa3db38
commit 2b19baff03
18 changed files with 205 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/test/java/com/alibaba/druid/sql/OnlineSQLTest.java
+44 -3
View File
@@ -2,7 +2,9 @@ package com.alibaba.druid.sql;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.List;
@@ -20,20 +22,23 @@ public class OnlineSQLTest extends TestCase {
private String password = "dragoon";
public void test_list_sql() throws Exception {
//reset();
Connection conn = DriverManager.getConnection(url, user, password);
int count = 0;
String sql = "SELECT id, value FROM m_sql_const";
String sql = "SELECT id, value FROM m_sql_const WHERE flag IS NULL LIMIT 1";
Statement stmt = conn.createStatement();
ResultSet rs = stmt.executeQuery(sql);
while (rs.next()) {
int id = rs.getInt(1);
String value = rs.getString(2);
System.out.println(value);
System.out.println();
try {
validate(value);
validate(id, value);
} catch (Exception e) {
e.printStackTrace();
continue;
}
count++;
}
@@ -44,8 +49,43 @@ public class OnlineSQLTest extends TestCase {
conn.close();
}
void reset() throws SQLException {
Connection conn = DriverManager.getConnection(url, user, password);
void validate(String sql) throws Exception {
String sql = "UPDATE m_sql_const SET flag = NULL, value2 = NULL";
PreparedStatement stmt = conn.prepareStatement(sql);
stmt.executeUpdate();
stmt.close();
conn.close();
}
void update(int id, String value2) throws SQLException {
Connection conn = DriverManager.getConnection(url, user, password);
String sql = "UPDATE m_sql_const SET flag = 1, value2 = ? WHERE id = ?";
PreparedStatement stmt = conn.prepareStatement(sql);
stmt.setString(1, value2);
stmt.setInt(2, id);
stmt.executeUpdate();
stmt.close();
conn.close();
}
void validate(int id, String sql) throws Exception {
sql = sql.trim();
boolean sqlFlag = false;
String lowerSql = sql.toLowerCase();
if (lowerSql.startsWith("insert") || lowerSql.startsWith("select") || lowerSql.startsWith("upate") || lowerSql.startsWith("delete") || lowerSql.startsWith("create") || lowerSql.startsWith("drop")) {
sqlFlag = true;
}
if (!sqlFlag) {
update(id, sql);
return;
}
try {
MySqlStatementParser parser = new MySqlStatementParser(sql);
@@ -58,6 +98,7 @@ public class OnlineSQLTest extends TestCase {
MySqlParameterizedOutputVisitor visitor = new MySqlParameterizedOutputVisitor(out);
statemen.accept(visitor);
update(id, out.toString());
System.out.println(sql);
System.out.println(out.toString());
} catch (Exception e) {
src/test/java/com/alibaba/druid/sql/mysql/bvt/CastFunctionsAndOperatorsTest.java
+1 -1
View File
@@ -112,7 +112,7 @@ public class CastFunctionsAndOperatorsTest extends TestCase {
String text = output(stmtList);
Assert.assertEquals("SELECT enum_col\nFROM tbl_name ORDER BY CAST(enum_col AS CHAR);", text);
Assert.assertEquals("SELECT enum_col\nFROM tbl_name\nORDER BY CAST(enum_col AS CHAR);", text);
}
public void test_8() throws Exception {
src/test/java/com/alibaba/druid/sql/mysql/bvt/MathematicalFunctionsTest.java
+1 -1
View File
@@ -310,7 +310,7 @@ public class MathematicalFunctionsTest extends TestCase {
String text = output(stmtList);
Assert.assertEquals("SELECT *\nFROM tbl_name ORDER BY RAND();", text);
Assert.assertEquals("SELECT *\nFROM tbl_name\nORDER BY RAND();", text);
}
public void test_26() throws Exception {
src/test/java/com/alibaba/druid/sql/mysql/bvt/parser/MySqlParserResourceTest.java
+56
View File
@@ -0,0 +1,56 @@
package com.alibaba.druid.sql.mysql.bvt.parser;
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.mysql.parser.MySqlStatementParser;
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlOutputVisitor;
import com.alibaba.druid.util.JdbcUtils;
public class MySqlParserResourceTest extends TestCase {
public void test_0() throws Exception {
exec_test("bvt/parser/mysql-0.txt");
}
public void exec_test(String resource) throws Exception {
System.out.println(resource);
InputStream is = null;
is = Thread.currentThread().getContextClassLoader().getResourceAsStream(resource);
Reader reader = new InputStreamReader(is, "UTF-8");
String input = JdbcUtils.read(reader);
JdbcUtils.close(reader);
String[] items = input.split("---------------------------");
String sql = items[0].trim();
String expect = items[1].trim();
MySqlStatementParser parser = new MySqlStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
Assert.assertEquals(1, statementList.size());
String text = output(statementList);
System.out.println(text);
Assert.assertEquals(expect, text.trim());
}
private String output(List<SQLStatement> stmtList) {
StringBuilder out = new StringBuilder();
MySqlOutputVisitor visitor = new MySqlOutputVisitor(out);
for (SQLStatement stmt : stmtList) {
stmt.accept(visitor);
}
return out.toString();
}
}
src/test/java/com/alibaba/druid/sql/oracle/bvt/OracleSQLParserResourceTest.java
+14 -14
View File
@@ -16,20 +16,20 @@ import com.alibaba.druid.util.JdbcUtils;
public class OracleSQLParserResourceTest extends TestCase {
public void test_0() throws Exception {
// exec_test("bvt/parser/oracle-0.txt");
// exec_test("bvt/parser/oracle-1.txt");
// exec_test("bvt/parser/oracle-2.txt");
// exec_test("bvt/parser/oracle-3.txt");
// exec_test("bvt/parser/oracle-4.txt");
// exec_test("bvt/parser/oracle-5.txt");
// exec_test("bvt/parser/oracle-6.txt"); // PARTITION
// exec_test("bvt/parser/oracle-7.txt");
// exec_test("bvt/parser/oracle-8.txt");
// exec_test("bvt/parser/oracle-9.txt");
// exec_test("bvt/parser/oracle-10.txt");
// exec_test("bvt/parser/oracle-11.txt");
// exec_test("bvt/parser/oracle-12.txt");
// exec_test("bvt/parser/oracle-13.txt");
exec_test("bvt/parser/oracle-0.txt");
exec_test("bvt/parser/oracle-1.txt");
exec_test("bvt/parser/oracle-2.txt");
exec_test("bvt/parser/oracle-3.txt");
exec_test("bvt/parser/oracle-4.txt");
exec_test("bvt/parser/oracle-5.txt");
exec_test("bvt/parser/oracle-6.txt"); // PARTITION
exec_test("bvt/parser/oracle-7.txt");
exec_test("bvt/parser/oracle-8.txt");
exec_test("bvt/parser/oracle-9.txt");
exec_test("bvt/parser/oracle-10.txt");
exec_test("bvt/parser/oracle-11.txt");
exec_test("bvt/parser/oracle-12.txt");
exec_test("bvt/parser/oracle-13.txt");
exec_test("bvt/parser/oracle-14.txt");
}