Search in sources :

Example 1 with OracleAlterTableTruncatePartition

use of com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleAlterTableTruncatePartition in project druid by alibaba.

the class OracleStatementParser method parseAlterTable.

private SQLStatement parseAlterTable() {
    lexer.nextToken();
    SQLAlterTableStatement stmt = new SQLAlterTableStatement(getDbType());
    stmt.setName(this.exprParser.name());
    for (; ; ) {
        if (identifierEquals("ADD")) {
            lexer.nextToken();
            if (lexer.token() == Token.LPAREN) {
                lexer.nextToken();
                SQLAlterTableAddColumn item = parseAlterTableAddColumn();
                stmt.addItem(item);
                accept(Token.RPAREN);
            } else if (lexer.token() == Token.CONSTRAINT) {
                OracleConstraint constraint = ((OracleExprParser) this.exprParser).parseConstaint();
                SQLAlterTableAddConstraint item = new SQLAlterTableAddConstraint();
                constraint.setParent(item);
                item.setParent(stmt);
                item.setConstraint(constraint);
                stmt.addItem(item);
            } else if (lexer.token() == Token.IDENTIFIER) {
                SQLAlterTableAddColumn item = parseAlterTableAddColumn();
                stmt.addItem(item);
            } else {
                throw new ParserException("TODO : " + lexer.token() + " " + lexer.stringVal());
            }
            continue;
        } else if (identifierEquals("MOVE")) {
            lexer.nextToken();
            if (lexer.token() == Token.TABLESPACE) {
                lexer.nextToken();
                OracleAlterTableMoveTablespace item = new OracleAlterTableMoveTablespace();
                item.setName(this.exprParser.name());
                stmt.addItem(item);
            } else {
                throw new ParserException("TODO : " + lexer.token() + " " + lexer.stringVal());
            }
        } else if (identifierEquals("RENAME")) {
            stmt.addItem(parseAlterTableRename());
        } else if (identifierEquals("MODIFY")) {
            lexer.nextToken();
            OracleAlterTableModify item = new OracleAlterTableModify();
            if (lexer.token() == Token.LPAREN) {
                lexer.nextToken();
                for (; ; ) {
                    SQLColumnDefinition columnDef = this.exprParser.parseColumn();
                    item.addColumn(columnDef);
                    if (lexer.token() == Token.COMMA) {
                        lexer.nextToken();
                        continue;
                    }
                    break;
                }
                accept(Token.RPAREN);
            } else {
                SQLColumnDefinition columnDef = this.exprParser.parseColumn();
                item.addColumn(columnDef);
            }
            stmt.addItem(item);
            continue;
        } else if (identifierEquals("SPLIT")) {
            parseAlterTableSplit(stmt);
            continue;
        } else if (lexer.token() == Token.TRUNCATE) {
            lexer.nextToken();
            if (identifierEquals("PARTITION")) {
                lexer.nextToken();
                OracleAlterTableTruncatePartition item = new OracleAlterTableTruncatePartition();
                item.setName(this.exprParser.name());
                stmt.addItem(item);
            } else {
                throw new ParserException("TODO : " + lexer.token() + " " + lexer.stringVal());
            }
            continue;
        } else if (lexer.token() == Token.DROP) {
            parseAlterDrop(stmt);
            continue;
        } else if (lexer.token() == Token.DISABLE) {
            lexer.nextToken();
            if (lexer.token() == Token.CONSTRAINT) {
                lexer.nextToken();
                SQLAlterTableEnableConstraint item = new SQLAlterTableEnableConstraint();
                item.setConstraintName(this.exprParser.name());
                stmt.addItem(item);
            } else {
                throw new ParserException("TODO : " + lexer.token() + " " + lexer.stringVal());
            }
        } else if (lexer.token() == Token.ENABLE) {
            lexer.nextToken();
            if (lexer.token() == Token.CONSTRAINT) {
                lexer.nextToken();
                SQLAlterTableDisableConstraint item = new SQLAlterTableDisableConstraint();
                item.setConstraintName(this.exprParser.name());
                stmt.addItem(item);
            } else {
                throw new ParserException("TODO : " + lexer.token() + " " + lexer.stringVal());
            }
        }
        break;
    }
    if (lexer.token() == Token.UPDATE) {
        lexer.nextToken();
        if (identifierEquals("GLOBAL")) {
            lexer.nextToken();
            acceptIdentifier("INDEXES");
            stmt.setUpdateGlobalIndexes(true);
        } else {
            throw new ParserException("TODO : " + lexer.token() + " " + lexer.stringVal());
        }
    }
    return stmt;
}
Also used : ParserException(com.alibaba.druid.sql.parser.ParserException) OracleAlterTableMoveTablespace(com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleAlterTableMoveTablespace) OracleAlterTableModify(com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleAlterTableModify) OracleConstraint(com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleConstraint) OracleAlterTableTruncatePartition(com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleAlterTableTruncatePartition)

Aggregations

OracleAlterTableModify (com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleAlterTableModify)1 OracleAlterTableMoveTablespace (com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleAlterTableMoveTablespace)1 OracleAlterTableTruncatePartition (com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleAlterTableTruncatePartition)1 OracleConstraint (com.alibaba.druid.sql.dialect.oracle.ast.stmt.OracleConstraint)1 ParserException (com.alibaba.druid.sql.parser.ParserException)1