package com.netflix.astyanax.cql.test;
import java.util.concurrent.atomic.AtomicLong;
import org.junit.AfterClass;
import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.netflix.astyanax.ColumnListMutation;
import com.netflix.astyanax.ExceptionCallback;
import com.netflix.astyanax.MutationBatch;
import com.netflix.astyanax.RowCallback;
import com.netflix.astyanax.connectionpool.OperationResult;
import com.netflix.astyanax.connectionpool.exceptions.ConnectionException;
import com.netflix.astyanax.model.ColumnFamily;
import com.netflix.astyanax.model.Row;
import com.netflix.astyanax.model.Rows;
import com.netflix.astyanax.serializers.StringSerializer;
public class AllRowsQueryTest extends KeyspaceTests {
private static final Logger LOG = LoggerFactory.getLogger(AllRowsQueryTest.class);
public static ColumnFamily<String, String> CF_ALL_ROWS = ColumnFamily
.newColumnFamily(
"allrows",
StringSerializer.get(),
StringSerializer.get());
@BeforeClass
public static void init() throws Exception {
initContext();
keyspace.createColumnFamily(CF_ALL_ROWS, null);
CF_ALL_ROWS.describe(keyspace);
/** POPULATE ROWS FOR TESTS */
MutationBatch m = keyspace.prepareMutationBatch();
for (char keyName = 'A'; keyName <= 'Z'; keyName++) {
String rowKey = Character.toString(keyName);
ColumnListMutation<String> cfmStandard = m.withRow(CF_ALL_ROWS, rowKey);
for (char cName = 'a'; cName <= 'z'; cName++) {
cfmStandard.putColumn(Character.toString(cName), (int) (cName - 'a') + 1, null);
}
m.withCaching(true);
m.execute();
m.discardMutations();
}
}
@AfterClass
public static void tearDown() throws Exception {
keyspace.dropColumnFamily(CF_ALL_ROWS);
}
@Test
public void getAllWithCallback() throws Exception {
final AtomicLong counter = new AtomicLong();
keyspace.prepareQuery(CF_ALL_ROWS).getAllRows()
.setRowLimit(30)
.setRepeatLastToken(false)
.setConcurrencyLevel(2)
//.withColumnRange(new RangeBuilder().setLimit(2).build())
.executeWithCallback(new RowCallback<String, String>() {
@Override
public void success(Rows<String, String> rows) {
for (Row<String, String> row : rows) {
LOG.info("ROW: " + row.getKey() + " "
+ row.getColumns().size());
counter.incrementAndGet();
}
}
@Override
public boolean failure(ConnectionException e) {
LOG.error(e.getMessage(), e);
return false;
}
});
LOG.info("Read " + counter.get() + " keys");
Assert.assertEquals(26, counter.get());
}
@Test
public void getAll() throws Exception {
AtomicLong counter = new AtomicLong(0);
OperationResult<Rows<String, String>> rows = keyspace
.prepareQuery(CF_ALL_ROWS).getAllRows().setConcurrencyLevel(2).setRowLimit(30)
//.withColumnRange(new RangeBuilder().setLimit(0).build())
.setExceptionCallback(new ExceptionCallback() {
@Override
public boolean onException(ConnectionException e) {
Assert.fail(e.getMessage());
return true;
}
}).execute();
for (Row<String, String> row : rows.getResult()) {
counter.incrementAndGet();
LOG.info("ROW: " + row.getKey() + " " + row.getColumns().size());
}
LOG.info("Read " + counter.get() + " keys");
Assert.assertEquals(26, counter.get());
}
}