class QueueWithTimeout
Borrowed from spin.atomicobject.com/2017/06/28/queue-pop-with-timeout-fixed/
Public Class Methods
new()
click to toggle source
# File lib/signalfx/signalflow/queue.rb, line 7 def initialize @mutex = Mutex.new @queue = [] @received = ConditionVariable.new end
Public Instance Methods
<<(x)
click to toggle source
# File lib/signalfx/signalflow/queue.rb, line 13 def <<(x) @mutex.synchronize do @queue << x @received.signal end end
pop(non_block = false)
click to toggle source
# File lib/signalfx/signalflow/queue.rb, line 20 def pop(non_block = false) pop_with_timeout(non_block ? 0 : nil) end
pop_with_timeout(timeout = nil)
click to toggle source
# File lib/signalfx/signalflow/queue.rb, line 24 def pop_with_timeout(timeout = nil) @mutex.synchronize do if timeout.nil? # wait indefinitely until there is an element in the queue while @queue.empty? @received.wait(@mutex) end elsif @queue.empty? && timeout != 0 # wait for element or timeout timeout_time = timeout + Time.now.to_f while @queue.empty? && (remaining_time = timeout_time - Time.now.to_f) > 0 @received.wait(@mutex, remaining_time) end end #if we're still empty after the timeout, raise exception raise ThreadError, "queue empty" if @queue.empty? @queue.shift end end