spx-resque-multi-job-forks 0.3.2

data/lib/resque-multi-job-forks.rb

@@ -0,0 +1,88 @@
+require 'resque'
+require 'resque/worker'
+
+module Resque
+  class Worker
+    attr_accessor :seconds_per_fork
+    attr_accessor :jobs_per_fork
+    attr_reader   :jobs_processed
+
+    unless method_defined?(:shutdown_without_multi_job_forks)
+      def perform_with_multi_job_forks(job = nil)
+        perform_without_multi_job_forks(job)
+        hijack_fork unless fork_hijacked?
+        @jobs_processed += 1
+      end
+      alias_method :perform_without_multi_job_forks, :perform
+      alias_method :perform, :perform_with_multi_job_forks
+
+      def shutdown_with_multi_job_forks
+        release_fork if fork_hijacked? && fork_job_limit_reached?
+        shutdown_without_multi_job_forks
+      end
+      alias_method :shutdown_without_multi_job_forks, :shutdown?
+      alias_method :shutdown?, :shutdown_with_multi_job_forks
+    end
+
+    def fork_hijacked?
+      @release_fork_limit
+    end
+
+    def hijack_fork
+      log 'hijack fork.'
+      @suppressed_fork_hooks = [Resque.after_fork, Resque.before_fork]
+      Resque.after_fork = Resque.before_fork = nil
+      @release_fork_limit = fork_job_limit
+      @jobs_processed = 0
+      @cant_fork = true
+    end
+
+    def release_fork
+      log "jobs processed by child: #{jobs_processed}"
+      run_hook :before_child_exit, self
+      Resque.after_fork, Resque.before_fork = *@suppressed_fork_hooks
+      @release_fork_limit = @jobs_processed = @cant_fork = nil
+      log 'hijack over, counter terrorists win.'
+      @shutdown = true unless $TESTING
+    end
+
+    def fork_job_limit
+      jobs_per_fork.nil? ? Time.now.to_i + seconds_per_fork : jobs_per_fork
+    end
+
+    def fork_job_limit_reached?
+      fork_job_limit_remaining <= 0 ? true : false
+    end
+
+    def fork_job_limit_remaining
+      jobs_per_fork.nil? ? @release_fork_limit - Time.now.to_i : jobs_per_fork - @jobs_processed
+    end
+
+    def seconds_per_fork
+      @seconds_per_fork ||= minutes_per_fork * 60
+    end
+
+    def minutes_per_fork
+      ENV['MINUTES_PER_FORK'].nil? ? 1 : ENV['MINUTES_PER_FORK'].to_i
+    end
+
+    def jobs_per_fork
+      @jobs_per_fork ||= ENV['JOBS_PER_FORK'].nil? ? nil : ENV['JOBS_PER_FORK'].to_i
+    end
+  end
+
+  # the `before_child_exit` hook will run in the child process
+  # right before the child process terminates
+  #
+  # Call with a block to set the hook.
+  # Call with no arguments to return the hook.
+  def self.before_child_exit(&block)
+    block ? (@before_child_exit = block) : @before_child_exit
+  end
+
+  # Set the before_child_exit proc.
+  def self.before_child_exit=(before_child_exit)
+    @before_child_exit = before_child_exit
+  end
+
+end

data/test/helper.rb

@@ -0,0 +1,60 @@
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+
+$TESTING = true
+
+require 'rubygems'
+require 'bundler'
+Bundler.setup
+Bundler.require
+require 'test/unit'
+require 'resque-multi-job-forks'
+
+# setup redis & resque.
+redis = Redis.new(:db => 1)
+Resque.redis = redis
+
+# adds simple STDOUT logging to test workers.
+# set `VERBOSE=true` when running the tests to view resques log output.
+module Resque
+  class Worker
+    def log(msg)
+      puts "*** #{msg}" unless ENV['VERBOSE'].nil?
+    end
+    alias_method :log!, :log
+  end
+end
+
+# stores a record of the job processing sequence.
+# you may wish to reset this in the test `setup` method.
+$SEQUENCE = []
+
+# test job, tracks sequence.
+class SequenceJob
+  @queue = :jobs
+  def self.perform(i)
+    $SEQUENCE << "work_#{i}".to_sym
+    sleep(2)
+  end
+end
+
+class QuickSequenceJob
+  @queue = :jobs
+  def self.perform(i)
+    $SEQUENCE << "work_#{i}".to_sym
+  end
+end
+
+
+# test hooks, tracks sequence.
+Resque.after_fork do
+  $SEQUENCE << :after_fork
+end
+
+Resque.before_fork do
+  $SEQUENCE << :before_fork
+end
+
+Resque.before_child_exit do |worker|
+  $SEQUENCE << "before_child_exit_#{worker.jobs_processed}".to_sym
+end

