DatasetWriter<TestEntity> writer = ds.newWriter();
writer.open();
try {
for (int i = 2; i < 10; ++i) {
TestEntity entity = createSpecificEntity(i);
writer.write(entity);
}
} finally {
writer.close();
}
// ensure the new entities are what we expect with get operations
for (int i = 0; i < 10; ++i) {
String iStr = Long.toString(i);
Key key = new Key.Builder(ds)
.add("part1", "part1_" + iStr)
.add("part2", "part2_" + iStr).build();
compareEntitiesWithString(i, ds.get(key));
}
// ensure the new entities are what we expect with scan operations
int cnt = 0;
DatasetReader<TestEntity> reader = ds.newReader();
reader.open();
try {
for (TestEntity entity : reader) {
compareEntitiesWithString(cnt, entity);
cnt++;
}
assertEquals(10, cnt);
} finally {
reader.close();
}
Key key = new Key.Builder(ds)
.add("part1", "part1_5")
.add("part2", "part2_5").build();
// test delete
ds.delete(key);
TestEntity deletedRecord = ds.get(key);
assertNull(deletedRecord);
}