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

support pgsql truncate statement

Browse Source
This commit is contained in:
温少
2012-02-22 20:02:23 +08:00
parent c9db35f1af
commit a6ba872902
19 changed files with 636 additions and 323 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/sql/dialect/postgresql/ast/PGSQLStatement.java
+7
View File
@@ -0,0 +1,7 @@
package com.alibaba.druid.sql.dialect.postgresql.ast;
import com.alibaba.druid.sql.ast.SQLStatement;
public interface PGSQLStatement extends SQLStatement, PGSQLObject {
}
src/main/java/com/alibaba/druid/sql/dialect/postgresql/ast/PGSQLStatementImpl.java
+8
View File
@@ -0,0 +1,8 @@
package com.alibaba.druid.sql.dialect.postgresql.ast;
public abstract class PGSQLStatementImpl extends PGSQLObjectImpl implements PGSQLStatement {
private static final long serialVersionUID = 1L;
}
src/main/java/com/alibaba/druid/sql/dialect/postgresql/ast/PGTruncateStatement.java
+43
View File
@@ -0,0 +1,43 @@
package com.alibaba.druid.sql.dialect.postgresql.ast;
import com.alibaba.druid.sql.ast.statement.SQLTruncateStatement;
import com.alibaba.druid.sql.dialect.postgresql.visitor.PGASTVisitor;
public class PGTruncateStatement extends SQLTruncateStatement implements PGSQLStatement{
private static final long serialVersionUID = 1L;
private boolean only;
private Boolean restartIdentity;
private Boolean cascade;
public boolean isOnly() {
return only;
}
public void setOnly(boolean only) {
this.only = only;
}
public Boolean getRestartIdentity() {
return restartIdentity;
}
public void setRestartIdentity(Boolean restartIdentity) {
this.restartIdentity = restartIdentity;
}
public Boolean getCascade() {
return cascade;
}
public void setCascade(Boolean cascade) {
this.cascade = cascade;
}
public void accept0(PGASTVisitor visitor) {
if (visitor.visit(this)) {
acceptChild(visitor, tableNames);
}
}
}
src/main/java/com/alibaba/druid/sql/dialect/postgresql/parser/PGLexer.java
+1
View File
@@ -81,6 +81,7 @@ public class PGLexer extends Lexer {
map.put("LOCK", Token.LOCK);
map.put("SOME", Token.SOME);
map.put("ANY", Token.ANY);
map.put("TRUNCATE", Token.TRUNCATE);
map.put("LIMIT", Token.LIMIT);
map.put("OFFSET", Token.OFFSET);
src/main/java/com/alibaba/druid/sql/dialect/postgresql/parser/PGSQLStatementParser.java
+49
View File
@@ -2,7 +2,9 @@ package com.alibaba.druid.sql.dialect.postgresql.parser;
import java.util.List;
import com.alibaba.druid.sql.ast.SQLName;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGTruncateStatement;
import com.alibaba.druid.sql.parser.Lexer;
import com.alibaba.druid.sql.parser.ParserException;
import com.alibaba.druid.sql.parser.SQLStatementParser;
@@ -22,6 +24,53 @@ public class PGSQLStatementParser extends SQLStatementParser {
return new PGSelectParser(this.lexer);
}
public SQLStatement parseTruncate() {
accept(Token.TRUNCATE);
PGTruncateStatement stmt = new PGTruncateStatement();
if (lexer.token() == Token.TABLE) {
lexer.nextToken();
}
if (lexer.token() == Token.ONLY) {
lexer.nextToken();
stmt.setOnly(true);
}
for (;;) {
SQLName name = this.exprParser.name();
stmt.getTableNames().add(name);
if (lexer.token() == Token.COMMA) {
lexer.nextToken();
continue;
}
break;
}
if (lexer.token() == Token.RESTART) {
lexer.nextToken();
accept(Token.IDENTITY);
stmt.setRestartIdentity(Boolean.TRUE);
} else if (lexer.token() == Token.SHARE) {
lexer.nextToken();
accept(Token.IDENTITY);
stmt.setRestartIdentity(Boolean.FALSE);
}
if (lexer.token() == Token.CASCADE) {
lexer.nextToken();
stmt.setCascade(Boolean.TRUE);
} else if (lexer.token() == Token.RESTRICT) {
lexer.nextToken();
stmt.setCascade(Boolean.FALSE);
}
return stmt;
}
public boolean parseStatementListDialect(List<SQLStatement> statementList) {
if (lexer.token() == Token.WITH) {
SQLStatement stmt = parseSelect();
src/main/java/com/alibaba/druid/sql/dialect/postgresql/visitor/PGASTVisitor.java
+24 -18
View File
@@ -1,30 +1,36 @@
package com.alibaba.druid.sql.dialect.postgresql.visitor;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGTruncateStatement;
import com.alibaba.druid.sql.visitor.SQLASTVisitor;
public interface PGASTVisitor extends SQLASTVisitor {
void endVisit(PGSelectQueryBlock.WindowClause x);
boolean visit(PGSelectQueryBlock.WindowClause x);
void endVisit(PGSelectQueryBlock.WindowClause x);
void endVisit(PGSelectQueryBlock.FetchClause x);
boolean visit(PGSelectQueryBlock.WindowClause x);
boolean visit(PGSelectQueryBlock.FetchClause x);
void endVisit(PGSelectQueryBlock.FetchClause x);
void endVisit(PGSelectQueryBlock.ForClause x);
boolean visit(PGSelectQueryBlock.FetchClause x);
boolean visit(PGSelectQueryBlock.ForClause x);
void endVisit(PGSelectQueryBlock.WithQuery x);
boolean visit(PGSelectQueryBlock.WithQuery x);
void endVisit(PGSelectQueryBlock.WithClause x);
boolean visit(PGSelectQueryBlock.WithClause x);
void endVisit(PGSelectQueryBlock.IntoClause x);
boolean visit(PGSelectQueryBlock.IntoClause x);
void endVisit(PGSelectQueryBlock.ForClause x);
boolean visit(PGSelectQueryBlock.ForClause x);
void endVisit(PGSelectQueryBlock.WithQuery x);
boolean visit(PGSelectQueryBlock.WithQuery x);
void endVisit(PGSelectQueryBlock.WithClause x);
boolean visit(PGSelectQueryBlock.WithClause x);
void endVisit(PGSelectQueryBlock.IntoClause x);
boolean visit(PGSelectQueryBlock.IntoClause x);
void endVisit(PGTruncateStatement x);
boolean visit(PGTruncateStatement x);
}
src/main/java/com/alibaba/druid/sql/dialect/postgresql/visitor/PGASTVisitorAdapter.java
+59 -48
View File
@@ -6,75 +6,86 @@ import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock.IntoClaus
import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock.WindowClause;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock.WithClause;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock.WithQuery;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGTruncateStatement;
import com.alibaba.druid.sql.visitor.SQLASTVisitorAdapter;
public class PGASTVisitorAdapter extends SQLASTVisitorAdapter implements
PGASTVisitor {
public class PGASTVisitorAdapter extends SQLASTVisitorAdapter implements PGASTVisitor {
@Override
public void endVisit(WindowClause x) {
@Override
public void endVisit(WindowClause x) {
}
}
@Override
public boolean visit(WindowClause x) {
@Override
public boolean visit(WindowClause x) {
return true;
}
return true;
}
@Override
public void endVisit(FetchClause x) {
@Override
public void endVisit(FetchClause x) {
}
}
@Override
public boolean visit(FetchClause x) {
@Override
public boolean visit(FetchClause x) {
return true;
}
return true;
}
@Override
public void endVisit(ForClause x) {
@Override
public void endVisit(ForClause x) {
}
}
@Override
public boolean visit(ForClause x) {
@Override
public boolean visit(ForClause x) {
return true;
}
return true;
}
@Override
public void endVisit(WithQuery x) {
@Override
public void endVisit(WithQuery x) {
}
}
@Override
public boolean visit(WithQuery x) {
@Override
public boolean visit(WithQuery x) {
return true;
}
return true;
}
@Override
public void endVisit(WithClause x) {
@Override
public void endVisit(WithClause x) {
}
}
@Override
public boolean visit(WithClause x) {
@Override
public boolean visit(WithClause x) {
return true;
}
@Override
public void endVisit(IntoClause x) {
}
@Override
public boolean visit(IntoClause x) {
return true;
}
return true;
}
@Override
public void endVisit(IntoClause x) {
}
@Override
public boolean visit(IntoClause x) {
return true;
}
@Override
public void endVisit(PGTruncateStatement x) {
}
@Override
public boolean visit(PGTruncateStatement x) {
return true;
}
}
src/main/java/com/alibaba/druid/sql/dialect/postgresql/visitor/PGOutputVisitor.java
+33
View File
@@ -9,6 +9,7 @@ import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock.IntoClaus
import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock.WindowClause;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock.WithClause;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock.WithQuery;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGTruncateStatement;
import com.alibaba.druid.sql.visitor.SQLASTOutputVisitor;
public class PGOutputVisitor extends SQLASTOutputVisitor implements
@@ -232,4 +233,36 @@ public class PGOutputVisitor extends SQLASTOutputVisitor implements
x.getTable().accept(this);
return false;
}
@Override
public void endVisit(PGTruncateStatement x) {
}
@Override
public boolean visit(PGTruncateStatement x) {
print("TRUNCATE TABLE ");
if (x.isOnly()) {
print("ONLY ");
}
printlnAndAccept(x.getTableNames(), ", ");
if (x.getRestartIdentity() != null) {
if (x.getRestartIdentity().booleanValue()) {
print(" RESTART IDENTITY");
} else {
print(" CONTINUE IDENTITY");
}
}
if (x.getCascade() != null) {
if (x.getCascade().booleanValue()) {
print(" CASCADE");
} else {
print(" RESTRICT");
}
}
return false;
}
}
src/main/java/com/alibaba/druid/sql/dialect/postgresql/visitor/PGSchemaStatVisitor.java
+64 -53
View File
@@ -1,86 +1,97 @@
package com.alibaba.druid.sql.dialect.postgresql.visitor;
import com.alibaba.druid.sql.ast.statement.SQLTruncateStatement;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock.FetchClause;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock.ForClause;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock.IntoClause;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock.WindowClause;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock.WithClause;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGSelectQueryBlock.WithQuery;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGTruncateStatement;
import com.alibaba.druid.sql.visitor.SchemaStatVisitor;
import com.alibaba.druid.stat.TableStat;
public class PGSchemaStatVisitor extends SchemaStatVisitor implements
PGASTVisitor {
public class PGSchemaStatVisitor extends SchemaStatVisitor implements PGASTVisitor {
@Override
public void endVisit(WindowClause x) {
@Override
public void endVisit(WindowClause x) {
}
}
@Override
public boolean visit(WindowClause x) {
return true;
}
@Override
public boolean visit(WindowClause x) {
return true;
}
@Override
public void endVisit(FetchClause x) {
@Override
public void endVisit(FetchClause x) {
}
}
@Override
public boolean visit(FetchClause x) {
return true;
}
@Override
public boolean visit(FetchClause x) {
return true;
}
@Override
public void endVisit(ForClause x) {
@Override
public void endVisit(ForClause x) {
}
}
@Override
public boolean visit(ForClause x) {
@Override
public boolean visit(ForClause x) {
return true;
}
return true;
}
@Override
public void endVisit(WithQuery x) {
@Override
public void endVisit(WithQuery x) {
}
}
@Override
public boolean visit(WithQuery x) {
@Override
public boolean visit(WithQuery x) {
return true;
}
return true;
}
@Override
public void endVisit(WithClause x) {
@Override
public void endVisit(WithClause x) {
}
}
@Override
public boolean visit(WithClause x) {
@Override
public boolean visit(WithClause x) {
return true;
}
return true;
}
@Override
public void endVisit(IntoClause x) {
}
@Override
public void endVisit(IntoClause x) {
@Override
public boolean visit(IntoClause x) {
String ident = x.getTable().toString();
TableStat stat = tableStats.get(ident);
if (stat == null) {
stat = new TableStat();
tableStats.put(new TableStat.Name(ident), stat);
}
stat.incrementInsertCount();
return false;
}
}
@Override
public boolean visit(IntoClause x) {
String ident = x.getTable().toString();
TableStat stat = tableStats.get(ident);
if (stat == null) {
stat = new TableStat();
tableStats.put(new TableStat.Name(ident), stat);
}
stat.incrementInsertCount();
return false;
}
@Override
public void endVisit(PGTruncateStatement x) {
}
@Override
public boolean visit(PGTruncateStatement x) {
this.visit((SQLTruncateStatement) x);
return false;
}
}