Examples with OffsetToken io.shardingjdbc.core.parsing.parser.token.OffsetToken used on opensource projects

Example 11 with OffsetToken

use of io.shardingjdbc.core.parsing.parser.token.OffsetToken in project sharding-jdbc by shardingjdbc.

the class SQLRewriteEngine method rewrite.

/**
 * rewrite SQL.
 *
 * @param isRewriteLimit is rewrite limit
 * @return SQL builder
 */
public SQLBuilder rewrite(final boolean isRewriteLimit) {
    SQLBuilder result = new SQLBuilder();
    if (sqlTokens.isEmpty()) {
        result.appendLiterals(originalSQL);
        return result;
    }
    int count = 0;
    sortByBeginPosition();
    for (SQLToken each : sqlTokens) {
        if (0 == count) {
            result.appendLiterals(originalSQL.substring(0, each.getBeginPosition()));
        }
        if (each instanceof TableToken) {
            appendTablePlaceholder(result, (TableToken) each, count, sqlTokens);
        } else if (each instanceof SchemaToken) {
            appendSchemaPlaceholder(result, (SchemaToken) each, count, sqlTokens);
        } else if (each instanceof IndexToken) {
            appendIndexPlaceholder(result, (IndexToken) each, count, sqlTokens);
        } else if (each instanceof ItemsToken) {
            appendItemsToken(result, (ItemsToken) each, count, sqlTokens);
        } else if (each instanceof RowCountToken) {
            appendLimitRowCount(result, (RowCountToken) each, count, sqlTokens, isRewriteLimit);
        } else if (each instanceof OffsetToken) {
            appendLimitOffsetToken(result, (OffsetToken) each, count, sqlTokens, isRewriteLimit);
        } else if (each instanceof OrderByToken) {
            appendOrderByToken(result, count, sqlTokens);
        }
        count++;
    }
    return result;
}

Example 12 with OffsetToken

use of io.shardingjdbc.core.parsing.parser.token.OffsetToken in project sharding-jdbc by shardingjdbc.

the class SQLRewriteEngineTest method assertRewriteForLimitForMemoryGroupBy.

@Test
public void assertRewriteForLimitForMemoryGroupBy() {
    selectStatement.setLimit(new Limit(DatabaseType.MySQL));
    selectStatement.getLimit().setOffset(new LimitValue(2, -1, true));
    selectStatement.getLimit().setRowCount(new LimitValue(2, -1, false));
    selectStatement.getOrderByItems().add(new OrderItem("x", "id", OrderDirection.ASC, OrderDirection.ASC, Optional.<String>absent()));
    selectStatement.getGroupByItems().add(new OrderItem("x", "id", OrderDirection.DESC, OrderDirection.ASC, Optional.<String>absent()));
    selectStatement.getSqlTokens().add(new TableToken(17, "table_x"));
    selectStatement.getSqlTokens().add(new OffsetToken(33, 2));
    selectStatement.getSqlTokens().add(new RowCountToken(36, 2));
    SQLRewriteEngine rewriteEngine = new SQLRewriteEngine(shardingRule, "SELECT x.id FROM table_x x LIMIT 2, 2", DatabaseType.MySQL, selectStatement);
    assertThat(rewriteEngine.rewrite(true).toSQL(tableTokens, null), is("SELECT x.id FROM table_1 x LIMIT 0, 2147483647"));
}

Example 13 with OffsetToken

use of io.shardingjdbc.core.parsing.parser.token.OffsetToken in project sharding-jdbc by shardingjdbc.

the class SQLRewriteEngineTest method assertRewriteForRowNumForNotRewriteLimit.

@Test
public void assertRewriteForRowNumForNotRewriteLimit() {
    selectStatement.setLimit(new Limit(DatabaseType.Oracle));
    selectStatement.getLimit().setOffset(new LimitValue(2, -1, true));
    selectStatement.getLimit().setRowCount(new LimitValue(4, -1, false));
    selectStatement.getSqlTokens().add(new TableToken(68, "table_x"));
    selectStatement.getSqlTokens().add(new OffsetToken(119, 2));
    selectStatement.getSqlTokens().add(new RowCountToken(98, 4));
    SQLRewriteEngine rewriteEngine = new SQLRewriteEngine(shardingRule, "SELECT * FROM (SELECT row_.*, rownum rownum_ FROM (SELECT x.id FROM table_x x) row_ WHERE rownum<=4) t WHERE t.rownum_>2", DatabaseType.Oracle, selectStatement);
    assertThat(rewriteEngine.rewrite(false).toSQL(tableTokens, null), is("SELECT * FROM (SELECT row_.*, rownum rownum_ FROM (SELECT x.id FROM table_1 x) row_ WHERE rownum<=4) t WHERE t.rownum_>2"));
}

Example 14 with OffsetToken

use of io.shardingjdbc.core.parsing.parser.token.OffsetToken in project sharding-jdbc by shardingjdbc.

the class SelectStatement method resetLimitTokens.

private void resetLimitTokens(final SelectStatement selectStatement, final List<SQLToken> limitSQLTokens) {
    int count = 0;
    List<Integer> toBeRemovedIndexes = new LinkedList<>();
    for (SQLToken each : selectStatement.getSqlTokens()) {
        if (each instanceof RowCountToken || each instanceof OffsetToken) {
            toBeRemovedIndexes.add(count);
        }
        count++;
    }
    for (int each : toBeRemovedIndexes) {
        selectStatement.getSqlTokens().remove(each);
    }
    selectStatement.getSqlTokens().addAll(limitSQLTokens);
}

Example 15 with OffsetToken

use of io.shardingjdbc.core.parsing.parser.token.OffsetToken in project sharding-jdbc by shardingjdbc.

the class MySQLLimitClauseParser method getLimitWithOffset.

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