Examples with Command org.h2.command.Command used on opensource projects

Example 36 with Command

use of org.h2.command.Command in project h2database by h2database.

the class Parser method parseSelectUnionExtension.

private Query parseSelectUnionExtension(Query command, int start, boolean unionOnly) {
    while (true) {
        if (readIf("UNION")) {
            SelectUnion union = new SelectUnion(session, command);
            if (readIf("ALL")) {
                union.setUnionType(SelectUnion.UnionType.UNION_ALL);
            } else {
                readIf("DISTINCT");
                union.setUnionType(SelectUnion.UnionType.UNION);
            }
            union.setRight(parseSelectSub());
            command = union;
        } else if (readIf("MINUS") || readIf("EXCEPT")) {
            SelectUnion union = new SelectUnion(session, command);
            union.setUnionType(SelectUnion.UnionType.EXCEPT);
            union.setRight(parseSelectSub());
            command = union;
        } else if (readIf("INTERSECT")) {
            SelectUnion union = new SelectUnion(session, command);
            union.setUnionType(SelectUnion.UnionType.INTERSECT);
            union.setRight(parseSelectSub());
            command = union;
        } else {
            break;
        }
    }
    if (!unionOnly) {
        parseEndOfQuery(command);
    }
    setSQL(command, null, start);
    return command;
}

Example 37 with Command

use of org.h2.command.Command in project h2database by h2database.

the class Parser method parseCreateTrigger.

private CreateTrigger parseCreateTrigger(boolean force) {
    boolean ifNotExists = readIfNotExists();
    String triggerName = readIdentifierWithSchema(null);
    Schema schema = getSchema();
    boolean insteadOf, isBefore;
    if (readIf("INSTEAD")) {
        read("OF");
        isBefore = true;
        insteadOf = true;
    } else if (readIf("BEFORE")) {
        insteadOf = false;
        isBefore = true;
    } else {
        read("AFTER");
        insteadOf = false;
        isBefore = false;
    }
    int typeMask = 0;
    boolean onRollback = false;
    do {
        if (readIf("INSERT")) {
            typeMask |= Trigger.INSERT;
        } else if (readIf("UPDATE")) {
            typeMask |= Trigger.UPDATE;
        } else if (readIf("DELETE")) {
            typeMask |= Trigger.DELETE;
        } else if (readIf("SELECT")) {
            typeMask |= Trigger.SELECT;
        } else if (readIf("ROLLBACK")) {
            onRollback = true;
        } else {
            throw getSyntaxError();
        }
    } while (readIf(","));
    read("ON");
    String tableName = readIdentifierWithSchema();
    checkSchema(schema);
    CreateTrigger command = new CreateTrigger(session, getSchema());
    command.setForce(force);
    command.setTriggerName(triggerName);
    command.setIfNotExists(ifNotExists);
    command.setInsteadOf(insteadOf);
    command.setBefore(isBefore);
    command.setOnRollback(onRollback);
    command.setTypeMask(typeMask);
    command.setTableName(tableName);
    if (readIf("FOR")) {
        read("EACH");
        read("ROW");
        command.setRowBased(true);
    } else {
        command.setRowBased(false);
    }
    if (readIf("QUEUE")) {
        command.setQueueSize(readPositiveInt());
    }
    command.setNoWait(readIf("NOWAIT"));
    if (readIf("AS")) {
        command.setTriggerSource(readString());
    } else {
        read("CALL");
        command.setTriggerClassName(readUniqueIdentifier());
    }
    return command;
}

Example 38 with Command

use of org.h2.command.Command in project h2database by h2database.

the class Parser method parseWithStatementOrQuery.

private Prepared parseWithStatementOrQuery() {
    int paramIndex = parameters.size();
    Prepared command = parseWith();
    ArrayList<Parameter> params = New.arrayList();
    for (int i = paramIndex, size = parameters.size(); i < size; i++) {
        params.add(parameters.get(i));
    }
    command.setParameterList(params);
    if (command instanceof Query) {
        Query query = (Query) command;
        query.init();
    }
    return command;
}

Example 39 with Command

use of org.h2.command.Command in project h2database by h2database.

the class Parser method parseGrantRevoke.

private GrantRevoke parseGrantRevoke(int operationType) {
    GrantRevoke command = new GrantRevoke(session);
    command.setOperationType(operationType);
    boolean tableClauseExpected = addRoleOrRight(command);
    while (readIf(",")) {
        addRoleOrRight(command);
        if (command.isRightMode() && command.isRoleMode()) {
            throw DbException.get(ErrorCode.ROLES_AND_RIGHT_CANNOT_BE_MIXED);
        }
    }
    if (tableClauseExpected) {
        if (readIf("ON")) {
            if (readIf("SCHEMA")) {
                Schema schema = database.getSchema(readAliasIdentifier());
                command.setSchema(schema);
            } else {
                do {
                    Table table = readTableOrView();
                    command.addTable(table);
                } while (readIf(","));
            }
        }
    }
    if (operationType == CommandInterface.GRANT) {
        read("TO");
    } else {
        read("FROM");
    }
    command.setGranteeName(readUniqueIdentifier());
    return command;
}

Example 40 with Command

use of org.h2.command.Command in project h2database by h2database.

the class Parser method parseCommit.

private TransactionCommand parseCommit() {
    TransactionCommand command;
    if (readIf("TRANSACTION")) {
        command = new TransactionCommand(session, CommandInterface.COMMIT_TRANSACTION);
        command.setTransactionName(readUniqueIdentifier());
        return command;
    }
    command = new TransactionCommand(session, CommandInterface.COMMIT);
    readIf("WORK");
    return command;
}