2020-05-06 22:14:24 +02:00
|
|
|
defmodule Pleroma.Gun.ConnectionPool.WorkerSupervisor do
|
2020-07-15 14:24:47 +02:00
|
|
|
@moduledoc "Supervisor for pool workers. Does not do anything except enforce max connection limit"
|
2020-05-06 22:14:24 +02:00
|
|
|
|
|
|
|
use DynamicSupervisor
|
|
|
|
|
|
|
|
def start_link(opts) do
|
|
|
|
DynamicSupervisor.start_link(__MODULE__, opts, name: __MODULE__)
|
|
|
|
end
|
|
|
|
|
|
|
|
def init(_opts) do
|
|
|
|
DynamicSupervisor.init(
|
|
|
|
strategy: :one_for_one,
|
|
|
|
max_children: Pleroma.Config.get([:connections_pool, :max_connections])
|
|
|
|
)
|
|
|
|
end
|
|
|
|
|
2020-05-07 15:11:48 +02:00
|
|
|
def start_worker(opts, retry \\ false) do
|
2020-05-06 22:14:24 +02:00
|
|
|
case DynamicSupervisor.start_child(__MODULE__, {Pleroma.Gun.ConnectionPool.Worker, opts}) do
|
|
|
|
{:error, :max_children} ->
|
2020-05-07 15:11:48 +02:00
|
|
|
if retry or free_pool() == :error do
|
|
|
|
:telemetry.execute([:pleroma, :connection_pool, :provision_failure], %{opts: opts})
|
|
|
|
{:error, :pool_full}
|
|
|
|
else
|
|
|
|
start_worker(opts, true)
|
2020-05-06 22:14:24 +02:00
|
|
|
end
|
|
|
|
|
|
|
|
res ->
|
|
|
|
res
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
defp free_pool do
|
2020-07-08 13:22:42 +02:00
|
|
|
wait_for_reclaimer_finish(Pleroma.Gun.ConnectionPool.Reclaimer.start_monitor())
|
2020-05-06 22:14:24 +02:00
|
|
|
end
|
|
|
|
|
2020-07-08 13:22:42 +02:00
|
|
|
defp wait_for_reclaimer_finish({pid, mon}) do
|
2020-05-06 22:14:24 +02:00
|
|
|
receive do
|
2020-07-08 13:22:42 +02:00
|
|
|
{:DOWN, ^mon, :process, ^pid, :no_unused_conns} ->
|
2020-05-06 22:14:24 +02:00
|
|
|
:error
|
|
|
|
|
2020-07-08 13:22:42 +02:00
|
|
|
{:DOWN, ^mon, :process, ^pid, :normal} ->
|
2020-05-06 22:14:24 +02:00
|
|
|
:ok
|
|
|
|
end
|
|
|
|
end
|
|
|
|
end
|