async-limiter 1.5.2 → 1.5.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 180ea4ebb784a29df337fe94e8190358482658fd07547ccaeaddc0974b271696
-  data.tar.gz: b02a1c5978f4b14e103852bbaa3685d8e1258b6d0190d1bb1e3c157a18b786e9
+  metadata.gz: dc95096df427d6b99502f8eb6f17f3cc0ec70598e093fd1d8aae5db50324d668
+  data.tar.gz: 21dfb9413922d3b20f1ee3f4ac9dbfd1ecdc5c2e08d12e9760c4b347cd941c0a
 SHA512:
-  metadata.gz: e0cd30fb437861e2077f1d15bd39a2bd9314ac64a7d48d2d303302a2c13c875cae02f042ad0d93531bf125475b62c1cc1b2e532279b044623f0b7b8765732bbb
-  data.tar.gz: abd67afa695c11d045f176cf5371046238050f951626cc5bc2f2b10a484ff567cd63e26ceb8b9896fb052b55682081d9bc421e9d6e489cdbc34c68fee38bc922
+  metadata.gz: b3c074890ffa3befca024121f8988222539991ed43a99b99ccdcf2855b7e8219151ed118a2d549b490ad4992a2801c2af7e8fb43f315978176e1dab8db9657ec
+  data.tar.gz: 44ddac8caf55fa453604e57565e66c8bccd708cd710c76f65ad13f2408daa5442e3596dde2cd45e7a5bd33ed90e034a226c30cc98f85f57196e34059f7db4a7a

data/lib/async/limiter/version.rb

@@ -1,5 +1,5 @@
 module Async
   module Limiter
-    VERSION = "1.5.2"
+    VERSION = "1.5.3"
   end
 end

data/lib/async/limiter/window.rb

@@ -1,4 +1,5 @@
 require "async/clock"
+require "async/notification"
 require "async/task"
 require_relative "constants"
 
@@ -26,6 +27,8 @@ module Async
 
         @waiting = queue
         @scheduler = nil
+        @yield_wait = false
+        @yield_notification = Notification.new
 
         @window_frame_start_time = NULL_TIME
         @window_start_time = NULL_TIME
@@ -158,7 +161,14 @@ module Async
         # slipping in operations before other waiting fibers get resumed.
         if blocking? || @waiting.any?
           @waiting.push(fiber, *queue_args) # queue_args used for custom queues
+          @yield_wait = true
           schedule if schedule?
+
+          # Non-blocking signal, prevents race condition where scheduler would
+          # start resuming waiting fibers before below 'Task.yield' was reached.
+          @yield_notification.signal
+          @yield_wait = false # we're out of the woods
+
           loop do
             Task.yield # run this line at least once
             break unless blocking?
@@ -186,6 +196,11 @@ module Async
           while @waiting.any? && !limit_blocking?
             delay = [next_acquire_time - Clock.now, 0].max
             task.sleep(delay) if delay.positive?
+
+            # Waits for the task that started the scheduler to yield.
+            # See #wait for more details.
+            @yield_wait && @yield_notification&.wait
+
             resume_waiting
           end
         ensure

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: async-limiter
 version: !ruby/object:Gem::Version
-  version: 1.5.2
+  version: 1.5.3
 platform: ruby
 authors:
 - Bruno Sutic