forked from alibaba/druid
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
27 changed files
with
380 additions
and
323 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
39 changes: 0 additions & 39 deletions
39
core/src/main/java/com/alibaba/druid/sql/dialect/bigquery/BigQueryCreateTableParser.java
This file was deleted.
Oops, something went wrong.
72 changes: 72 additions & 0 deletions
72
...rc/main/java/com/alibaba/druid/sql/dialect/bigquery/parser/BigQueryCreateTableParser.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,72 @@ | ||
package com.alibaba.druid.sql.dialect.bigquery.parser; | ||
|
||
import com.alibaba.druid.sql.ast.SQLName; | ||
import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr; | ||
import com.alibaba.druid.sql.ast.statement.SQLColumnDefinition; | ||
import com.alibaba.druid.sql.ast.statement.SQLCreateTableStatement; | ||
import com.alibaba.druid.sql.ast.statement.SQLSelectOrderByItem; | ||
import com.alibaba.druid.sql.dialect.db2.parser.DB2ExprParser; | ||
import com.alibaba.druid.sql.parser.SQLCreateTableParser; | ||
import com.alibaba.druid.sql.parser.SQLExprParser; | ||
import com.alibaba.druid.sql.parser.Token; | ||
|
||
public class BigQueryCreateTableParser extends SQLCreateTableParser { | ||
public BigQueryCreateTableParser(String sql) { | ||
super(new DB2ExprParser(sql)); | ||
} | ||
|
||
public BigQueryCreateTableParser(SQLExprParser exprParser) { | ||
super(exprParser); | ||
} | ||
|
||
protected void parseCreateTableRest(SQLCreateTableStatement stmt) { | ||
for (;;) { | ||
if (lexer.token() == Token.PARTITION) { | ||
lexer.nextToken(); | ||
accept(Token.BY); | ||
|
||
boolean brace = lexer.nextIf(Token.LPAREN); | ||
for (; ; ) { | ||
SQLName name; | ||
name = exprParser.name(); | ||
if (name instanceof SQLIdentifierExpr | ||
&& ((SQLIdentifierExpr) name).getName().equalsIgnoreCase("DATE") | ||
&& lexer.nextIf(Token.LPAREN) | ||
) { | ||
name = exprParser.name(); | ||
accept(Token.RPAREN); | ||
name.putAttribute("function", "DATE"); | ||
} | ||
stmt.addPartitionColumn(new SQLColumnDefinition(name)); | ||
if (lexer.nextIf(Token.COMMA)) { | ||
continue; | ||
} | ||
break; | ||
} | ||
if (brace) { | ||
accept(Token.RPAREN); | ||
} | ||
continue; | ||
} | ||
|
||
if (lexer.nextIfIdentifier("CLUSTERED")) { | ||
accept(Token.BY); | ||
SQLSelectOrderByItem item = exprParser.parseSelectOrderByItem(); | ||
item.setParent(stmt); | ||
stmt.getClusteredBy().add(item); | ||
} | ||
|
||
if (lexer.nextIfIdentifier("OPTIONS")) { | ||
exprParser.parseAssignItem(stmt.getTableOptions(), stmt); | ||
continue; | ||
} | ||
|
||
if (lexer.nextIfIdentifier("CLONE")) { | ||
stmt.setLike(exprParser.name()); | ||
continue; | ||
} | ||
|
||
break; | ||
} | ||
} | ||
} |
2 changes: 1 addition & 1 deletion
2
.../dialect/bigquery/BigQueryExprParser.java → ...t/bigquery/parser/BigQueryExprParser.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
2 changes: 1 addition & 1 deletion
2
...d/sql/dialect/bigquery/BigQueryLexer.java → ...ialect/bigquery/parser/BigQueryLexer.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
2 changes: 1 addition & 1 deletion
2
...ialect/bigquery/BigQuerySelectParser.java → ...bigquery/parser/BigQuerySelectParser.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
2 changes: 1 addition & 1 deletion
2
...ect/bigquery/BigQueryStatementParser.java → ...query/parser/BigQueryStatementParser.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
74 changes: 74 additions & 0 deletions
74
core/src/main/java/com/alibaba/druid/sql/dialect/bigquery/visitor/BigQueryOutputVisitor.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,74 @@ | ||
package com.alibaba.druid.sql.dialect.bigquery.visitor; | ||
|
||
import com.alibaba.druid.DbType; | ||
import com.alibaba.druid.sql.ast.statement.SQLColumnDefinition; | ||
import com.alibaba.druid.sql.ast.statement.SQLCreateTableStatement; | ||
import com.alibaba.druid.sql.ast.statement.SQLExprTableSource; | ||
import com.alibaba.druid.sql.visitor.SQLASTOutputVisitor; | ||
|
||
import java.util.List; | ||
|
||
public class BigQueryOutputVisitor extends SQLASTOutputVisitor { | ||
public BigQueryOutputVisitor(StringBuilder appender) { | ||
super(appender, DbType.db2); | ||
} | ||
|
||
public BigQueryOutputVisitor(StringBuilder appender, boolean parameterized) { | ||
super(appender, parameterized); | ||
this.dbType = DbType.bigquery; | ||
} | ||
|
||
protected void printPartitionedBy(SQLCreateTableStatement x) { | ||
List<SQLColumnDefinition> partitionColumns = x.getPartitionColumns(); | ||
int partitionSize = partitionColumns.size(); | ||
if (partitionSize == 0) { | ||
return; | ||
} | ||
|
||
println(); | ||
print0(ucase ? "PARTITION BY (" : "partition by ("); | ||
this.indentCount++; | ||
println(); | ||
for (int i = 0; i < partitionSize; ++i) { | ||
SQLColumnDefinition column = partitionColumns.get(i); | ||
printPartitoinedByColumn(column); | ||
|
||
if (i != partitionSize - 1) { | ||
print(','); | ||
} | ||
if (this.isPrettyFormat() && column.hasAfterComment()) { | ||
print(' '); | ||
printlnComment(column.getAfterCommentsDirect()); | ||
} | ||
|
||
if (i != partitionSize - 1) { | ||
println(); | ||
} | ||
} | ||
this.indentCount--; | ||
println(); | ||
print(')'); | ||
} | ||
|
||
protected void printPartitoinedByColumn(SQLColumnDefinition column) { | ||
String function = (String) column.getName().getAttribute("function"); | ||
if (function != null) { | ||
print0(function); | ||
print('('); | ||
} | ||
column.accept(this); | ||
if (function != null) { | ||
print(')'); | ||
} | ||
} | ||
|
||
protected void printCreateTableLike(SQLCreateTableStatement x) { | ||
SQLExprTableSource like = x.getLike(); | ||
if (like == null) { | ||
return; | ||
} | ||
println(); | ||
print0(ucase ? "CLONE " : "clone "); | ||
like.accept(this); | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.