feat: randomized batch sizes

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