chore: simplify batch creation (#50)

Logflare · Jan 10, 2024 · d8f3f52 · d8f3f52
1 parent ae95e46
commit d8f3f52
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/lib/pipeline.ex b/lib/pipeline.ex
@@ -17,11 +17,9 @@ defmodule Loadfest.Pipeline do
     end
 
     def handle_demand(demand, state) when demand > 0 do
-      messages =
-        for batch_size <- Loadfest.Worker.stream_batch_sizes() |> Enum.take(demand),
-            events = Loadfest.Worker.stream_batch() |> Enum.take(batch_size) do
-          %Broadway.Message{data: events, acknowledger: {__MODULE__, :ack, 3}}
-        end
+      messages = for _i <- 1..demand do
+          %Broadway.Message{data: Loadfest.Worker.make_batch(), acknowledger: {__MODULE__, :ack, 3}}
+      end
 
       {:noreply, messages, state}
     end