feat: randomize batch size

Logflare · Jan 25, 2024 · 6deede4 · 6deede4
1 parent 1e4023a
commit 6deede4
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/lib/pipeline.ex b/lib/pipeline.ex
@@ -17,9 +17,15 @@ defmodule Loadfest.Pipeline do
     end
 
     def handle_demand(demand, state) when demand > 0 do
-      messages = for _i <- 1..demand do
-          %Broadway.Message{data: Loadfest.Worker.make_batch(250), acknowledger: {__MODULE__, :ack, 3}}
-      end
+      num = Enum.random([1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 50, 50, 50, 250])
+
+      messages =
+        for _i <- 1..demand do
+          %Broadway.Message{
+            data: Loadfest.Worker.make_batch(num),
+            acknowledger: {__MODULE__, :ack, 3}
+          }
+        end
 
       {:noreply, messages, state}
     end