mlanett-hive 0.3.0

data/spec/support/jobs.rb

@@ -0,0 +1,68 @@
+# -*- encoding: utf-8 -*-
+
+require "support/redis"
+
+class QuitJob
+  def initialize( options = {} )
+  end
+  def call( context = {} )
+    context[:worker].quit!
+  end
+end
+
+class TrueJob
+  def initialize( options = {} )
+  end
+  def call
+    true
+  end
+end
+
+class QuitJobWithSet
+  def initialize( options = {} )
+  end
+  include RedisClient
+  def call( context = {} )
+    redis.set("QuitJobWithSet",Process.pid)
+    context[:worker].quit!
+  end
+end
+
+class ForeverJobWithSet
+  def initialize( options = {} )
+  end
+  include RedisClient
+  def call( context = {} )
+    redis.set("ForeverJobWithSet",Process.pid)
+    false
+  end
+end
+
+class ForeverUntilQuitJob
+  def initialize( options = {} )
+  end
+  include RedisClient
+  def call( context = {} )
+    if redis.get("ForeverUntilQuitJob") then
+      context[:worker].quit!
+    else
+      false
+    end
+  end
+end
+
+class ListenerJob
+  include RedisClient
+
+  def initialize( context = {} )
+    storage = Hive::Redis::Storage.new(redis)
+    @mq = Hive::Messager.new( storage, my_address: context[:worker].key )
+    @mq.expect("Quit")   { |message| context[:worker].quit! }
+    @mq.expect("Exit!")  { |message| Kernel.exit! }
+    @mq.expect("State?") { |message| @mq.reply "State: #{context[:worker].state}", to: message }
+  end
+
+  def call( context = {} )
+    @mq.receive
+  end
+end

data/spec/support/redis.rb

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+
+require "redis"
+
+module RedisClient
+
+  TEST_REDIS = { url: "redis://127.0.0.1:6379/1" }
+
+  def redis
+    @redis ||= ::Redis.connect(TEST_REDIS)
+  end
+
+  def with_clean_redis(&block)
+    redis.client.disconnect # auto connect after fork
+    redis.flushall
+    yield
+  ensure
+    redis.flushall
+    redis.quit              # quit (close) connection, not server
+  end
+
+end # RedisClient

data/spec/support/timing.rb

@@ -0,0 +1,32 @@
+# -*- encoding: utf-8 -*-
+
+module Timing
+
+  def time(&block)
+    _time(&block)
+    elapsed
+  end
+
+  protected
+
+  def start
+    @start = Time.now.to_f
+  end
+
+  def finish
+    @finish = Time.now.to_f
+  end
+
+  def elapsed
+    @finish - @start
+  end
+
+  def _time(&block)
+    # elapsed time should be known whether or not it raises an error
+    start
+    yield
+  ensure
+    finish
+  end
+
+end # Timing

data/spec/utilities/observer_base_spec.rb

@@ -0,0 +1,50 @@
+# -*- encoding: utf-8 -*-
+
+require "helper"
+require "hive"
+
+class TestObserver < Hive::Utilities::ObserverBase
+  attr :alpha
+  attr :beta
+  def initialize( alpha = 1, beta = 2 )
+    @alpha = alpha
+    @beta = beta
+  end
+end
+
+describe Hive::Utilities::ObserverBase do
+
+  it "can instaniate from a class name" do
+    o = Hive::Utilities::ObserverBase.resolve TestObserver
+    o.should be_instance_of TestObserver
+    o.alpha.should eq(1)
+    o.beta.should eq(2)
+  end
+
+  it "can instaniate from a string" do
+    o = Hive::Utilities::ObserverBase.resolve "TestObserver"
+    o.should be_instance_of TestObserver
+    o.alpha.should eq(1)
+    o.beta.should eq(2)
+  end
+
+  it "can instaniate from a symbol" do
+    o = Hive::Utilities::ObserverBase.resolve :log
+    o.should be_instance_of Hive::Utilities::LogObserver
+  end
+
+  it "can instaniate from a block" do
+    o = Hive::Utilities::ObserverBase.resolve (->() { TestObserver.new })
+    o.should be_instance_of TestObserver
+    o.alpha.should eq(1)
+    o.beta.should eq(2)
+  end
+
+  it "can instaniate from an array" do
+    o = Hive::Utilities::ObserverBase.resolve [ TestObserver, 2, 4 ]
+    o.should be_instance_of TestObserver
+    o.alpha.should eq(2)
+    o.beta.should eq(4)
+  end
+  
+end

data/spec/utilities/process_spec.rb

