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

improved sql parser support CreateView

Browse Source
This commit is contained in:
shaojin.wensj
2015-07-06 00:51:09 +08:00
parent ee0a32e0c7
commit 68c5630cd1
6 changed files with 247 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/sql/ast/statement/SQLCreateViewStatement.java
+71 -9
View File
@@ -20,25 +20,31 @@ import java.util.List;
import com.alibaba.druid.sql.ast.SQLExpr;
import com.alibaba.druid.sql.ast.SQLName;
import com.alibaba.druid.sql.ast.SQLObjectImpl;
import com.alibaba.druid.sql.ast.SQLStatementImpl;
import com.alibaba.druid.sql.ast.expr.SQLCharExpr;
import com.alibaba.druid.sql.ast.expr.SQLLiteralExpr;
import com.alibaba.druid.sql.visitor.SQLASTVisitor;
public class SQLCreateViewStatement extends SQLStatementImpl implements SQLDDLStatement {
private boolean orReplace = false;
protected SQLName name;
protected SQLSelect subQuery;
private boolean orReplace = false;
protected SQLName name;
protected SQLSelect subQuery;
protected boolean ifNotExists = false;
protected final List<SQLExpr> columns = new ArrayList<SQLExpr>();
protected final List<Column> columns = new ArrayList<Column>();
private Level with;
private Level with;
private SQLLiteralExpr comment;
public SQLCreateViewStatement(){
}
public SQLCreateViewStatement(String dbType){
super (dbType);
super(dbType);
}
public boolean isOrReplace() {
@@ -73,10 +79,29 @@ public class SQLCreateViewStatement extends SQLStatementImpl implements SQLDDLSt
this.subQuery = subQuery;
}
public List<SQLExpr> getColumns() {
public List<Column> getColumns() {
return columns;
}
public boolean isIfNotExists() {
return ifNotExists;
}
public void setIfNotExists(boolean ifNotExists) {
this.ifNotExists = ifNotExists;
}
public SQLLiteralExpr getComment() {
return comment;
}
public void setComment(SQLLiteralExpr comment) {
if (comment != null) {
comment.setParent(this);
}
this.comment = comment;
}
public void output(StringBuffer buf) {
buf.append("CREATE VIEW ");
this.name.output(buf);
@@ -106,12 +131,49 @@ public class SQLCreateViewStatement extends SQLStatementImpl implements SQLDDLSt
if (visitor.visit(this)) {
acceptChild(visitor, this.name);
acceptChild(visitor, this.columns);
acceptChild(visitor, this.comment);
acceptChild(visitor, this.subQuery);
}
visitor.endVisit(this);
}
public static enum Level {
CASCADED, LOCAL
CASCADED, LOCAL
}
public static class Column extends SQLObjectImpl {
private SQLExpr expr;
private SQLCharExpr comment;
public SQLExpr getExpr() {
return expr;
}
public void setExpr(SQLExpr expr) {
if (expr != null) {
expr.setParent(this);
}
this.expr = expr;
}
public SQLCharExpr getComment() {
return comment;
}
public void setComment(SQLCharExpr comment) {
if (comment != null) {
comment.setParent(this);
}
this.comment = comment;
}
@Override
protected void accept0(SQLASTVisitor visitor) {
if (visitor.visit(this)) {
acceptChild(visitor, expr);
acceptChild(visitor, comment);
}
}
}
}
src/main/java/com/alibaba/druid/sql/parser/SQLStatementParser.java
+40 -1
View File
@@ -22,6 +22,7 @@ import com.alibaba.druid.sql.ast.SQLExpr;
import com.alibaba.druid.sql.ast.SQLName;
import com.alibaba.druid.sql.ast.SQLObject;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.ast.expr.SQLCharExpr;
import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
import com.alibaba.druid.sql.ast.expr.SQLQueryExpr;
import com.alibaba.druid.sql.ast.expr.SQLVariantRefExpr;
@@ -1344,6 +1345,11 @@ public class SQLStatementParser extends SQLParser {
lexer.reset(markBp, markChar, Token.CREATE);
return parseCreateProcedure();
}
if (lexer.token() == Token.VIEW) {
lexer.reset(markBp, markChar, Token.CREATE);
return parseCreateView();
}
// lexer.reset(mark_bp, mark_ch, Token.CREATE);
throw new ParserException("TODO " + lexer.token() + " " + lexer.stringVal());
@@ -1600,14 +1606,47 @@ public class SQLStatementParser extends SQLParser {
}
this.accept(Token.VIEW);
if (lexer.token() == Token.IF || identifierEquals("IF")) {
lexer.nextToken();
accept(Token.NOT);
accept(Token.EXISTS);
createView.setIfNotExists(true);
}
createView.setName(exprParser.name());
if (lexer.token() == Token.LPAREN) {
lexer.nextToken();
this.exprParser.exprList(createView.getColumns(), createView);
for (;;) {
SQLCreateViewStatement.Column column = new SQLCreateViewStatement.Column();
SQLExpr expr = this.exprParser.expr();
column.setExpr(expr);
if (lexer.token() == Token.COMMENT) {
lexer.nextToken();
column.setComment((SQLCharExpr) exprParser.primary());
}
column.setParent(createView);
createView.getColumns().add(column);
if (lexer.token() == Token.COMMA) {
lexer.nextToken();
} else {
break;
}
}
accept(Token.RPAREN);
}
if (lexer.token() == Token.COMMENT) {
lexer.nextToken();
SQLCharExpr comment = (SQLCharExpr) exprParser.primary();
createView.setComment(comment);
}
this.accept(Token.AS);
src/main/java/com/alibaba/druid/sql/visitor/SQLASTOutputVisitor.java
+46 -11
View File
@@ -156,7 +156,7 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Printab
private boolean prettyFormat = true;
protected int selectListNumberOfLine = 5;
private List<Object> parameters;
private List<Object> parameters;
public SQLASTOutputVisitor(Appendable appender){
this.appender = appender;
@@ -726,7 +726,7 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Printab
if (alias != null && alias.length() > 0) {
print(" AS ");
if (alias.indexOf(' ') == -1 || alias.charAt(0) == '"' || alias.charAt(0) == '\'') {
print(alias);
print(alias);
} else {
print('"');
print(alias);
@@ -1070,7 +1070,7 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Printab
println();
print(")");
}
if (x.getInherits() != null) {
print(" INHERITS (");
x.getInherits().accept(this);
@@ -1411,7 +1411,7 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Printab
public boolean visit(SQLAlterTableDropColumnItem x) {
print("DROP COLUMN ");
this.printAndAccept(x.getColumns(), ", ");
if (x.isCascade()) {
print(" CASCADE");
}
@@ -1481,20 +1481,55 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Printab
print("OR REPLACE ");
}
print("VIEW ");
if (x.isIfNotExists()) {
print("IF NOT EXISTS ");
}
x.getName().accept(this);
if (x.getColumns().size() > 0) {
print(" (");
printAndAccept(x.getColumns(), ", ");
println();
print("(");
incrementIndent();
println();
for (int i = 0; i < x.getColumns().size(); ++i) {
if (i != 0) {
print(", ");
println();
}
x.getColumns().get(i).accept(this);
}
decrementIndent();
println();
print(")");
}
print(" AS ");
if (x.getComment() != null) {
println();
print("COMMENT ");
x.getComment().accept(this);
}
println();
print("AS");
println();
x.getSubQuery().accept(this);
return false;
}
public boolean visit(SQLCreateViewStatement.Column x) {
x.getExpr().accept(this);
if (x.getComment() != null) {
print(" COMMENT ");
x.getComment().accept(this);
}
return false;
}
@Override
public boolean visit(SQLAlterTableDropIndex x) {
print("DROP INDEX ");
@@ -1600,7 +1635,7 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Printab
public boolean visit(SQLAlterTableAlterColumn x) {
print("ALTER COLUMN ");
x.getColumn().accept(this);
if (x.isSetNotNull()) { // postgresql
print(" SET NOT NULL");
}
@@ -1614,7 +1649,7 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Printab
if (x.isDropDefault()) { // postgresql
print(" DROP DEFAULT");
}
return false;
}
@@ -2115,7 +2150,7 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Printab
return false;
}
@Override
public boolean visit(SQLBinaryExpr x) {
print("b'");
@@ -2124,7 +2159,7 @@ public class SQLASTOutputVisitor extends SQLASTVisitorAdapter implements Printab
return false;
}
@Override
public boolean visit(SQLAlterTableRename x) {
print("RENAME TO ");
src/main/java/com/alibaba/druid/sql/visitor/SQLASTVisitor.java
+4
View File
@@ -292,6 +292,10 @@ public interface SQLASTVisitor {
boolean visit(SQLCreateViewStatement x);
void endVisit(SQLCreateViewStatement x);
boolean visit(SQLCreateViewStatement.Column x);
void endVisit(SQLCreateViewStatement.Column x);
boolean visit(NotNullConstraint x);
src/main/java/com/alibaba/druid/sql/visitor/SQLASTVisitorAdapter.java
+7
View File
@@ -391,6 +391,13 @@ public class SQLASTVisitorAdapter implements SQLASTVisitor {
public void endVisit(SQLCreateViewStatement x) {
}
public boolean visit(SQLCreateViewStatement.Column x) {
return true;
}
public void endVisit(SQLCreateViewStatement.Column x) {
}
public boolean visit(NotNullConstraint x) {
return true;