Browse Source

gun ConnectionPool: replace casts with calls

The slowdown from this is most likely immesurable, however it eliminates
possible false positives when tracking dead clients.
frontends/logic-flow
rinpatch 3 years ago
parent
commit
4ce6179dc7
2 changed files with 21 additions and 23 deletions
  1. +2
    -2
      lib/pleroma/gun/connection_pool.ex
  2. +19
    -21
      lib/pleroma/gun/connection_pool/worker.ex

+ 2
- 2
lib/pleroma/gun/connection_pool.ex View File

@@ -19,7 +19,7 @@ defmodule Pleroma.Gun.ConnectionPool do
get_gun_pid_from_worker(worker_pid, true) get_gun_pid_from_worker(worker_pid, true)


[{worker_pid, {gun_pid, _used_by, _crf, _last_reference}}] -> [{worker_pid, {gun_pid, _used_by, _crf, _last_reference}}] ->
GenServer.cast(worker_pid, {:add_client, self(), false})
GenServer.call(worker_pid, :add_client)
{:ok, gun_pid} {:ok, gun_pid}


[] -> [] ->
@@ -70,7 +70,7 @@ defmodule Pleroma.Gun.ConnectionPool do


case query_result do case query_result do
[worker_pid] -> [worker_pid] ->
GenServer.cast(worker_pid, {:remove_client, self()})
GenServer.call(worker_pid, :remove_client)


[] -> [] ->
:ok :ok


+ 19
- 21
lib/pleroma/gun/connection_pool/worker.ex View File

@@ -36,7 +36,16 @@ defmodule Pleroma.Gun.ConnectionPool.Worker do
end end


@impl true @impl true
def handle_cast({:add_client, client_pid, send_pid_back}, %{key: key} = state) do
def handle_cast({:add_client, client_pid, send}, state) do
case handle_call(:add_client, {client_pid, nil}, state) do
{:reply, conn_pid, state, :hibernate} ->
if send, do: send(client_pid, {:conn_pid, conn_pid})
{:noreply, state, :hibernate}
end
end

@impl true
def handle_call(:add_client, {client_pid, _}, %{key: key} = state) do
time = :erlang.monotonic_time(:millisecond) time = :erlang.monotonic_time(:millisecond)


{{conn_pid, _, _, _}, _} = {{conn_pid, _, _, _}, _} =
@@ -44,8 +53,6 @@ defmodule Pleroma.Gun.ConnectionPool.Worker do
{conn_pid, [client_pid | used_by], crf(time - last_reference, crf), time} {conn_pid, [client_pid | used_by], crf(time - last_reference, crf), time}
end) end)


if send_pid_back, do: send(client_pid, {:conn_pid, conn_pid})

state = state =
if state.timer != nil do if state.timer != nil do
Process.cancel_timer(state[:timer]) Process.cancel_timer(state[:timer])
@@ -57,11 +64,11 @@ defmodule Pleroma.Gun.ConnectionPool.Worker do
ref = Process.monitor(client_pid) ref = Process.monitor(client_pid)


state = put_in(state.client_monitors[client_pid], ref) state = put_in(state.client_monitors[client_pid], ref)
{:noreply, state, :hibernate}
{:reply, conn_pid, state, :hibernate}
end end


@impl true @impl true
def handle_cast({:remove_client, client_pid}, %{key: key} = state) do
def handle_call(:remove_client, {client_pid, _}, %{key: key} = state) do
{{_conn_pid, used_by, _crf, _last_reference}, _} = {{_conn_pid, used_by, _crf, _last_reference}, _} =
Registry.update_value(@registry, key, fn {conn_pid, used_by, crf, last_reference} -> Registry.update_value(@registry, key, fn {conn_pid, used_by, crf, last_reference} ->
{conn_pid, List.delete(used_by, client_pid), crf, last_reference} {conn_pid, List.delete(used_by, client_pid), crf, last_reference}
@@ -78,7 +85,7 @@ defmodule Pleroma.Gun.ConnectionPool.Worker do
nil nil
end end


{:noreply, %{state | timer: timer}, :hibernate}
{:reply, :ok, %{state | timer: timer}, :hibernate}
end end


@impl true @impl true
@@ -102,22 +109,13 @@ defmodule Pleroma.Gun.ConnectionPool.Worker do


@impl true @impl true
def handle_info({:DOWN, _ref, :process, pid, reason}, state) do def handle_info({:DOWN, _ref, :process, pid, reason}, state) do
# Sometimes the client is dead before we demonitor it in :remove_client, so the message
# arrives anyway

case state.client_monitors[pid] do
nil ->
{:noreply, state, :hibernate}
:telemetry.execute(
[:pleroma, :connection_pool, :client_death],
%{client_pid: pid, reason: reason},
%{key: state.key}
)


_ref ->
:telemetry.execute(
[:pleroma, :connection_pool, :client_death],
%{client_pid: pid, reason: reason},
%{key: state.key}
)

handle_cast({:remove_client, pid}, state)
end
handle_cast({:remove_client, pid, false}, state)
end end


# LRFU policy: https://citeseerx.ist.psu.edu/viewdoc/summary?doi=10.1.1.55.1478 # LRFU policy: https://citeseerx.ist.psu.edu/viewdoc/summary?doi=10.1.1.55.1478


Loading…
Cancel
Save