@@ -0,0 +1,17 @@
+# -*- encoding: utf-8 -*-
+
+require "helper"
+require "hive"
+
+describe Hive::Utilities::Process do
+  it "does not fail like Process.wait2" do
+    system("false")
+    pid = $?.pid
+    expect { ::Process.wait2( pid ) }.to raise_exception
+  end
+  it "can handle dead processes" do
+    system("false")
+    pid = $?.pid
+    expect { Hive::Utilities::Process.wait_and_terminate(pid) }.to_not raise_exception
+  end
+end

data/spec/worker_spec.rb

@@ -0,0 +1,121 @@
+# -*- encoding: utf-8 -*-
+
+require "helper"
+
+describe Hive::Worker do
+
+  it "should run once" do
+    count  = 0
+    worker = nil
+    job    = ->(context={}) { count += 1; worker.quit! }
+    worker = Hive::Worker.new( job )
+    worker.run
+    count.should eq 1
+  end
+
+  it "should run with a classname" do
+    worker = Hive::Worker.new("QuitJob")
+    expect { worker.run }.should_not raise_error
+  end
+
+  it "should run with a class" do
+    worker = Hive::Worker.new(QuitJob)
+    expect { worker.run }.should_not raise_error
+  end
+
+  it "should run with a lambda" do
+    job    = ->(context) { context[:worker].quit! }
+    worker = Hive::Worker.new( job )
+    expect { worker.run }.should_not raise_error
+  end
+
+  it "should pass a context with a worker" do
+    ok     = false
+    worker = nil
+    job    = ->(context) { worker.should eq context[:worker]; worker.quit! }
+    worker = Hive::Worker.new( job )
+    worker.run
+  end
+
+  it "should use observers" do
+    obsrvr  = Hive::Utilities::ObserverBase.new
+    obsrvr.should_receive(:notify).with(anything,:worker_started).ordered
+    obsrvr.should_receive(:notify).with(anything,:worker_heartbeat).ordered
+    obsrvr.should_receive(:notify).with(anything,:worker_stopped).ordered
+
+    job     = ->(context) { context[:worker].quit! }
+    policy  = Hive::Policy.resolve({ observers: [ obsrvr ] })
+    worker  = Hive::Worker.new job, policy: policy
+
+    worker.run
+  end
+
+  it "should exit when the policy says to run out (of jobs)" do
+    count  = 0
+    job    = ->(context) { count += 1; true }
+    policy = Hive::Policy.resolve({ worker_max_jobs: 5 })
+    worker = Hive::Worker.new job, policy: policy
+    worker.run
+    count.should be <= 5
+  end
+
+  it "should execute after_fork blocks"
+
+  describe "when testing lifetime", time: true do
+    it "should exit when the policy says to run out (of time)" do
+      overhead = 1
+      worker_max_lifetime = 2
+      count    = 0
+      job      = ->(context) { count += 1; true }
+      policy   = Hive::Policy.resolve worker_max_lifetime: worker_max_lifetime, worker_max_jobs: 1e9
+      worker   = Hive::Worker.new job, policy: policy
+      time { worker.run }
+      elapsed.should be <= worker_max_lifetime + overhead
+    end
+  end
+
+  describe "when spawning a process", redis: true do
+
+    before do
+      @policy = Hive::Policy.resolve worker_max_lifetime: 4, worker_max_jobs: 100, storage: :redis
+    end
+
+    it "should spawn a new process" do
+      Hive::Worker.spawn( QuitJobWithSet )
+      wait_until { redis.get("QuitJobWithSet").to_i > 0 }
+      redis.get("QuitJobWithSet").to_i.should be > 0
+    end
+
+    it "should respond to TERM" do
+      Hive::Worker.spawn( ForeverJobWithSet )
+
+      wait_until { redis.get("ForeverJobWithSet").to_i != 0 }
+      pid = redis.get("ForeverJobWithSet").to_i
+      Hive::Utilities::Process.alive?(pid).should be_true
+
+      Process.kill( "TERM", pid )
+      wait_until { ! Hive::Utilities::Process.alive?(pid) }
+      Hive::Utilities::Process.alive?(pid).should be_false
+    end
+
+    it "uses the registry" do
+      job       = ForeverUntilQuitJob
+      registry  = Hive::Registry.new( job.to_s, @policy.storage )
+      registry.workers.size.should eq(0)
+
+      Hive::Worker.spawn ForeverUntilQuitJob, registry: registry, policy: @policy
+
+      wait_until { registry.workers.size > 0 }
+      registry.workers.size.should eq(1)
+
+      key = registry.workers.first
+      key.name.should eq(job.to_s)
+
+      redis.set("ForeverUntilQuitJob",true)
+      wait_until { registry.workers.size == 0 }
+      registry.workers.size.should eq(0)
+    end
+
+  end
+
+end

data/unused/times.rb

