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

增强支持CONNECT

Browse Source 
git-svn-id: http://code.alibabatech.com/svn/druid/trunk@131 b9813039-fb51-4c41-a8b9-e21c2acb5095
This commit is contained in:
wenshao
2011-05-19 02:34:44 +00:00
parent 39067f1133
commit 6311c67f54
6 changed files with 127 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/ast/stmt/OracleSelectHierachicalQueryClause.java
+9
View File
@@ -25,12 +25,21 @@ public class OracleSelectHierachicalQueryClause extends OracleSQLObject {
private SQLExpr startWith;
private SQLExpr connectBy;
private boolean prior = false;
private boolean noCycle = false;
public OracleSelectHierachicalQueryClause(){
}
public boolean isPrior() {
return prior;
}
public void setPrior(boolean prior) {
this.prior = prior;
}
protected void accept0(OracleASTVisitor visitor) {
if (visitor.visit(this)) {
acceptChild(visitor, this.startWith);
src/main/java/com/alibaba/druid/sql/dialect/oracle/ast/visitor/OracleOutputVisitor.java
+6
View File
@@ -561,9 +561,15 @@ public class OracleOutputVisitor extends SQLASTOutputVisitor implements OracleAS
public boolean visit(OracleSelectHierachicalQueryClause x) {
print("CONNECT BY ");
if (x.isPrior()) {
print("PRIOR ");
}
if (x.isNoCycle()) {
print("NOCYCLE ");
}
x.getConnectBy().accept(this);
print(" START WITH ");
x.getStartWith().accept(this);
src/main/java/com/alibaba/druid/sql/dialect/oracle/parser/OracleLexer.java
+86
View File
@@ -1,16 +1,102 @@
package com.alibaba.druid.sql.dialect.oracle.parser;
import java.util.HashMap;
import java.util.Map;
import com.alibaba.druid.sql.parser.Keywords;
import com.alibaba.druid.sql.parser.Lexer;
import com.alibaba.druid.sql.parser.Token;
public class OracleLexer extends Lexer {
public final static Keywords DEFAULT_ORACLE_KEYWORDS;
static {
Map<String, Token> map = new HashMap<String, Token>();
map.put("EXISTS", Token.EXISTS);
map.put("THEN", Token.THEN);
map.put("AS", Token.AS);
map.put("GROUP", Token.GROUP);
map.put("BY", Token.BY);
map.put("HAVING", Token.HAVING);
map.put("DELETE", Token.DELETE);
map.put("ORDER", Token.ORDER);
map.put("INDEX", Token.INDEX);
map.put("FOR", Token.FOR);
map.put("SCHEMA", Token.SCHEMA);
map.put("FOREIGN", Token.FOREIGN);
map.put("REFERENCE", Token.REFERENCE);
map.put("REFERENCES", Token.REFERENCES);
map.put("CHECK", Token.CHECK);
map.put("PRIMARY", Token.PRIMARY);
map.put("KEY", Token.KEY);
map.put("CONSTRAINT", Token.CONSTRAINT);
map.put("DEFAULT", Token.DEFAULT);
map.put("VIEW", Token.VIEW);
map.put("CREATE", Token.CREATE);
map.put("VALUES", Token.VALUES);
map.put("ALTER", Token.ALTER);
map.put("TABLE", Token.TABLE);
map.put("DROP", Token.DROP);
map.put("SET", Token.SET);
map.put("INTO", Token.INTO);
map.put("UPDATE", Token.UPDATE);
map.put("NULL", Token.NULL);
map.put("IS", Token.IS);
map.put("NOT", Token.NOT);
map.put("SELECT", Token.SELECT);
map.put("INSERT", Token.INSERT);
map.put("FROM", Token.FROM);
map.put("WHERE", Token.WHERE);
map.put("AND", Token.AND);
map.put("OR", Token.OR);
map.put("XOR", Token.XOR);
map.put("DISTINCT", Token.DISTINCT);
map.put("UNIQUE", Token.UNIQUE);
map.put("ALL", Token.ALL);
map.put("UNION", Token.UNION);
map.put("INTERSECT", Token.INTERSECT);
map.put("MINUS", Token.MINUS);
map.put("INNER", Token.INNER);
map.put("LEFT", Token.LEFT);
map.put("RIGHT", Token.RIGHT);
map.put("FULL", Token.FULL);
map.put("ON", Token.ON);
map.put("OUTER", Token.OUTER);
map.put("JOIN", Token.JOIN);
map.put("NEW", Token.NEW);
map.put("CASE", Token.CASE);
map.put("WHEN", Token.WHEN);
map.put("END", Token.END);
map.put("WHEN", Token.WHEN);
map.put("ELSE", Token.ELSE);
map.put("EXISTS", Token.EXISTS);
map.put("CAST", Token.CAST);
map.put("IN", Token.IN);
map.put("ASC", Token.ASC);
map.put("DESC", Token.DESC);
map.put("LIKE", Token.LIKE);
map.put("ESCAPE", Token.ESCAPE);
map.put("BETWEEN", Token.BETWEEN);
map.put("INTERVAL", Token.INTERVAL);
map.put("LOCK", Token.LOCK);
map.put("SOME", Token.SOME);
map.put("ANY", Token.ANY);
map.put("START", Token.START);
map.put("CONNECT", Token.CONNECT);
map.put("PRIOR", Token.PRIOR);
DEFAULT_ORACLE_KEYWORDS = new Keywords(map);
}
public OracleLexer(char[] input, int inputLength, boolean skipComment){
super(input, inputLength, skipComment);
super.keywods = DEFAULT_ORACLE_KEYWORDS;
}
public OracleLexer(String input){
super(input);
super.keywods = DEFAULT_ORACLE_KEYWORDS;
}
public void scanComment() {
src/main/java/com/alibaba/druid/sql/dialect/oracle/parser/OracleSelectParser.java
+8 -2
View File
@@ -217,7 +217,7 @@ public class OracleSelectParser extends SQLSelectParser {
hierachical.setConnectBy(this.createExprParser().expr());
}
if (identifierEquals("START")) {
if (lexer.token() == Token.START) {
lexer.nextToken();
if (hierachical == null) {
hierachical = new OracleSelectHierachicalQueryClause();
@@ -227,13 +227,19 @@ public class OracleSelectParser extends SQLSelectParser {
hierachical.setStartWith(this.createExprParser().expr());
}
if (identifierEquals("CONNECT")) {
if (lexer.token() == Token.CONNECT) {
if (hierachical == null) {
hierachical = new OracleSelectHierachicalQueryClause();
}
lexer.nextToken();
accept(Token.BY);
if (lexer.token() == Token.PRIOR) {
lexer.nextToken();
hierachical.setPrior(true);
}
if (identifierEquals("NOCYCLE")) {
hierachical.setNoCycle(true);
src/main/java/com/alibaba/druid/sql/parser/Token.java
+6
View File
@@ -93,8 +93,14 @@ public enum Token {
SOME("SOME"),
ANY("ANY"),
// mysql
TRUE("TRUE"),
FALSE("FALSE"),
// oracle
START("START"),
PRIOR("PRIOR"),
CONNECT("CONNECT"),
EOF, ERROR,
IDENTIFIER,