procrastinator 0.5.0 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 9ffc8c7bca7bcfddac8f0b86e428920ad3c12554
-  data.tar.gz: e68d47728ebcedda262ed1c7e3843162a64feb05
+  metadata.gz: 085c07da4d7c83dcccb46159c9bad1fc17717013
+  data.tar.gz: aaa2554edfe76056856da018705a21bef6a5345f
 SHA512:
-  metadata.gz: c74017a1c0d6dab41e5fbd7f7ef8c20fe61b8dc284b1e1ef72a856fed0123f450bca2144d5f8bdeff33b6de60f3b0c8acae677fb43f28236358244dc704bc3f0
-  data.tar.gz: 1e43eb8665474cfd59c29a12721f25b6ebe4134ead5fa89114d9631cabe227ed6f37a4e318ab6004a0493b615ed9407f402778e619eebc954e80e9bf4b6e9756
+  metadata.gz: 27cf0c1198f7c0f7e3c1208ff52f93a1c9fa7117ec2dcd8c7433f334cfeefb4b6e3dd43ed126be7058aeb60b7e78038fc3db138588e5eb267346fd7e43c7a0d6
+  data.tar.gz: cd0a15608dfa630cd70958d1040b06a63e127efca81e08f45642ee4fc427ead3a1ddf0d3f46243752f8a77d265cbcf562ecb2d01b51fed06d97ca1c9159bcfaf

data/lib/procrastinator/environment.rb

@@ -4,14 +4,7 @@ module Procrastinator
 
       DEFAULT_LOG_DIRECTORY = 'log/'
 
-      def initialize(persister:, test_mode: false)
-         raise ArgumentError.new('persister cannot be nil') if persister.nil?
-
-         [:read_tasks, :create_task, :update_task, :delete_task].each do |method|
-            raise MalformedPersisterError.new("persister must repond to ##{method}") unless persister.respond_to? method
-         end
-
-         @persister         = persister
+      def initialize(test_mode: false)
          @test_mode         = test_mode
          @queue_definitions = {}
          @queue_workers     = []
@@ -20,6 +13,12 @@ module Procrastinator
          @log_level         = Logger::INFO
       end
 
+      def persister_factory(&block)
+         @persister_factory = block
+
+         build_persister
+      end
+
       def define_queue(name, properties={})
          raise ArgumentError.new('queue name cannot be nil') if name.nil?
 
@@ -35,6 +34,7 @@ module Procrastinator
          else
             @queue_definitions.each do |name, props|
                pid = fork do
+                  build_persister
                   worker = QueueWorker.new(props.merge(name:      name,
                                                        persister: @persister,
                                                        log_dir:   @log_dir,
@@ -131,6 +131,16 @@ module Procrastinator
 
          heartbeat_thread.abort_on_exception = true
       end
+
+      def build_persister
+         @persister = @persister_factory.call
+
+         raise ArgumentError.new('persister cannot be nil') if @persister.nil?
+
+         [:read_tasks, :create_task, :update_task, :delete_task].each do |method|
+            raise MalformedPersisterError.new("persister must repond to ##{method}") unless @persister.respond_to? method
+         end
+      end
    end
 
    class MalformedPersisterError < StandardError

data/lib/procrastinator/version.rb

@@ -1,3 +1,3 @@
 module Procrastinator
-   VERSION = '0.5.0'
+   VERSION = '0.6.0'
 end

data/lib/procrastinator.rb

@@ -8,15 +8,15 @@ require 'logger'
 module Procrastinator
    @@test_mode = false
 
-   def self.setup(persister, &block)
+   def self.setup(&block)
       raise ArgumentError.new('Procrastinator.setup must be given a block') if block.nil?
 
-      env = Environment.new(persister: persister, test_mode: @@test_mode)
+      env = Environment.new(test_mode: @@test_mode)
 
       yield(env)
 
-      raise RuntimeError.new('setup block did not define any queues') if env.queue_definitions.empty?
-
+      raise RuntimeError.new('setup block must call #persister_factory on the environment') if env.persister.nil?
+      raise RuntimeError.new('setup block must call #define_queue on the environment') if env.queue_definitions.empty?
       env.spawn_workers
 
       env

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: procrastinator
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.6.0
 platform: ruby
 authors:
 - Robin Miller