@@ -0,0 +1,45 @@
+start, time
+stop,  time
+start, time
+stop,  time
+start, time
+stop,  time
+
+def generate
+  now = Time.now.to_i
+  10.times do
+    puts -now
+    now += rand(7)
+    puts now
+    now += rand(3)
+  end
+end
+
+times = [-1318798764,1318798766,-1318798768,1318798768,-1318798768,1318798772,-1318798773,1318798776,-1318798777,1318798781,-1318798781,1318798781,-1318798783,1318798784,-1318798786,1318798792,-1318798793,1318798796,-1318798796,1318798800]
+
+def analyze( start, times )
+  whats = []
+  whens = []
+  last  = nil
+  ( times + [Time.now.to_i] ).each do |time|
+    if time < 0 then
+      if last && whats[last] == :start then
+        whats << :stop
+        whens << -time
+      end
+      whats << :start
+      whens << -time
+    else
+      if last && whats[last] == :stop then
+        whats << :start
+        whens << time
+      end
+      whats << :stop
+      whens << time
+    end
+    last = (last || -1) + 1
+  end
+  [ whats, whens ]
+end
+
+analyze(times)

metadata

@@ -0,0 +1,148 @@
+--- !ruby/object:Gem::Specification
+name: mlanett-hive
+version: !ruby/object:Gem::Version
+  version: 0.3.0
+  prerelease: 
+platform: ruby
+authors:
+- Mark Lanett
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-02-07 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: redis
+  requirement: &70198172375900 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70198172375900
+- !ruby/object:Gem::Dependency
+  name: redis-namespace
+  requirement: &70198172375480 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70198172375480
+description: 
+email:
+- mark.lanett@gmail.com
+executables:
+- hive
+extensions: []
+extra_rdoc_files: []
+files:
+- .autotest
+- .gitignore
+- .rspec
+- Gemfile
+- Guardfile
+- README
+- Rakefile
+- bin/hive
+- demo/demo
+- demo/demo.rb
+- demo/demo3
+- demo/job1.rb
+- demo/job2.rb
+- demo/job3.rb
+- demo/populate.rb
+- hive.gemspec
+- lib/hive.rb
+- lib/hive/checker.rb
+- lib/hive/configuration.rb
+- lib/hive/idler.rb
+- lib/hive/key.rb
+- lib/hive/lifecycle_observer.rb
+- lib/hive/log.rb
+- lib/hive/messager.rb
+- lib/hive/mocks/storage.rb
+- lib/hive/monitor.rb
+- lib/hive/policy.rb
+- lib/hive/pool.rb
+- lib/hive/redis/storage.rb
+- lib/hive/registry.rb
+- lib/hive/squiggly.rb
+- lib/hive/trace.rb
+- lib/hive/utilities/airbrake_observer.rb
+- lib/hive/utilities/hoptoad_observer.rb
+- lib/hive/utilities/log_observer.rb
+- lib/hive/utilities/observeable.rb
+- lib/hive/utilities/observer_base.rb
+- lib/hive/utilities/process.rb
+- lib/hive/utilities/resolver.rb
+- lib/hive/utilities/signal_hook.rb
+- lib/hive/utilities/storage_base.rb
+- lib/hive/version.rb
+- lib/hive/worker.rb
+- spec/checker_spec.rb
+- spec/configuration_spec.rb
+- spec/helper.rb
+- spec/idler_spec.rb
+- spec/key_spec.rb
+- spec/messager_spec.rb
+- spec/mocks/storage_spec.rb
+- spec/monitor_spec.rb
+- spec/policy_spec.rb
+- spec/pool_spec.rb
+- spec/redis/storage_spec.rb
+- spec/registry_spec.rb
+- spec/support/jobs.rb
+- spec/support/redis.rb
+- spec/support/timing.rb
+- spec/utilities/observer_base_spec.rb
+- spec/utilities/process_spec.rb
+- spec/worker_spec.rb
+- unused/times.rb
+homepage: ''
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.15
+signing_key: 
+specification_version: 3
+summary: Manage a collection of worker processes
+test_files:
+- spec/checker_spec.rb
+- spec/configuration_spec.rb
+- spec/helper.rb
+- spec/idler_spec.rb
+- spec/key_spec.rb
+- spec/messager_spec.rb
+- spec/mocks/storage_spec.rb
+- spec/monitor_spec.rb
+- spec/policy_spec.rb
+- spec/pool_spec.rb
+- spec/redis/storage_spec.rb
+- spec/registry_spec.rb
+- spec/support/jobs.rb
+- spec/support/redis.rb
+- spec/support/timing.rb
+- spec/utilities/observer_base_spec.rb
+- spec/utilities/process_spec.rb
+- spec/worker_spec.rb