Examples with SQLParsingException io.shardingjdbc.core.parsing.parser.exception.SQLParsingException used on opensource projects

Example 1 with SQLParsingException

use of io.shardingjdbc.core.parsing.parser.exception.SQLParsingException in project sharding-jdbc by shardingjdbc.

the class MySQLLimitClauseParser method getLimitWithComma.

private Limit getLimitWithComma(final int index, final int valueBeginPosition, final int value, final boolean isParameterForValue, final SelectStatement selectStatement) {
    int rowCountBeginPosition = lexerEngine.getCurrentToken().getEndPosition();
    int rowCountValue;
    int rowCountIndex = -1;
    boolean isParameterForRowCount = false;
    if (lexerEngine.equalAny(Literals.INT)) {
        rowCountValue = Integer.parseInt(lexerEngine.getCurrentToken().getLiterals());
        rowCountBeginPosition = rowCountBeginPosition - (rowCountValue + "").length();
    } else if (lexerEngine.equalAny(Symbol.QUESTION)) {
        rowCountIndex = -1 == index ? selectStatement.getParametersIndex() : index + 1;
        rowCountValue = -1;
        rowCountBeginPosition--;
        isParameterForRowCount = true;
    } else {
        throw new SQLParsingException(lexerEngine);
    }
    lexerEngine.nextToken();
    if (isParameterForValue) {
        selectStatement.increaseParametersIndex();
    } else {
        selectStatement.getSqlTokens().add(new OffsetToken(valueBeginPosition, value));
    }
    if (isParameterForRowCount) {
        selectStatement.increaseParametersIndex();
    } else {
        selectStatement.getSqlTokens().add(new RowCountToken(rowCountBeginPosition, rowCountValue));
    }
    Limit result = new Limit(DatabaseType.MySQL);
    result.setRowCount(new LimitValue(rowCountValue, rowCountIndex, false));
    result.setOffset(new LimitValue(value, index, true));
    return result;
}

Example 2 with SQLParsingException

use of io.shardingjdbc.core.parsing.parser.exception.SQLParsingException in project sharding-jdbc by shardingjdbc.

the class MySQLLimitClauseParser method parse.

/**
 * Parse limit.
 *
 * @param selectStatement select statement
 */
public void parse(final SelectStatement selectStatement) {
    if (!lexerEngine.skipIfEqual(MySQLKeyword.LIMIT)) {
        return;
    }
    int valueIndex = -1;
    int valueBeginPosition = lexerEngine.getCurrentToken().getEndPosition();
    int value;
    boolean isParameterForValue = false;
    if (lexerEngine.equalAny(Literals.INT)) {
        value = Integer.parseInt(lexerEngine.getCurrentToken().getLiterals());
        valueBeginPosition = valueBeginPosition - (value + "").length();
    } else if (lexerEngine.equalAny(Symbol.QUESTION)) {
        valueIndex = selectStatement.getParametersIndex();
        value = -1;
        valueBeginPosition--;
        isParameterForValue = true;
    } else {
        throw new SQLParsingException(lexerEngine);
    }
    lexerEngine.nextToken();
    if (lexerEngine.skipIfEqual(Symbol.COMMA)) {
        selectStatement.setLimit(getLimitWithComma(valueIndex, valueBeginPosition, value, isParameterForValue, selectStatement));
        return;
    }
    if (lexerEngine.skipIfEqual(MySQLKeyword.OFFSET)) {
        selectStatement.setLimit(getLimitWithOffset(valueIndex, valueBeginPosition, value, isParameterForValue, selectStatement));
        return;
    }
    if (isParameterForValue) {
        selectStatement.increaseParametersIndex();
    } else {
        selectStatement.getSqlTokens().add(new RowCountToken(valueBeginPosition, value));
    }
    Limit limit = new Limit(DatabaseType.MySQL);
    limit.setRowCount(new LimitValue(value, valueIndex, false));
    selectStatement.setLimit(limit);
}

Example 3 with SQLParsingException

use of io.shardingjdbc.core.parsing.parser.exception.SQLParsingException in project sharding-jdbc by shardingjdbc.

the class PostgreSQLLimitClauseParser method buildRowCount.

