Examples with VirtualLeafRecord com.swirlds.virtualmap.datasource.VirtualLeafRecord used on opensource projects

Example 6 with VirtualLeafRecord

use of com.swirlds.virtualmap.datasource.VirtualLeafRecord in project hedera-services by hashgraph.

the class VirtualDataSourceNewAPIBench method setup.

@Setup(Level.Trial)
public void setup() {
    System.out.println("------- Setup -----------------------------");
    storePath = Path.of("store-" + impl);
    try {
        final boolean storeExists = Files.exists(storePath);
        // get slot index suppliers
        switch(impl) {
            case "lmdb":
                dataSource = new VFCDataSourceLmdb<>(// max seralized size
                1 + 8 + 32, // max seralized size
                ContractKey::new, ContractValue.SERIALIZED_SIZE, ContractValue::new, storePath);
                break;
            // storePath);
            case "jasperdb":
                VirtualLeafRecordSerializer<ContractKey, ContractValue> virtualLeafRecordSerializer = new VirtualLeafRecordSerializer<>((short) 1, DigestType.SHA_384, (short) 1, DataFileCommon.VARIABLE_DATA_SIZE, new ContractKeySupplier(), (short) 1, ContractValue.SERIALIZED_SIZE, new ContractValueSupplier(), true);
                JasperDbBuilder<ContractKey, ContractValue> dbBuilder = new JasperDbBuilder<>();
                dbBuilder.virtualLeafRecordSerializer(virtualLeafRecordSerializer).virtualInternalRecordSerializer(new VirtualInternalRecordSerializer()).keySerializer(new ContractKeySerializer()).storageDir(storePath).maxNumOfKeys(numEntities + 10_000_000).preferDiskBasedIndexes(false).internalHashesRamToDiskThreshold(Long.MAX_VALUE).mergingEnabled(true);
                dataSource = dbBuilder.build("jdb", "4ApiBench");
                break;
            default:
                throw new IllegalStateException("Unexpected value: " + impl);
        }
        ;
        // create data
        if (!storeExists) {
            System.out.println("================================================================================");
            System.out.println("Creating data ...");
            // create internal nodes and leaves
            long iHaveWritten = 0;
            while (iHaveWritten < numEntities) {
                final long start = System.currentTimeMillis();
                final long batchSize = Math.min(WRITE_BATCH_SIZE, numEntities - iHaveWritten);
                dataSource.saveRecords(numEntities, numEntities * 2, LongStream.range(iHaveWritten, iHaveWritten + batchSize).mapToObj(i -> new VirtualInternalRecord(i, hash((int) i))), LongStream.range(iHaveWritten, iHaveWritten + batchSize).mapToObj(i -> new VirtualLeafRecord<>(i + numEntities, hash((int) i), new ContractKey(i, i), new ContractValue(i))), Stream.empty());
                iHaveWritten += batchSize;
                printUpdate(start, batchSize, ContractValue.SERIALIZED_SIZE, "Created " + iHaveWritten + " Nodes");
            }
            System.out.println("================================================================================");
            // set nextPath
            nextPath = numEntities;
            // let merge catch up
            try {
                System.out.println("Waiting for merge");
                Thread.sleep(30000);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        } else {
            System.out.println("Loaded existing data");
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
}

Example 7 with VirtualLeafRecord

use of com.swirlds.virtualmap.datasource.VirtualLeafRecord in project hedera-services by hashgraph.

the class DatabaseState method setupDatabase.

@Setup(Level.Trial)
public void setupDatabase() throws IOException {
    System.out.println("dataSourcePath = " + dataSourcePath + " mergingEnabled=" + mergingEnabled);
    if (Files.exists(dataSourcePath)) {
        System.err.println("!!!!!!!!!!!!! Deleting old db.");
        deleteDirectoryAndContents(dataSourcePath);
    }
    if (Files.exists(dataSourceSnapshotPath)) {
        System.err.println("!!!!!!!!!!!!! Deleting old db snapshot.");
        deleteDirectoryAndContents(dataSourceSnapshotPath);
    }
    // create data source
    VirtualLeafRecordSerializer<ContractKey, ContractValue> virtualLeafRecordSerializer = new VirtualLeafRecordSerializer<>((short) 1, DigestType.SHA_384, (short) 1, DataFileCommon.VARIABLE_DATA_SIZE, new ContractKeySupplier(), (short) 1, ContractValue.SERIALIZED_SIZE, new ContractValueSupplier(), true);
    JasperDbBuilder<ContractKey, ContractValue> dbBuilder = new JasperDbBuilder<>();
    dbBuilder.virtualLeafRecordSerializer(virtualLeafRecordSerializer).virtualInternalRecordSerializer(new VirtualInternalRecordSerializer()).keySerializer(new ContractKeySerializer()).storageDir(dataSourcePath).maxNumOfKeys(500_000_000).preferDiskBasedIndexes(false).internalHashesRamToDiskThreshold(0).mergingEnabled(mergingEnabled);
    dataSource = dbBuilder.build("jdb", "4dbState");
    // populate with initial data
    System.out.printf("Creating initial data set of %,d leaves\n", initialDataSize);
    progressPercentage = 0;
    final long firstLeafPath = initialDataSize;
    final long lastLeafPath = firstLeafPath + initialDataSize;
    var internalRecordStream = LongStream.range(0, firstLeafPath).mapToObj(path -> new VirtualInternalRecord(path, hash((int) path))).peek(internalRecord -> printProgress(internalRecord.getPath(), lastLeafPath));
    var leafRecordStream = LongStream.range(firstLeafPath, lastLeafPath + 1).mapToObj(path -> new VirtualLeafRecord<>(path, hash((int) path), new ContractKey(path, path), new ContractValue(path))).peek(leaf -> printProgress(leaf.getPath(), lastLeafPath));
    dataSource.saveRecords(firstLeafPath, lastLeafPath, internalRecordStream, leafRecordStream, Stream.empty());
    System.out.printf("Done creating initial data set of %,d leaves\n", initialDataSize);
}

Example 8 with VirtualLeafRecord

use of com.swirlds.virtualmap.datasource.VirtualLeafRecord in project hedera-services by hashgraph.

the class VFCDataSourceLmdb method loadLeafRecord.

// ==================================================================================================================
// Private methods
/**
 * load a leaf record by path, using the provided key or if null deserializing the key.
 */
private VirtualLeafRecord<K, V> loadLeafRecord(Txn<ByteBuffer> txn, long path, K key) throws IOException {
    ByteBuffer keyHashValueBuffer = leafPathToKeyHashValueMap.get(txn, getPathNativeOrderBytes(path));
    if (keyHashValueBuffer == null) {
        return null;
    }
    // deserialize
    keyHashValueBuffer.clear();
    // deserialize key
    if (key != null) {
        // jump past the key because we don't need to deserialize it
        keyHashValueBuffer.position(keySizeBytes);
    } else {
        final int keySerializationVersion = keyHashValueBuffer.getInt();
        key = keyConstructor.get();
        key.deserialize(keyHashValueBuffer, keySerializationVersion);
    }
    // deserialize hash
    final Hash hash = new Hash(DigestType.SHA_384);
    keyHashValueBuffer.get(hash.getValue());
    // deserialize value
    final int valueSerializationVersion = keyHashValueBuffer.getInt();
    final V value = valueConstructor.get();
    value.deserialize(keyHashValueBuffer, valueSerializationVersion);
    // return new VirtualLeafRecord
    return new VirtualLeafRecord<>(path, hash, key, value);
}