fixed progress bar for replication listener

redis · May 4, 2021 · 9f293ec · 9f293ec
1 parent 9fc2357
commit 9f293ec
Showing 1 changed file with 1 addition and 7 deletions.
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
@@ -65,6 +65,7 @@ private Flow replicationFlow(StepBuilderFactory stepBuilderFactory) {
     private TaskletStep scanStep(StepBuilderFactory stepBuilderFactory) {
         StepBuilder stepBuilder = stepBuilderFactory.get("scan-replication-step");
         RiotStepBuilder<T, T> 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 extends KeyValueItemReader.KeyValueItemReaderBuilder> B configure(B
         return readerOptions.configure(builder);
     }
 
-    @Override
-    protected <I, O> RiotStepBuilder<I, O> riotStep(StepBuilder stepBuilder, String taskName) {
-        RiotStepBuilder<I, O> riotStepBuilder = super.riotStep(stepBuilder, taskName);
-        riotStepBuilder.initialMax(readerOptions.initialMaxSupplier(getRedisOptions()));
-        return riotStepBuilder;
-    }
-
 }