marvin_load_test

defmodule Marvin.Runner do
  use DynamicSupervisor, restart: :transient
  alias Marvin.Worker
  alias Marvin.Config
  alias Marvin.Reporter
  alias Marvin.HttpClient

  @spec start_link(any) :: :ignore | {:error, any} | {:ok, pid}
  def start_link(init_args) do
    DynamicSupervisor.start_link(__MODULE__, init_args, name: __MODULE__)
  end

  @spec setup_workers([Config.scenario()]) :: [:ok]
  def setup_workers(scenarios) do
    scenarios
    |> build_worker_specs()
    |> Task.async_stream(&DynamicSupervisor.start_child(__MODULE__, &1))
    |> Enum.to_list()
  end

  defp build_worker_specs(scenarios) do
    Enum.flat_map(scenarios, fn scenario ->
      Enum.map(1..scenario.concurrency, &worker_spec(&1, scenario))
    end)
  end

  defp worker_spec(id, scenario) do
    args = [
      id: "#{scenario.name}_#{id}" |> String.to_atom(),
      reporter: Reporter,
      endpoint: scenario.endpoint,
      http_client: HttpClient
    ]

    {Marvin.Worker, args}
  end

  @spec run_workers :: :ok
  def run_workers() do
    __MODULE__
    |> DynamicSupervisor.which_children()
    |> Enum.map(fn {_id, pid, _, _} -> pid end)
    |> Task.async_stream(&Worker.run_loop/1)
    |> Stream.run()
  end

  @doc """
  When a supervisor shuts down, it terminates all children in the opposite order they are listed.
  The termination happens by sending a shutdown exit signal, via
     Process.exit(child_pid, :shutdown)
  and then awaiting for a time interval (defaults to 5000ms).
  The supervisor will abruptly terminates the child with reason :kill
  if the child process does not terminate in this interval.
  """
  @spec stop :: nil
  def stop() do
    children =
      __MODULE__
      |> DynamicSupervisor.which_children()
      |> Enum.map(fn {_id, pid, _, _} -> pid end)

    DynamicSupervisor.stop(__MODULE__, :shutdown)

    ensure_all_stopped(children)
  end

  @impl true
  def init(_) do
    DynamicSupervisor.init(strategy: :one_for_one)
  end

  defp ensure_all_stopped(children) do
    if Enum.any?(children, &Process.alive?/1) do
      :timer.sleep(100)
      ensure_all_stopped(children)
    end
  end
end