bug fixed for visitor

This commit is contained in:
温少
2012-02-23 07:09:40 +08:00
parent eff085fe9b
commit 3c587b5a8e
11 changed files with 132 additions and 4 deletions
@@ -7,6 +7,14 @@ public class PGCurrentOfExpr extends PGSQLObjectImpl implements SQLExpr {
private static final long serialVersionUID = 1L;
private SQLExpr cursor;
public PGCurrentOfExpr() {
}
public PGCurrentOfExpr(SQLExpr cursor) {
this.cursor = cursor;
}
public SQLExpr getCursor() {
return cursor;
@@ -14,8 +14,17 @@ public class PGDeleteStatement extends SQLDeleteStatement implements PGSQLStatem
private PGWithClause with;
private boolean only = false;
private List<SQLName> using = new ArrayList<SQLName>(2);
private boolean returning;
private String alias;
public boolean isReturning() {
return returning;
}
public void setReturning(boolean returning) {
this.returning = returning;
}
public String getAlias() {
return alias;
}
@@ -47,7 +56,7 @@ public class PGDeleteStatement extends SQLDeleteStatement implements PGSQLStatem
public void setWith(PGWithClause with) {
this.with = with;
}
protected void accept0(SQLASTVisitor visitor) {
accept0((PGASTVisitor) visitor);
}
@@ -105,6 +105,8 @@ public class PGLexer extends Lexer {
map.put("CASCADE", Token.CASCADE);
map.put("RESTRICT", Token.RESTRICT);
map.put("USING", Token.USING);
map.put("CURRENT", Token.CURRENT);
map.put("RETURNING", Token.RETURNING);
DEFAULT_MYSQL_KEYWORDS = new Keywords(map);
}
@@ -6,6 +6,7 @@ import com.alibaba.druid.sql.ast.SQLExpr;
import com.alibaba.druid.sql.ast.SQLName;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.ast.statement.SQLDeleteStatement;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGCurrentOfExpr;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGDeleteStatement;
import com.alibaba.druid.sql.dialect.postgresql.ast.PGTruncateStatement;
import com.alibaba.druid.sql.parser.Lexer;
@@ -59,8 +60,23 @@ public class PGSQLStatementParser extends SQLStatementParser {
if (lexer.token() == (Token.WHERE)) {
lexer.nextToken();
SQLExpr where = this.exprParser.expr();
deleteStatement.setWhere(where);
if (lexer.token() == Token.CURRENT) {
lexer.nextToken();
accept(Token.OF);
SQLName cursorName = this.exprParser.name();
SQLExpr where = new PGCurrentOfExpr(cursorName);
deleteStatement.setWhere(where);
} else {
SQLExpr where = this.exprParser.expr();
deleteStatement.setWhere(where);
}
}
if (lexer.token() == Token.RETURNING) {
lexer.nextToken();
accept(Token.STAR);
deleteStatement.setReturning(true);
}
return deleteStatement;
@@ -303,6 +303,11 @@ public class PGOutputVisitor extends SQLASTOutputVisitor implements PGASTVisitor
print("WHERE ");
x.getWhere().accept(this);
}
if (x.isReturning()) {
println();
print("RETURNING *");
}
return false;
}