Java Code Examples for com.googlecode.cqengine.resultset.ResultSet#size()

/**
 * Returns a Collection-like view of the given ResultSet.
 * <p/>
 * The collection simply delegates to the ResultSet, which in turn will reflect
 * any changes made to the underlying IndexedCollection by other threads.
 * For example consecutive calls to the size() method
 * may return different values if objects are added to or removed from the IndexedCollection.
 *
 * @param resultSet The ResultSet to wrap
 * @return A Collection-like view of the given ResultSet
 */
public static <O> Collection<O> asCollection(final ResultSet<O> resultSet) {
    return new AbstractCollection<O>() {
        @Override
        public Iterator<O> iterator() {
            return resultSet.iterator();
        }
        @Override
        public int size() {
            return resultSet.size();
        }

        @Override
        public boolean contains(Object o) {
            @SuppressWarnings("unchecked")
            O object = (O)o;
            return resultSet.contains(object);
        }

        @Override
        public boolean isEmpty() {
            return resultSet.isEmpty();
        }
    };
}
 

@Test
public void testNewResultSet_FilterQuery_Size() throws Exception{

    // Mocks
    FilterQuery<Car, String> filterQuery = mockFilterQuery();
    ConnectionManager connectionManager = mock(ConnectionManager.class);
    Connection connection = mock(Connection.class);
    Statement statement = mock(PreparedStatement.class);
    java.sql.ResultSet resultSet = mock(java.sql.ResultSet.class);

    // Behaviour//
    when(connectionManager.getConnection(any(SQLiteIndex.class), anyQueryOptions())).thenReturn(connection);
    when(connection.createStatement()).thenReturn(statement);
    when(statement.executeQuery("SELECT objectKey, value FROM " + TABLE_NAME + " ORDER BY objectKey;")).thenReturn(resultSet);
    when(resultSet.getStatement()).thenReturn(statement);
    when(resultSet.next()).thenReturn(true).thenReturn(true).thenReturn(true).thenReturn(true).thenReturn(true).thenReturn(true).thenReturn(false);
    when(resultSet.getInt(1)).thenReturn(1).thenReturn(1).thenReturn(2).thenReturn(3).thenReturn(4).thenReturn(5);
    when(resultSet.getString(2)).thenReturn("abs").thenReturn("gps").thenReturn("airbags").thenReturn("abs").thenReturn("").thenReturn("gps");

    ResultSet<Car> carsWithAbs = new SQLiteIndex<String, Car, Integer>(
            Car.FEATURES,
            OBJECT_TO_ID,
            ID_TO_OBJECT,
            "")

            .retrieve(filterQuery, createQueryOptions(connectionManager));


    assertNotNull(carsWithAbs);
    int size = carsWithAbs.size();

    assertEquals(3, size);
    verify(connection, times(0)).close();

}
 

public Optional<ShipData> getShipFromChunk(long chunkLong) {
    Query<ShipData> query = equal(ShipData.CHUNKS, chunkLong);
    ResultSet<ShipData> resultSet = allShips.retrieve(query);

    if (resultSet.size() > 1) {
        throw new IllegalStateException("How the heck did we get 2 or more ships both managing the chunk at " + chunkLong);
    }
    if (resultSet.isEmpty()) {
        return Optional.empty();
    } else {
        return Optional.of(resultSet.uniqueResult());
    }
}
 

@Test
public void testNewResultSet_Size() throws Exception{

    // Mocks
    ConnectionManager connectionManager = mock(ConnectionManager.class);
    Connection connection = mock(Connection.class);
    PreparedStatement preparedStatement = mock(PreparedStatement.class);
    java.sql.ResultSet resultSet = mock(java.sql.ResultSet.class);

    // Behaviour
    when(connectionManager.getConnection(any(SQLiteIndex.class), anyQueryOptions())).thenReturn(connection);
    when(connection.prepareStatement("SELECT COUNT(1) AS countDistinct FROM (SELECT objectKey FROM " + TABLE_NAME + " WHERE value = ? GROUP BY objectKey);")).thenReturn(preparedStatement);
    when(preparedStatement.executeQuery()).thenReturn(resultSet);
    when(resultSet.getStatement()).thenReturn(preparedStatement);
    when(resultSet.next()).thenReturn(true);
    when(resultSet.getInt(1)).thenReturn(3);

    ResultSet<Car> carsWithAbs = new SQLiteIndex<String, Car, Integer>(
            Car.FEATURES,
            OBJECT_TO_ID,
            ID_TO_OBJECT,
            "")

            .retrieve(equal(Car.FEATURES, "abs"), createQueryOptions(connectionManager));


    assertNotNull(carsWithAbs);
    int size = carsWithAbs.size();

    assertEquals(3, size);
    verify(connection, times(0)).close();

}
 

@Override
public int runQueryCountResults_CQEngineStatistics() {
    ResultSet<Car> results = indexedCollection.retrieve(query);
    return results.size();
}
 

@Override
public int runQueryCountResults_CQEngineStatistics() {
    ResultSet<Car> results = indexedCollection.retrieve(query);
    return results.size();
}
 

@Override
public int runQueryCountResults_CQEngineStatistics() {
    ResultSet<Car> results = indexedCollection.retrieve(query);
    return results.size();
}
 

@Override
public int runQueryCountResults_CQEngineStatistics() {
    ResultSet<Car> results = indexedCollection.retrieve(query);
    return results.size();
}
 

@Override
public int runQueryCountResults_CQEngineStatistics() {
    ResultSet<Car> results = indexedCollection.retrieve(query);
    return results.size();
}
 

@Override
public int runQueryCountResults_CQEngineStatistics() {
    ResultSet<Car> results = indexedCollection.retrieve(query);
    return results.size();
}
 

@Override
public int runQueryCountResults_CQEngineStatistics() {
    ResultSet<Car> results = indexedCollection.retrieve(query);
    return results.size();
}
 

@Override
public int runQueryCountResults_CQEngineStatistics() {
    ResultSet<Car> results = indexedCollection.retrieve(query);
    return results.size();
}
 

@Override
public int runQueryCountResults_CQEngineStatistics() {
    ResultSet<Car> results = indexedCollection.retrieve(query);
    return results.size();
}
 

@Override
public int runQueryCountResults_CQEngineStatistics() {
    ResultSet<Car> results = indexedCollection.retrieve(query);
    return results.size();
}
 

@Override
public int runQueryCountResults_CQEngineStatistics() {
    ResultSet<Car> results = indexedCollection.retrieve(query);
    return results.size();
}
 

@Override
public int runQueryCountResults_CQEngineStatistics() {
    ResultSet<Car> results = indexedCollection.retrieve(query);
    return results.size();
}
 

@Override
public int runQueryCountResults_CQEngineStatistics() {
    ResultSet<Car> results = indexedCollection.retrieve(query);
    return results.size();
}