Examples with Insert io.pravega.client.tables.Insert used on opensource projects

Example 6 with Insert

use of io.pravega.client.tables.Insert in project pravega by pravega.

the class KeyValueTableTestBase method testIterators.

@Test
public void testIterators() {
    @Cleanup val kvt = createKeyValueTable();
    val iteration = new AtomicInteger(0);
    // Populate everything.
    val versions = new Versions();
    forEveryPrimaryKey((pk, secondaryKeys) -> {
        List<TableModification> inserts = secondaryKeys.stream().map(sk -> new Insert(new TableKey(pk, sk), getValue(pk, sk, iteration.get()))).collect(Collectors.toList());
        val keyVersions = kvt.update(inserts).join();
        for (int i = 0; i < inserts.size(); i++) {
            versions.add(getUniqueKeyId(pk, inserts.get(i).getKey().getSecondaryKey()), keyVersions.get(i));
        }
    });
    // Check the Primary Key iterator.
    checkPrimaryKeyIterator(kvt, versions, iteration.get());
    // Check the Global iterators.
    checkGlobalIterator(kvt, versions, iteration.get());
}