Module: Enumerable

Defined in:
lib/threach.rb

Instance Method Summary collapse

Instance Method Details

#threach(threads = 0, iterator = :each, &blk) ⇒ Object 



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
 
# File 'lib/threach.rb', line 4

def threach(threads=0, iterator=:each, &blk)
  if threads == 0
    self.send(iterator) do |*args|
      blk.call *args
    end
  else
    bq = SizedQueue.new(threads * 2)
    consumers = []
    threads.times do |i|
      consumers << Thread.new(i) do |i|
        until (a = bq.pop) === :end_of_data
          blk.call(*a)
        end
      end          
    end
  
    # The producer
    count = 0
    self.send(iterator) do |*x|
      bq.push x
      count += 1
    end
    # Now end it
    threads.times do 
      bq << :end_of_data
    end
    # Do the join
    consumers.each {|t| t.join}
  end
end