How to use

createSplitReader

method

in

co.cask.cdap.data2.dataset2.lib.table.ObjectStoreDataset

public RecordScanner<KeyValue<byte[], T>> createSplitRecordScanner(Split split) {
 return Scannables.splitRecordScanner(createSplitReader(split), new ObjectRecordMaker());
}

public RecordScanner<KeyValue<byte[], T>> createSplitRecordScanner(Split split) {
 return Scannables.splitRecordScanner(createSplitReader(split), new ObjectRecordMaker());
}

private void verifySplits(ObjectStoreDataset<String> t, List<Split> splits, SortedSet<Long> keysToVerify)
 throws InterruptedException {
 // read each split and verify the keys, remove all read keys from the set
 for (Split split : splits) {
  SplitReader<byte[], String> reader = t.createSplitReader(split);
  reader.initialize(split);
  while (reader.nextKeyValue()) {
   byte[] key = reader.getCurrentKey();
   String value = reader.getCurrentValue();
   // verify each row has the two columns written
   Assert.assertEquals(Long.toString(Bytes.toLong(key)), value);
   Assert.assertTrue(keysToVerify.remove(Bytes.toLong(key)));
  }
 }
 // verify all keys have been read
 if (!keysToVerify.isEmpty()) {
  System.out.println("Remaining [" + keysToVerify.size() + "]: " + keysToVerify);
 }
 Assert.assertTrue(keysToVerify.isEmpty());
}