data/test/test_resque-multi-job-forks.rb

@@ -0,0 +1,58 @@
+require File.dirname(__FILE__) + '/helper'
+
+class TestResqueMultiJobForks < Test::Unit::TestCase
+  def setup
+    $SEQUENCE = []
+    Resque.redis.flushdb
+    @worker = Resque::Worker.new(:jobs)
+  end
+
+  def test_timeout_limit_sequence_of_events
+    # only allow enough time for 3 jobs to process.
+    @worker.seconds_per_fork = 3
+
+    Resque.enqueue(SequenceJob, 1)
+    Resque.enqueue(SequenceJob, 2)
+    Resque.enqueue(SequenceJob, 3)
+    Resque.enqueue(SequenceJob, 4)
+
+    # make sure we don't take longer then 15 seconds.
+    begin
+      Timeout::timeout(15) { @worker.work(1) }
+    rescue Timeout::Error
+    end
+
+    # test the sequence is correct.
+    assert_equal([:before_fork, :after_fork, :work_1, :work_2, :work_3,
+                  :before_child_exit_3, :before_fork, :after_fork, :work_4,
+                  :before_child_exit_1], $SEQUENCE, 'correct sequence')
+  end
+
+  # test we can also limit fork job process by a job limit.
+  def test_job_limit_sequence_of_events
+    # only allow enough time for 3 jobs to process.
+    ENV['JOBS_PER_FORK'] = '20'
+
+    # queue 40 jobs.
+    (1..40).each { |i| Resque.enqueue(QuickSequenceJob, i) }
+
+    begin
+      Timeout::timeout(3) { @worker.work(1) }
+    rescue Timeout::Error
+    end
+
+    assert_equal :before_fork, $SEQUENCE[0], 'first before_fork call.'
+    assert_equal :after_fork, $SEQUENCE[1], 'first after_fork call.'
+    assert_equal :work_20, $SEQUENCE[21], '20th chunk of work.'
+    assert_equal :before_child_exit_20, $SEQUENCE[22], 'first before_child_exit call.'
+    assert_equal :before_fork, $SEQUENCE[23], 'final before_fork call.'
+    assert_equal :after_fork, $SEQUENCE[24], 'final after_fork call.'
+    assert_equal :work_40, $SEQUENCE[44], '40th chunk of work.'
+    assert_equal :before_child_exit_20, $SEQUENCE[45], 'final before_child_exit call.'
+  end
+
+  def teardown
+    # make sure we don't clobber any other tests.
+    ENV['JOBS_PER_FORK'] = nil
+  end
+end

metadata

@@ -0,0 +1,117 @@
+--- !ruby/object:Gem::Specification
+name: spx-resque-multi-job-forks
+version: !ruby/object:Gem::Version
+  version: 0.3.2
+  prerelease: 
+platform: ruby
+authors:
+- Mick Staugaard
+- Luke Antins
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-06-28 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: resque
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.20.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.20.0
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: When your resque jobs are frequent and fast, the overhead of forking
+  and running your after_fork might get too big.
+email:
+- mick@zendesk.com
+- luke@lividpenguin.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/resque-multi-job-forks.rb
+- test/helper.rb
+- test/test_resque-multi-job-forks.rb
+homepage: http://github.com/staugaard/resque-multi-job-forks
+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.24
+signing_key: 
+specification_version: 3
+summary: Have your resque workers process more that one job
+test_files:
+- test/helper.rb
+- test/test_resque-multi-job-forks.rb