diff options
Diffstat (limited to 'lib/thread.rb')
-rw-r--r-- | lib/thread.rb | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/lib/thread.rb b/lib/thread.rb index 4f294cc9a3..8f7f6cdd6a 100644 --- a/lib/thread.rb +++ b/lib/thread.rb @@ -13,6 +13,10 @@ unless defined? ThreadError end end +if $DEBUG + Thread.abort_on_exception = true +end + class Mutex def initialize @waiting = [] @@ -107,4 +111,30 @@ class Queue def length @que.length end + alias size length +end + +class SizedQueue<Queue + def initialize(max) + @max = max + @queue_wait = [] + super() + end + + def push(obj) + while @que.length >= @max + @queue_wait.push Thread.current + Thread.stop + end + super + end + + def pop(*args) + if @que.length < @max + t = @queue_wait.shift + t.run if t + end + pop = super + pop + end end |