in src/main/java/org/apache/flink/state/benchmark/RocksdbStateBackendRescalingBenchmarkExecutor.java [49:65]
public void setUp() throws Exception {
EmbeddedRocksDBStateBackend stateBackend = new EmbeddedRocksDBStateBackend(true);
benchmark =
new RescalingBenchmarkBuilder<byte[]>()
.setMaxParallelism(128)
.setParallelismBefore(rescaleType.getParallelismBefore())
.setParallelismAfter(rescaleType.getParallelismAfter())
.setManagedMemorySize(512 * 1024 * 1024)
.setCheckpointStorageAccess(
new FileSystemCheckpointStorage("file://" + prepareDirectory("rescaleDb").getAbsolutePath())
.createCheckpointStorage(new JobID()))
.setStateBackend(stateBackend)
.setStreamRecordGenerator(new ByteArrayRecordGenerator(numberOfKeys, keyLen))
.setStateProcessFunctionSupplier(TestKeyedFunction::new)
.build();
benchmark.setUp();
}