private Optional<LimitValue> buildRowCount(final SelectStatement selectStatement) {
    int parameterIndex = selectStatement.getParametersIndex();
    int rowCountValue = -1;
    int rowCountIndex = -1;
    int valueBeginPosition = lexerEngine.getCurrentToken().getEndPosition();
    if (lexerEngine.equalAny(DefaultKeyword.ALL)) {
        lexerEngine.nextToken();
    } else {
        if (lexerEngine.equalAny(Literals.INT, Literals.FLOAT)) {
            rowCountValue = NumberUtil.roundHalfUp(lexerEngine.getCurrentToken().getLiterals());
            valueBeginPosition = valueBeginPosition - (rowCountValue + "").length();
            selectStatement.getSqlTokens().add(new RowCountToken(valueBeginPosition, rowCountValue));
        } else if (lexerEngine.equalAny(Symbol.QUESTION)) {
            rowCountIndex = parameterIndex++;
            selectStatement.setParametersIndex(parameterIndex);
            rowCountValue = -1;
        } else {
            throw new SQLParsingException(lexerEngine);
        }
        lexerEngine.nextToken();
    }
    return Optional.of(new LimitValue(rowCountValue, rowCountIndex, false));
}

Example 4 with SQLParsingException

use of io.shardingjdbc.core.parsing.parser.exception.SQLParsingException in project sharding-jdbc by shardingjdbc.

the class PostgreSQLLimitClauseParser method buildOffset.

private Optional<LimitValue> buildOffset(final SelectStatement selectStatement) {
    int parameterIndex = selectStatement.getParametersIndex();
    int offsetValue = -1;
    int offsetIndex = -1;
    int offsetBeginPosition = lexerEngine.getCurrentToken().getEndPosition();
    if (lexerEngine.equalAny(Literals.INT, Literals.FLOAT)) {
        offsetValue = NumberUtil.roundHalfUp(lexerEngine.getCurrentToken().getLiterals());
        offsetBeginPosition = offsetBeginPosition - (offsetValue + "").length();
        selectStatement.getSqlTokens().add(new OffsetToken(offsetBeginPosition, offsetValue));
    } else if (lexerEngine.equalAny(Symbol.QUESTION)) {
        offsetIndex = parameterIndex++;
        selectStatement.setParametersIndex(parameterIndex);
    } else {
        throw new SQLParsingException(lexerEngine);
    }
    lexerEngine.nextToken();
    lexerEngine.skipIfEqual(DefaultKeyword.ROW, PostgreSQLKeyword.ROWS);
    return Optional.of(new LimitValue(offsetValue, offsetIndex, true));
}

Example 5 with SQLParsingException

use of io.shardingjdbc.core.parsing.parser.exception.SQLParsingException in project sharding-jdbc by shardingjdbc.

the class SQLServerOffsetClauseParser method parse.

/**
 * Parse offset.
 *
 * @param selectStatement select statement
 */
public void parse(final SelectStatement selectStatement) {
    if (!lexerEngine.skipIfEqual(SQLServerKeyword.OFFSET)) {
        return;
    }
    int offsetValue = -1;
    int offsetIndex = -1;
    if (lexerEngine.equalAny(Literals.INT)) {
        offsetValue = Integer.parseInt(lexerEngine.getCurrentToken().getLiterals());
    } else if (lexerEngine.equalAny(Symbol.QUESTION)) {
        offsetIndex = selectStatement.getParametersIndex();
        selectStatement.increaseParametersIndex();
    } else {
        throw new SQLParsingException(lexerEngine);
    }
    lexerEngine.nextToken();
    Limit limit = new Limit(DatabaseType.SQLServer);
    if (lexerEngine.skipIfEqual(DefaultKeyword.FETCH)) {
        lexerEngine.nextToken();
        int rowCountValue = -1;
        int rowCountIndex = -1;
        lexerEngine.nextToken();
        if (lexerEngine.equalAny(Literals.INT)) {
            rowCountValue = Integer.parseInt(lexerEngine.getCurrentToken().getLiterals());
        } else if (lexerEngine.equalAny(Symbol.QUESTION)) {
            rowCountIndex = selectStatement.getParametersIndex();
            selectStatement.increaseParametersIndex();
        } else {
            throw new SQLParsingException(lexerEngine);
        }
        lexerEngine.nextToken();
        lexerEngine.nextToken();
        limit.setRowCount(new LimitValue(rowCountValue, rowCountIndex, false));
        limit.setOffset(new LimitValue(offsetValue, offsetIndex, true));
    } else {
        limit.setOffset(new LimitValue(offsetValue, offsetIndex, true));
    }
    selectStatement.setLimit(limit);
}