ZioKafkaConsumerbenchmark.throughput: increase count per batch

This is more consistent with the manual kafka client benchmark equivalent
zio · Nov 30, 2024 · 7df5e80 · 7df5e80
1 parent bdde6e3
commit 7df5e80
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/zio-kafka-bench/src/main/scala/zio/kafka/bench/ZioKafkaConsumerBenchmark.scala b/zio-kafka-bench/src/main/scala/zio/kafka/bench/ZioKafkaConsumerBenchmark.scala
@@ -42,8 +42,11 @@ class ZioKafkaConsumerBenchmark extends ConsumerZioBenchmark[Kafka with Producer
       counter <- Ref.make(0)
       _ <- Consumer
              .plainStream(Subscription.topics(topic1), Serde.byteArray, Serde.byteArray)
-             .tap { _ =>
-               counter.updateAndGet(_ + 1).flatMap(count => Consumer.stopConsumption.when(count == recordCount))
+             .chunks
+             .tap { batch =>
+               counter
+                 .updateAndGet(_ + batch.size)
+                 .flatMap(count => Consumer.stopConsumption.when(count == recordCount))
              }
              .runDrain
              .provideSome[Kafka](env)