Search in sources :

Example 36 with MergedResult

use of io.shardingjdbc.core.merger.MergedResult in project sharding-jdbc by shardingjdbc.

the class IteratorStreamMergedResultTest method assertNextForLastResultSetsNotEmptyOnly.

@Test
public void assertNextForLastResultSetsNotEmptyOnly() throws SQLException {
    when(queryResults.get(2).next()).thenReturn(true, false);
    mergeEngine = new DQLMergeEngine(queryResults, selectStatement);
    MergedResult actual = mergeEngine.merge();
    assertTrue(actual.next());
    assertFalse(actual.next());
}
Also used : MergedResult(io.shardingjdbc.core.merger.MergedResult) DQLMergeEngine(io.shardingjdbc.core.merger.dql.DQLMergeEngine) Test(org.junit.Test)

Example 37 with MergedResult

use of io.shardingjdbc.core.merger.MergedResult in project sharding-jdbc by shardingjdbc.

the class OrderByStreamMergedResultTest method assertNextForSomeResultSetsEmpty.

@Test
public void assertNextForSomeResultSetsEmpty() throws SQLException {
    mergeEngine = new DQLMergeEngine(queryResults, selectStatement);
    when(queryResults.get(0).next()).thenReturn(true, false);
    when(queryResults.get(0).getValue(1, Object.class)).thenReturn("2");
    when(queryResults.get(2).next()).thenReturn(true, true, false);
    when(queryResults.get(2).getValue(1, Object.class)).thenReturn("1", "1", "3", "3");
    MergedResult actual = mergeEngine.merge();
    assertTrue(actual.next());
    assertThat(actual.getValue(1, Object.class).toString(), is("1"));
    assertTrue(actual.next());
    assertThat(actual.getValue(1, Object.class).toString(), is("2"));
    assertTrue(actual.next());
    assertThat(actual.getValue(1, Object.class).toString(), is("3"));
    assertFalse(actual.next());
}
Also used : MergedResult(io.shardingjdbc.core.merger.MergedResult) DQLMergeEngine(io.shardingjdbc.core.merger.dql.DQLMergeEngine) Test(org.junit.Test)

Example 38 with MergedResult

use of io.shardingjdbc.core.merger.MergedResult in project sharding-jdbc by shardingjdbc.

the class LimitDecoratorMergedResultTest method assertNextWithRowCount.

@Test
public void assertNextWithRowCount() throws SQLException {
    Limit limit = new Limit(DatabaseType.MySQL);
    limit.setOffset(new LimitValue(2, -1, true));
    limit.setRowCount(new LimitValue(2, -1, false));
    selectStatement.setLimit(limit);
    mergeEngine = new DQLMergeEngine(queryResults, selectStatement);
    MergedResult actual = mergeEngine.merge();
    assertTrue(actual.next());
    assertTrue(actual.next());
    assertFalse(actual.next());
}
Also used : MergedResult(io.shardingjdbc.core.merger.MergedResult) DQLMergeEngine(io.shardingjdbc.core.merger.dql.DQLMergeEngine) Limit(io.shardingjdbc.core.parsing.parser.context.limit.Limit) LimitValue(io.shardingjdbc.core.parsing.parser.context.limit.LimitValue) Test(org.junit.Test)

Example 39 with MergedResult

use of io.shardingjdbc.core.merger.MergedResult in project sharding-jdbc by shardingjdbc.

the class LimitDecoratorMergedResultTest method assertNextForSkipAll.

@Test
public void assertNextForSkipAll() throws SQLException {
    Limit limit = new Limit(DatabaseType.MySQL);
    limit.setOffset(new LimitValue(Integer.MAX_VALUE, -1, true));
    selectStatement.setLimit(limit);
    mergeEngine = new DQLMergeEngine(queryResults, selectStatement);
    MergedResult actual = mergeEngine.merge();
    assertFalse(actual.next());
}
Also used : MergedResult(io.shardingjdbc.core.merger.MergedResult) DQLMergeEngine(io.shardingjdbc.core.merger.dql.DQLMergeEngine) Limit(io.shardingjdbc.core.parsing.parser.context.limit.Limit) LimitValue(io.shardingjdbc.core.parsing.parser.context.limit.LimitValue) Test(org.junit.Test)

Example 40 with MergedResult

use of io.shardingjdbc.core.merger.MergedResult in project sharding-jdbc by shardingjdbc.

the class RowNumberDecoratorMergedResultTest method assertNextForSkipAll.

@Test
public void assertNextForSkipAll() throws SQLException {
    Limit limit = new Limit(DatabaseType.Oracle);
    limit.setOffset(new LimitValue(Integer.MAX_VALUE, -1, true));
    selectStatement.setLimit(limit);
    mergeEngine = new DQLMergeEngine(queryResults, selectStatement);
    MergedResult actual = mergeEngine.merge();
    assertFalse(actual.next());
}
Also used : MergedResult(io.shardingjdbc.core.merger.MergedResult) DQLMergeEngine(io.shardingjdbc.core.merger.dql.DQLMergeEngine) Limit(io.shardingjdbc.core.parsing.parser.context.limit.Limit) LimitValue(io.shardingjdbc.core.parsing.parser.context.limit.LimitValue) Test(org.junit.Test)

Aggregations

MergedResult (io.shardingjdbc.core.merger.MergedResult)42 Test (org.junit.Test)41 Limit (io.shardingjdbc.core.parsing.parser.context.limit.Limit)27 DQLMergeEngine (io.shardingjdbc.core.merger.dql.DQLMergeEngine)26 GroupByMemoryMergedResult (io.shardingjdbc.core.merger.dql.groupby.GroupByMemoryMergedResult)15 GroupByStreamMergedResult (io.shardingjdbc.core.merger.dql.groupby.GroupByStreamMergedResult)15 IteratorStreamMergedResult (io.shardingjdbc.core.merger.dql.iterator.IteratorStreamMergedResult)15 OrderByStreamMergedResult (io.shardingjdbc.core.merger.dql.orderby.OrderByStreamMergedResult)15 LimitDecoratorMergedResult (io.shardingjdbc.core.merger.dql.pagination.LimitDecoratorMergedResult)15 RowNumberDecoratorMergedResult (io.shardingjdbc.core.merger.dql.pagination.RowNumberDecoratorMergedResult)15 TopAndRowNumberDecoratorMergedResult (io.shardingjdbc.core.merger.dql.pagination.TopAndRowNumberDecoratorMergedResult)15 LimitValue (io.shardingjdbc.core.parsing.parser.context.limit.LimitValue)11 OrderItem (io.shardingjdbc.core.parsing.parser.context.OrderItem)9 AggregationSelectItem (io.shardingjdbc.core.parsing.parser.context.selectitem.AggregationSelectItem)3 BigDecimal (java.math.BigDecimal)3 QueryResult (io.shardingjdbc.core.merger.QueryResult)2 TestQueryResult (io.shardingjdbc.core.merger.fixture.TestQueryResult)2 MySQLPacketQueryResult (io.shardingjdbc.proxy.backend.mysql.MySQLPacketQueryResult)1 DatabaseProtocolPacket (io.shardingjdbc.proxy.transport.common.packet.DatabaseProtocolPacket)1 ErrPacket (io.shardingjdbc.proxy.transport.mysql.packet.generic.ErrPacket)1