Do not untrap exits if they were already trapped before callingfix/do-not-untrap-trapped-processes

1 files changed, 10 insertions, 7 deletions

diff --git a/lib/concurrent_limiter.ex b/lib/concurrent_limiter.ex
index 4a367ea..f4d6fbc 100644
--- a/lib/concurrent_limiter.ex
+++ b/lib/concurrent_limiter.ex
@@ -105,19 +105,22 @@ defmodule ConcurrentLimiter do
           limiter: name
         })
 
-        Process.flag(:trap_exit, true)
+        old_value = 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
+          unless old_value do
+            Process.flag(:trap_exit, false)
+
+            receive do
+              {:EXIT, _, reason} ->
+                Process.exit(self(), reason)
+            after
+              0 -> :noop
+            end
           end
         end