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

add ibatis parameters support

Browse Source
This commit is contained in:
温少
2012-03-15 13:45:20 +08:00
parent 5f80737ead
commit 7c62f95ce7
4 changed files with 87 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/sql/dialect/oracle/parser/OracleLexer.java
+1 -1
View File
@@ -162,7 +162,7 @@ public class OracleLexer extends Lexer {
return;
}
if (ch != ':') {
if (ch != ':' && ch != '#') {
throw new SQLParseException("illegal variable");
}
src/main/java/com/alibaba/druid/sql/parser/Lexer.java
+4 -1
View File
@@ -277,6 +277,9 @@ public class Lexer {
}
}
return;
case '#':
scanVariable();
return;
case '.':
scanChar();
if (isDigit(ch)) {
@@ -551,7 +554,7 @@ public class Lexer {
public void scanVariable() {
final char first = ch;
if (ch != '@' && ch != ':') {
if (ch != '@' && ch != ':' && ch != '#') {
throw new SQLParseException("illegal variable");
}
src/test/java/com/alibaba/druid/bvt/sql/oracle/OracleIbatisSelectTest.java
+41
View File
@@ -0,0 +1,41 @@
package com.alibaba.druid.bvt.sql.oracle;
import java.util.List;
import junit.framework.Assert;
import com.alibaba.druid.sql.OracleTest;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
import com.alibaba.druid.sql.dialect.oracle.visitor.OracleSchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
public class OracleIbatisSelectTest extends OracleTest {
public void test_0() throws Exception {
String sql = "SELECT * FROM T WHERE F1 = #id# ";
OracleStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
SQLStatement statemen = statementList.get(0);
print(statementList);
Assert.assertEquals(1, statementList.size());
OracleSchemaStatVisitor visitor = new OracleSchemaStatVisitor();
statemen.accept(visitor);
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getColumns());
System.out.println("coditions : " + visitor.getConditions());
System.out.println("relationships : " + visitor.getRelationships());
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("T")));
Assert.assertEquals(1, visitor.getTables().size());
Assert.assertEquals(2, visitor.getColumns().size());
// Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("bonuses", "employee_id")));
}
}
src/test/java/com/alibaba/druid/bvt/sql/oracle/OracleIbatisSelectTest1.java
+41
View File
@@ -0,0 +1,41 @@
package com.alibaba.druid.bvt.sql.oracle;
import java.util.List;
import junit.framework.Assert;
import com.alibaba.druid.sql.OracleTest;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.dialect.oracle.parser.OracleStatementParser;
import com.alibaba.druid.sql.dialect.oracle.visitor.OracleSchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
public class OracleIbatisSelectTest1 extends OracleTest {
public void test_0() throws Exception {
String sql = "SELECT * FROM T WHERE F1 = $id$";
OracleStatementParser parser = new OracleStatementParser(sql);
List<SQLStatement> statementList = parser.parseStatementList();
SQLStatement statemen = statementList.get(0);
print(statementList);
Assert.assertEquals(1, statementList.size());
OracleSchemaStatVisitor visitor = new OracleSchemaStatVisitor();
statemen.accept(visitor);
System.out.println("Tables : " + visitor.getTables());
System.out.println("fields : " + visitor.getColumns());
System.out.println("coditions : " + visitor.getConditions());
System.out.println("relationships : " + visitor.getRelationships());
Assert.assertTrue(visitor.getTables().containsKey(new TableStat.Name("T")));
Assert.assertEquals(1, visitor.getTables().size());
//Assert.assertEquals(2, visitor.getColumns().size());
// Assert.assertTrue(visitor.getColumns().contains(new TableStat.Column("bonuses", "employee_id")));
}
}