marvin_load_test

defmodule Marvin.Reporter do
  use Agent, restart: :permanent, shutdown: 10_000

  @spec start_link(any) :: {:error, any} | {:ok, pid}
  def start_link(_) do
    init_state = %{
      n_failed_requests: 0,
      n_successful_requests: 0
    }

    Agent.start_link(fn -> init_state end, name: __MODULE__)
  end

  @spec log_result([{:failed, non_neg_integer()} | {:successful, non_neg_integer()}, ...]) :: :ok
  def log_result(successful: successful, failed: failed) do
    increase_count(:n_successful_requests, successful)
    increase_count(:n_failed_requests, failed)
  end

  @spec print_result(non_neg_integer(), non_neg_integer()) :: :ok
  def print_result(start, stop) do
    ensure_logs_all_collected()
    n_successful_requests = get_state(:n_successful_requests)
    n_failed_requests = get_state(:n_failed_requests)

    duration = ((stop - start) / 1_000_000) |> Float.round(2)

    rps = (n_successful_requests / duration) |> round
    rpm = (rps * 60 / 1000) |> Float.round(1)

    IO.puts("==========================================================")
    IO.puts("# Successful requests: #{n_successful_requests}")
    IO.puts("# Failed requests: #{n_failed_requests}")
    IO.puts("# Duration: #{duration} seconds")
    IO.puts("# Throughput: #{rps} rps, #{rpm}K rpm")
    IO.puts("==========================================================")
  end

  defp get_state(key) do
    Agent.get(__MODULE__, fn state -> state[key] end)
  end

  defp update_state(key, value) do
    Agent.update(__MODULE__, fn state -> Map.put(state, key, value) end)
  end

  defp increase_count(key, count) do
    old_count = get_state(key)
    update_state(key, old_count + count)
  end

  defp ensure_logs_all_collected do
    {:message_queue_len, msg_count} = Process.info(self(), :message_queue_len)

    if msg_count > 0 do
      :timer.sleep(100)
      ensure_logs_all_collected()
    end
  end
end