feat: Using no-op writer with sleep option

redis · Nov 4, 2024 · 145f2be · 145f2be
1 parent c53b4ed
commit 145f2be
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/core/riot-core/src/main/java/com/redis/riot/core/AbstractJobCommand.java b/core/riot-core/src/main/java/com/redis/riot/core/AbstractJobCommand.java
@@ -247,15 +247,16 @@ private <I, O> ItemReader<? extends I> reader(Step<I, O> step) {
 	}
 
 	private <I, O> ItemWriter<? super O> writer(Step<I, O> step) {
+		ItemWriter<O> writer = step.getWriter();
 		if (stepArgs.isDryRun()) {
 			log.info("Using no-op writer");
-			return new NoopItemWriter<>();
+			writer = new NoopItemWriter<>();
 		}
 		if (stepArgs.getSleep() > 0) {
-			log.info("Throttling writer with sleep {}", stepArgs.getSleep());
-			return new ThrottledItemWriter<>(step.getWriter(), stepArgs.getSleep());
+			log.info("Throttling writer with sleep={}ms", stepArgs.getSleep());
+			writer = new ThrottledItemWriter<>(writer, stepArgs.getSleep());
 		}
-		return step.getWriter();
+		return writer;
 	}
 
 	public String getJobName() {