AkkomaGang/concurrent-limiter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge branch 'fix/trap-exits' into 'master'

Browse source 
Fix counters not decrementing when the process is exited

See merge request pleroma/elixir-libraries/concurrent_limiter!1
This commit is contained in:
href 2020-09-05 07:30:49 +00:00
commit 6f45866724
1 changed files with 29 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
lib/concurrent_limiter.ex
View file

@ -101,25 +101,50 @@ defmodule ConcurrentLimiter do
cond do
counter <= max_running ->
:telemetry.execute([:concurrent_limiter, :execution], %{counter: counter}, %{limiter: name})
:telemetry.execute([:concurrent_limiter, :execution], %{counter: counter}, %{
limiter: name
})
Process.flag(:trap_exit, true)
try do
fun.()
after
dec(ref, name)
Process.flag(:trap_exit, false)
receive do
{:EXIT, _, reason} ->
Process.exit(self(), reason)
after
0 -> :noop
end
end
counter > max ->
:telemetry.execute([:concurrent_limiter, :overload], %{counter: counter}, %{limiter: name, scope: "max"})
:telemetry.execute([:concurrent_limiter, :overload], %{counter: counter}, %{
limiter: name,
scope: "max"
})
dec(ref, name)
{:error, :overload}
retries + 1 > max_retries ->
:telemetry.execute([:concurrent_limiter, :max_retries], %{counter: counter}, %{limiter: name, retries: retries + 1})
:telemetry.execute([:concurrent_limiter, :max_retries], %{counter: counter}, %{
limiter: name,
retries: retries + 1
})
dec(ref, name)
{:error, :overload}
counter > max_running ->
:telemetry.execute([:concurrent_limiter, :wait], %{counter: counter}, %{limiter: name, retries: retries + 1})
:telemetry.execute([:concurrent_limiter, :wait], %{counter: counter}, %{
limiter: name,
retries: retries + 1
})
wait(ref, name, fun, wait, opts, retries + 1)
end
end