summary refs log tree commit homepage
path: root/lib/rainbows/queue_pool.rb
diff options
context:
space:
mode:
authorEric Wong <normalperson@yhbt.net>2010-05-14 15:52:58 -0700
committerEric Wong <normalperson@yhbt.net>2010-05-14 18:17:08 -0700
commit01ee5327c018f64fa4b6aa767c0328c56f699170 (patch)
treee38767d95236443391ae7480bdba0804764423c2 /lib/rainbows/queue_pool.rb
parent5f854e876ba55180936ebfc82002df9bd882a4ae (diff)
It'll be useful later on for a variety of things!
Diffstat (limited to 'lib/rainbows/queue_pool.rb')
-rw-r--r--lib/rainbows/queue_pool.rb32
1 files changed, 32 insertions, 0 deletions
diff --git a/lib/rainbows/queue_pool.rb b/lib/rainbows/queue_pool.rb
new file mode 100644
index 0000000..806bbee
--- /dev/null
+++ b/lib/rainbows/queue_pool.rb
@@ -0,0 +1,32 @@
+# -*- encoding: binary -*-
+require 'thread'
+
+module Rainbows
+
+  # Thread pool class based on pulling off a single Ruby Queue.
+  # This is NOT used for the ThreadPool class, since that class does not
+  # need a userspace Queue.
+  class QueuePool < Struct.new(:queue, :threads)
+    G = Rainbows::G
+
+    def initialize(size = 20, &block)
+      q = Queue.new
+      self.threads = (1..size).map do
+        Thread.new do
+          while job = q.shift
+            block.call(job)
+          end
+        end
+      end
+      self.queue = q
+    end
+
+    def quit!
+      threads.each { |_| queue << nil }
+      threads.delete_if do |t|
+        G.tick
+        t.alive? ? t.join(0.01) : true
+      end until threads.empty?
+    end
+  end
+end