producer_consumer 0.1

data/LICENSE

@@ -0,0 +1,21 @@
+The MIT License
+
+Copyright (c) 2011 Kyle Kingsbury
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.markdown

@@ -0,0 +1,5 @@
+    ProducerConsumer.produce 4 do |p|
+      some_item or p.finish
+    end.consume 10 do |item|
+      puts item
+    end.run

data/lib/producer_consumer.rb

@@ -0,0 +1,115 @@
+class ProducerConsumer
+  require 'thread'
+
+  class Finished < Exception
+  end
+
+  class NoConsumer < RuntimeError
+  end
+
+  class NoProducer < RuntimeError
+  end
+
+  class Producer
+    def finish
+      raise ProducerConsumer::Finished
+    end 
+  end
+
+  def self.consume(*a, &b)
+    new.consume(*a, &b)
+  end
+
+  def self.produce(*a, &b)
+    new.produce(*a, &b)
+  end
+
+  # Oh god. It's a PATTERN.
+  def initialize
+    @queue = Queue.new 
+    @state = :free
+    @producers = []
+    @consumers = []
+  end
+
+  # Registers n producers
+  def produce(n = 1, &block)
+    @producers += (0...n).map do |i|
+      block
+    end
+
+    self
+  end
+
+  def producer(block)
+    Thread.new do
+      p = Producer.new
+      begin
+        loop do
+          @queue << block.call(p)
+        end
+      rescue Finished
+      end
+    end
+  end
+
+  # Registers n consumers
+  def consume(n = 1, &block)
+    @consumers += (0...n).map do |c|
+      block
+    end
+
+    self
+  end
+
+  def consumer(block)
+    Thread.new do
+      begin
+        loop do
+          x = @queue.pop
+
+          if x == Finished
+            raise Finished
+          end
+
+          block.call x
+        end
+      rescue Finished
+      end
+    end
+  end
+
+  # Run!
+  def run
+    # Can't run this more than once.
+    unless @state == :free
+      raise RuntimeError, "Already running"
+    end
+
+    # Validate we will actually do something.
+    raise NoConsumer if @consumers.empty?
+    raise NoProducer if @producers.empty?
+
+    # Start producers and consumers
+    @state = :starting
+    consumers = @consumers.map do |block|
+      consumer block
+    end
+    producers = @producers.map do |block|
+      producer block
+    end
+
+    # Wait for producers to finish
+    @state = :waiting_for_producers
+    producers.each { |t| t.join }
+
+    # Signal consumers to finish...
+    @state = :waiting_for_consumers
+    consumers.size.times do
+      @queue << Finished
+    end
+    consumers.each { |t| t.join }
+
+    @state = :free
+  end
+end

data/lib/producer_consumer/version.rb

@@ -0,0 +1,3 @@
+class ProducerConsumer
+  VERSION = '0.1'
+end

metadata

@@ -0,0 +1,71 @@
+--- !ruby/object:Gem::Specification 
+name: producer_consumer
+version: !ruby/object:Gem::Version 
+  hash: 9
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  version: "0.1"
+platform: ruby
+authors: 
+- Kyle Kingsbury
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-06-28 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: 
+email: aphyr@aphyr.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/producer_consumer.rb
+- lib/producer_consumer/version.rb
+- LICENSE
+- README.markdown
+has_rdoc: true
+homepage: https://github.com/aphyr/producer_consumer
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 57
+      segments: 
+      - 1
+      - 8
+      - 7
+      version: 1.8.7
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: producer_consumer
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: A tiny threaded queue-backed producer-consumer.
+test_files: []
+