)
end
- def start_worker(opts) do
+ def start_worker(opts, retry \\ false) do
case DynamicSupervisor.start_child(__MODULE__, {Pleroma.Gun.ConnectionPool.Worker, opts}) do
{:error, :max_children} ->
- case free_pool() do
- :ok -> start_worker(opts)
- :error -> {:error, :pool_full}
+ if retry or free_pool() == :error do
+ :telemetry.execute([:pleroma, :connection_pool, :provision_failure], %{opts: opts})
+ {:error, :pool_full}
+ else
+ start_worker(opts, true)
end
res ->
end
end
- @registry Pleroma.Gun.ConnectionPool
- @enforcer_key "enforcer"
defp free_pool do
- case Registry.lookup(@registry, @enforcer_key) do
- [] ->
- pid =
- spawn(fn ->
- {:ok, _pid} = Registry.register(@registry, @enforcer_key, nil)
- max_connections = Pleroma.Config.get([:connections_pool, :max_connections])
-
- reclaim_max =
- [:connections_pool, :reclaim_multiplier]
- |> Pleroma.Config.get()
- |> Kernel.*(max_connections)
- |> round
- |> max(1)
-
- unused_conns =
- Registry.select(
- @registry,
- [
- {{:_, :"$1", {:_, :"$2", :"$3", :"$4"}}, [{:==, :"$2", []}],
- [{{:"$1", :"$3", :"$4"}}]}
- ]
- )
-
- case unused_conns do
- [] ->
- exit(:no_unused_conns)
-
- unused_conns ->
- unused_conns
- |> Enum.sort(fn {_pid1, crf1, last_reference1}, {_pid2, crf2, last_reference2} ->
- crf1 <= crf2 and last_reference1 <= last_reference2
- end)
- |> Enum.take(reclaim_max)
- |> Enum.each(fn {pid, _, _} ->
- DynamicSupervisor.terminate_child(__MODULE__, pid)
- end)
- end
- end)
-
- wait_for_enforcer_finish(pid)
-
- [{pid, _}] ->
- wait_for_enforcer_finish(pid)
- end
+ wait_for_reclaimer_finish(Pleroma.Gun.ConnectionPool.Reclaimer.start_monitor())
end
- defp wait_for_enforcer_finish(pid) do
- ref = Process.monitor(pid)
-
+ defp wait_for_reclaimer_finish({pid, mon}) do
receive do
- {:DOWN, ^ref, :process, ^pid, :no_unused_conns} ->
+ {:DOWN, ^mon, :process, ^pid, :no_unused_conns} ->
:error
- {:DOWN, ^ref, :process, ^pid, :normal} ->
+ {:DOWN, ^mon, :process, ^pid, :normal} ->
:ok
end
end