diff --git a/riot-redis/src/main/java/com/redislabs/riot/redis/AbstractReplicateCommand.java b/riot-redis/src/main/java/com/redislabs/riot/redis/AbstractReplicateCommand.java index cd6b0bfdf..2b72fca37 100644 --- a/riot-redis/src/main/java/com/redislabs/riot/redis/AbstractReplicateCommand.java +++ b/riot-redis/src/main/java/com/redislabs/riot/redis/AbstractReplicateCommand.java @@ -65,6 +65,7 @@ private Flow replicationFlow(StepBuilderFactory stepBuilderFactory) { private TaskletStep scanStep(StepBuilderFactory stepBuilderFactory) { StepBuilder stepBuilder = stepBuilderFactory.get("scan-replication-step"); RiotStepBuilder scanStep = riotStep(stepBuilder, "Scanning"); + scanStep.initialMax(readerOptions.initialMaxSupplier(getRedisOptions())); return scanStep.reader(reader(getRedisOptions())).writer(writer(targetRedisOptions)).build().build(); } @@ -130,11 +131,4 @@ protected B configure(B return readerOptions.configure(builder); } - @Override - protected RiotStepBuilder riotStep(StepBuilder stepBuilder, String taskName) { - RiotStepBuilder riotStepBuilder = super.riotStep(stepBuilder, taskName); - riotStepBuilder.initialMax(readerOptions.initialMaxSupplier(getRedisOptions())); - return riotStepBuilder; - } - }