delayed_job_master 1.0.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 97335497d66ae19f719fdfd655d184507e985b42
+  data.tar.gz: d723ca1ed73489e737cf2653e4685ab0644a4a26
+SHA512:
+  metadata.gz: 56e8b33e80dee3d82b370f3275cc638fee882a458260c5304977c9379fcb6523aec185231038e0629fefb090c78d960ed4fa28e674e4b1a38d702b0492dacce0
+  data.tar.gz: 4b26b8ffbe453683d07809afc5a968dcb115429b0c41a1b2593da12996e5ba1d814e65126d9afa63ad63ae52d353322be2f02385ce4e630e7ff849e549d522be

data/.gitignore

@@ -0,0 +1,19 @@
+/.bundle/
+/.yardoc
+/.project
+/Gemfile.lock
+/gemfiles/.bundle
+/gemfiles/vendor
+/gemfiles/*gemfile.lock
+/_yardoc/
+/bin/
+/coverage/
+/doc/
+/pkg/
+/log/
+/tmp/
+/spec/reports/
+/spec/**/db/*.sqlite3
+/spec/**/log/*.log
+/spec/**/tmp/pids/*.pid
+/vendor/

data/.rspec

@@ -0,0 +1,4 @@
+--format documentation
+--color
+--require spec_helper
+--require rails_helper

data/.travis.yml

@@ -0,0 +1,16 @@
+language: ruby
+services:
+  - mongodb
+rvm:
+  - 2.3.4
+  - 2.4.1
+gemfile:
+  - gemfiles/active_record.gemfile
+  - gemfiles/mongoid.gemfile
+before_install:
+  - gem install bundler -v 1.15.3
+before_script:
+  - cd spec/dummy
+  - bundle exec rake db:migrate RAILS_ENV=test
+  - cd ../..
+script: bundle exec rspec

data/CODE_OF_CONDUCT.md

@@ -0,0 +1,49 @@
+# Contributor Code of Conduct
+
+As contributors and maintainers of this project, and in the interest of
+fostering an open and welcoming community, we pledge to respect all people who
+contribute through reporting issues, posting feature requests, updating
+documentation, submitting pull requests or patches, and other activities.
+
+We are committed to making participation in this project a harassment-free
+experience for everyone, regardless of level of experience, gender, gender
+identity and expression, sexual orientation, disability, personal appearance,
+body size, race, ethnicity, age, religion, or nationality.
+
+Examples of unacceptable behavior by participants include:
+
+* The use of sexualized language or imagery
+* Personal attacks
+* Trolling or insulting/derogatory comments
+* Public or private harassment
+* Publishing other's private information, such as physical or electronic
+  addresses, without explicit permission
+* Other unethical or unprofessional conduct
+
+Project maintainers have the right and responsibility to remove, edit, or
+reject comments, commits, code, wiki edits, issues, and other contributions
+that are not aligned to this Code of Conduct, or to ban temporarily or
+permanently any contributor for other behaviors that they deem inappropriate,
+threatening, offensive, or harmful.
+
+By adopting this Code of Conduct, project maintainers commit themselves to
+fairly and consistently applying these principles to every aspect of managing
+this project. Project maintainers who do not follow or enforce the Code of
+Conduct may be permanently removed from the project team.
+
+This code of conduct applies both within project spaces and in public spaces
+when an individual is representing the project or its community.
+
+Instances of abusive, harassing, or otherwise unacceptable behavior may be
+reported by contacting a project maintainer at kaneta@sitebridge.co.jp. All
+complaints will be reviewed and investigated and will result in a response that
+is deemed necessary and appropriate to the circumstances. Maintainers are
+obligated to maintain confidentiality with regard to the reporter of an
+incident.
+
+This Code of Conduct is adapted from the [Contributor Covenant][homepage],
+version 1.3.0, available at
+[http://contributor-covenant.org/version/1/3/0/][version]
+
+[homepage]: http://contributor-covenant.org
+[version]: http://contributor-covenant.org/version/1/3/0/

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in delayed_job_master.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2017 Yoshikazu Kaneta
+
+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.md

@@ -0,0 +1,158 @@
+# DelayedJobMaster
+
+A simple delayed_job master process to control multiple workers.
+
+## Features
+
+* Preload application and fork workers fastly.
+* Monitor workers and fork new workers if necessary.
+* Restart workers with memory limitation.
+* Trap USR1 signal to reopen log files.
+* Trap USR2 signal to restart master and workers.
+* Auto-scale worker processes.
+
+## Dependencies
+
+* ruby 2.3+
+* delayed_job 4.1
+
+Supported delayed_job backends:
+
+* delayed_job_active_record 4.1
+* delayed_job_mongoid 2.3
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'delayed_job_master', require: false
+```
+
+And then execute:
+
+    $ bundle
+
+Create default files:
+
+    $ rails generate delayed_job_master
+
+This command creates `bin/delayed_job_master` and `config/delayed_job_master.rb`.
+
+## Configuration
+
+Edit `config/delayed_job_master.rb`:
+
+```ruby
+# working directory
+working_directory Dir.pwd
+
+# monitor wait time in second
+monitor_wait 5
+
+# path to pid file
+pid_file "#{Dir.pwd}/tmp/pids/delayed_job_master.pid"
+
+# path to log file
+log_file "#{Dir.pwd}/log/delayed_job_master.log"
+
+# log level
+log_level :info
+
+# worker1
+add_worker do |worker|
+  # queue name for the worker
+  worker.queues %w(queue1)
+
+  # worker control (:static or :dynamic)
+  worker.control :static
+
+  # worker count
+  worker.count 1
+
+  # max memory in MB
+  worker.max_memory 300
+
+  # configs below are same as delayed_job, see https://github.com/collectiveidea/delayed_job
+  # worker.sleep_delay 5
+  # worker.read_ahead 5
+  # worker.max_attempts 25
+  # worker.max_run_time 4.hours
+  # worker.min_priority 1
+  # worker.max_priority 10
+end
+
+# worker2
+add_worker do |worker|
+  worker.queues %w(queue2)
+  worker.control :dynamic
+  worker.count 2
+end
+
+before_fork do |master, worker_info|
+  Delayed::Worker.before_fork if defined?(Delayed::Worker)
+end
+
+after_fork do |master, worker_info|
+  Delayed::Worker.after_fork if defined?(Delayed::Worker)
+end
+```
+
+## Usage
+
+Start master:
+
+    $ RAILS_ENV=production bin/delayed_job_master -c config/delayed_job_master.rb -D
+
+Command line options:
+
+* -c, --config: Specify configuration file.
+* -D, --daemon: Start master as a daemon.
+
+Stop master and workers gracefully:
+
+    $ kill -TERM `cat tmp/pids/delayed_job_master.pid`
+
+Stop master and workers forcefully:
+
+    $ kill -QUIT `cat tmp/pids/delayed_job_master.pid`
+
+Reopen log files:
+
+    $ kill -USR1 `cat tmp/pids/delayed_job_master.pid`
+
+Restart gracefully:
+
+    $ kill -USR2 `cat tmp/pids/delayed_job_master.pid`
+
+Workers handle each signal as follows:
+
+* TERM: Workers stop after finishing current jobs.
+* QUIT: Workers are killed immediately.
+* USR1: Workers reopen log files.
+* USR2: New workers start, old workers stop after finishing current jobs.
+
+## Worker status
+
+`ps` command shows worker status as follows:
+
+```
+$ ps aux
+... delayed_job.0 (queue1) [BUSY]  # BUSY process is currently proceeding some jobs
+```
+
+After graceful restart, you may find OLD process.
+
+```
+$ ps aux
+... delayed_job.0 (queue1) [BUSY] [OLD]  # OLD process will stop after finishing current jobs.
+```
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/kanety/delayed_job_master. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.
+
+## License
+
+The gem is available as open source under the terms of the [MIT License](http://opensource.org/licenses/MIT).
+

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/delayed_job_master.gemspec

@@ -0,0 +1,34 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'delayed/master/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "delayed_job_master"
+  spec.version       = Delayed::Master::VERSION
+  spec.authors       = ["Yoshikazu Kaneta"]
+  spec.email         = ["kaneta@sitebridge.co.jp"]
+
+  spec.summary       = %q{A simple delayed_job master process to control multiple workers}
+  spec.description   = %q{A simple delayed_job master process to control multiple workers}
+  spec.homepage      = "https://github.com/kanety/delayed_job_master"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.required_ruby_version = '>= 2.3'
+
+  spec.add_dependency "delayed_job", ">= 4.1"
+  spec.add_dependency "get_process_mem"
+
+  spec.add_development_dependency "rails"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec-rails"
+  spec.add_development_dependency "simplecov"
+  spec.add_development_dependency "sqlite3"
+  spec.add_development_dependency "delayed_job_mongoid", ">= 2.3"
+  spec.add_development_dependency "delayed_job_active_record", ">= 4.1"
+end

data/gemfiles/active_record.gemfile

@@ -0,0 +1,5 @@
+source 'https://rubygems.org'
+
+gem "delayed_job_active_record", "~> 4.1.0"
+
+gemspec path: "../"

data/gemfiles/mongoid.gemfile

@@ -0,0 +1,5 @@
+source 'https://rubygems.org'
+
+gem "delayed_job_mongoid", "~> 2.3.0"
+
+gemspec path: "../"

data/lib/delayed/master.rb

@@ -0,0 +1,115 @@
+require 'fileutils'
+require 'logger'
+require 'ostruct'
+require_relative 'util/file_reopener'
+require_relative 'master/version'
+require_relative 'master/command'
+require_relative 'master/callback'
+require_relative 'master/worker_info'
+require_relative 'master/worker_pool'
+require_relative 'master/signal_handler'
+
+module Delayed
+  class Master
+    attr_reader :config, :logger, :worker_infos
+
+    def initialize(argv)
+      config = Command.new(argv).config
+      @config = OpenStruct.new(config).freeze
+      @logger = setup_logger(@config.log_file, @config.log_level)
+      @worker_infos = []
+
+      @signal_handler = SignalHandler.new(self)
+      @worker_pool = WorkerPool.new(self, config)
+    end
+
+    def run
+      load_app
+      show_worker_configs
+      daemonize if @config.daemon
+
+      create_pid_file
+      @logger.info "started master #{Process.pid}"
+
+      @signal_handler.register
+      @worker_pool.init
+      @worker_pool.monitor_while { stop? }
+
+      remove_pid_file
+      @logger.info "shut down master"
+    end
+
+    def load_app
+      require File.join(@config.working_directory, 'config', 'environment')
+      require_relative 'master/job_counter'
+      require_relative 'worker/extension'
+    end
+
+    def prepared?
+      @worker_pool.prepared?
+    end
+
+    def quit
+      kill_workers
+      @stop = true
+    end
+
+    def stop
+      @signal_handler.dispatch('TERM')
+      @stop = true
+    end
+
+    def stop?
+      @stop
+    end
+
+    def reopen_files
+      @signal_handler.dispatch('USR1')
+      @logger.info "reopening files..."
+      Delayed::Util::FileReopener.reopen
+      @logger.info "reopened"
+    end
+
+    def restart
+      @signal_handler.dispatch('USR2')
+      @logger.info "restarting master..."
+      exec(*([$0] + ARGV))
+    end
+
+    def kill_workers
+      @signal_handler.dispatch('KILL')
+    end
+
+    def wait_workers
+      Process.waitall
+    end
+
+    private
+
+    def setup_logger(log_file, log_level)
+      FileUtils.mkdir_p(File.dirname(log_file))
+      logger = Logger.new(log_file)
+      logger.level = log_level
+      logger
+    end
+
+    def create_pid_file
+      FileUtils.mkdir_p(File.dirname(@config.pid_file))
+      File.write(@config.pid_file, Process.pid)
+    end
+
+    def remove_pid_file
+      File.delete(@config.pid_file) if File.exist?(@config.pid_file)
+    end
+
+    def daemonize
+      Process.daemon(true)
+    end
+
+    def show_worker_configs
+      @config.worker_configs.each do |config|
+        puts "#{config[:count]} worker for '#{config[:queues].join(',')}' under #{config[:control]} control"
+      end
+    end
+  end
+end

data/lib/delayed/master/callback.rb

@@ -0,0 +1,13 @@
+module Delayed
+  class Master
+    class Callback
+      def initialize(config = {})
+        @config = config.select { |k, _| [:before_fork, :after_fork].include?(k) }
+      end
+
+      def run(name, *args)
+        @config[name].call(*args) if @config[name]
+      end
+    end
+  end
+end

data/lib/delayed/master/command.rb

@@ -0,0 +1,32 @@
+require 'optparse'
+require_relative 'dsl'
+
+module Delayed
+  class Master
+    class Command
+      attr_reader :config
+
+      def initialize(args)
+        @config = {}
+
+        OptionParser.new { |opt|
+          opt.banner = <<-EOS
+            #{File.basename($PROGRAM_NAME)} #{Delayed::Master::VERSION}
+            Usage: #{File.basename($PROGRAM_NAME)} [options]
+          EOS
+
+          opt.on('-h', '--help', '-v', '--version', 'Show this message') do |boolean|
+            puts opt
+            exit
+          end
+          opt.on('-D', '--daemon', 'Start master as a daemon') do |boolean|
+            @config[:daemon] = boolean
+          end
+          opt.on('-c', '--config=FILE', 'Specify config file') do |file|
+            @config.merge!(DSL.new(file).config)
+          end
+        }.parse(args)
+      end
+    end
+  end
+end

data/lib/delayed/master/dsl.rb

@@ -0,0 +1,60 @@
+module Delayed
+  class Master
+    class DSL
+      SIMPLE_CONFIGS   = [:working_directory, :log_file, :log_level, :pid_file, :monitor_wait]
+      CALLBACK_CONFIGS = [:before_fork, :after_fork]
+
+      attr_reader :config
+
+      def initialize(config_file)
+        @config = { worker_configs: [] }
+        instance_eval(File.read(config_file))
+        @config
+      end
+
+      def add_worker
+        setting = WorkerSetting.new(control: :static, count: 1)
+        yield setting
+        setting.config[:exit_on_complete] = true if setting.config[:control] == :dynamic
+        @config[:worker_configs] << setting.config
+      end
+
+      SIMPLE_CONFIGS.each do |key|
+        define_method(key) do |value|
+          @config[key] = value
+        end
+      end
+
+      CALLBACK_CONFIGS.each do |key|
+        define_method(key) do |&block|
+          @config[key] = block
+        end
+      end
+
+      class WorkerSetting
+        SIMPLE_CONFIGS = [:control, :count, :max_memory,
+                          :min_priority, :max_priority, :sleep_delay, :read_ahead, :exit_on_complete,
+                          :max_attempts, :max_run_time]
+        ARRAY_CONFIGS  = [:queues]
+
+        attr_reader :config
+
+        def initialize(default = {})
+          @config = default
+        end
+
+        SIMPLE_CONFIGS.each do |key|
+          define_method(key) do |value|
+            @config[key] = value
+          end
+        end
+
+        ARRAY_CONFIGS.each do |key|
+          define_method(key) do |value|
+            @config[key] = Array(value)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/delayed/master/job_counter.rb

@@ -0,0 +1,8 @@
+case Delayed::Worker.backend.to_s
+when 'Delayed::Backend::ActiveRecord::Job'
+  require_relative 'job_counter/active_record'
+when 'Delayed::Backend::Mongoid::Job'
+  require_relative 'job_counter/mongoid'
+else
+  raise "Unsupported backend: #{Delayed::Worker.backend}"
+end

data/lib/delayed/master/job_counter/active_record.rb

@@ -0,0 +1,24 @@
+# JobCounter depends on delayed_job_active_record.
+# See https://github.com/collectiveidea/delayed_job_active_record/blob/master/lib/delayed/backend/active_record.rb
+module Delayed
+  class Master
+    class JobCounter
+      class << self
+        def count(config)
+          jobs = ready_to_run(config[:max_run_time] || Delayed::Worker::DEFAULT_MAX_RUN_TIME)
+          jobs.where!("priority >= ?", config[:min_priority]) if config[:min_priority]
+          jobs.where!("priority <= ?", config[:max_priority]) if config[:max_priority]
+          jobs.where!(queue: config[:queues]) if config[:queues].any?
+          jobs.count
+        end
+
+        private
+
+        def ready_to_run(max_run_time)
+          db_time_now = Delayed::Job.db_time_now
+          Delayed::Job.where("(run_at <= ? AND (locked_at IS NULL OR locked_at < ?)) AND failed_at IS NULL", db_time_now, db_time_now - max_run_time)
+        end
+      end
+    end
+  end
+end

data/lib/delayed/master/job_counter/mongoid.rb

@@ -0,0 +1,32 @@
+# JobCounter depends on delayed_job_mongoid.
+# See https://github.com/collectiveidea/delayed_job_mongoid/blob/master/lib/delayed/backend/mongoid.rb
+module Delayed
+  class Master
+    class JobCounter
+      class << self
+        def count(config)
+          right_now = Delayed::Job.db_time_now
+          jobs = reservation_criteria(right_now, config[:max_run_time] || Delayed::Worker::DEFAULT_MAX_RUN_TIME)
+          jobs = jobs.gte(priority: config[:min_priority].to_i) if config[:min_priority]
+          jobs = jobs.lte(priority: config[:max_priority].to_i) if config[:max_priority]
+          jobs = jobs.any_in(queue: config[:queues]) if config[:queues].any?
+          jobs.count
+        end
+
+        private
+
+        def reservation_criteria(right_now, max_run_time)
+          criteria = Delayed::Job.where(
+            run_at: { '$lte' => right_now },
+            failed_at: nil
+          ).any_of(
+            { locked_at: nil },
+            locked_at: { '$lt' => (right_now - max_run_time) }
+          )
+
+          criteria
+        end
+      end
+    end
+  end
+end

data/lib/delayed/master/signal_handler.rb

@@ -0,0 +1,43 @@
+module Delayed
+  class Master
+    class SignalHandler
+      def initialize(master)
+        @master = master
+        @logger = master.logger
+        @worker_infos = master.worker_infos
+      end
+
+      def register
+        %w(TERM INT QUIT USR1 USR2).each do |signal|
+          trap(signal) do
+            Thread.new do
+              @logger.info "received #{signal} signal"
+              case signal
+              when 'TERM', 'INT'
+                @master.stop
+              when 'QUIT'
+                @master.quit
+              when 'USR1'
+                @master.reopen_files
+              when 'USR2'
+                @master.restart
+              end
+            end
+          end
+        end
+      end
+
+      def dispatch(signal)
+        @worker_infos.each do |worker_info|
+          next unless worker_info.pid
+          begin
+            Process.kill signal, worker_info.pid
+            @logger.info "sent #{signal} signal to worker #{worker_info.pid}"
+          rescue
+            @logger.error "failed to send #{signal} signal to worker #{worker_info.pid}"
+          end
+        end
+      end
+    end
+  end
+end

data/lib/delayed/master/version.rb

@@ -0,0 +1,5 @@
+module Delayed
+  class Master
+    VERSION = "1.0.0"
+  end
+end

data/lib/delayed/master/worker_info.rb

@@ -0,0 +1,19 @@
+module Delayed
+  class Master
+    class WorkerInfo
+      attr_reader :index, :config
+      attr_accessor :pid
+
+      def initialize(index, config = {})
+        @index = index
+        @config = config
+      end
+
+      def title
+        titles = ["delayed_job.#{@index}"]
+        titles << "(#{@config[:queues].join(',')})" if @config[:queues]
+        titles.join(' ')
+      end
+    end
+  end
+end

data/lib/delayed/master/worker_pool.rb

@@ -0,0 +1,128 @@
+module Delayed
+  class Master
+    class WorkerPool
+      def initialize(master, config = {})
+        @master = master
+        @logger = master.logger
+        @worker_infos = master.worker_infos
+
+        @config = OpenStruct.new(config).freeze
+        @static_worker_configs, @dynamic_worker_configs = @config.worker_configs.partition { |wc| wc[:control] == :static }
+
+        @callback = Delayed::Master::Callback.new(config)
+      end
+
+      def init
+        @static_worker_configs.each_with_index do |config, i|
+          worker_info = Delayed::Master::WorkerInfo.new(i, config)
+          @worker_infos << worker_info
+          fork_worker(worker_info)
+          @logger.info "started worker #{worker_info.pid}"
+        end
+
+        @prepared = true
+        debug_worker_infos
+      end
+
+      def monitor_while(&block)
+        loop do
+          break if block.call
+          check_pid
+          check_dynamic_worker
+          sleep @config.monitor_wait.to_i
+        end
+      end
+
+      def prepared?
+        @prepared
+      end
+
+      private
+
+      def fork_worker(worker_info)
+        @callback.run(:before_fork, @master, worker_info)
+        worker_info.pid = fork do
+          @callback.run(:after_fork, @master, worker_info)
+          $0 = worker_info.title
+          worker = create_new_worker(worker_info)
+          worker.start
+        end
+      end
+
+      def create_new_worker(worker_info)
+        worker = Delayed::Worker.new(worker_info.config)
+        [:max_run_time, :max_attempts].each do |key|
+          value = worker_info.config[key]
+          Delayed::Worker.send("#{key}=", value) if value
+        end
+        [:max_memory].each do |key|
+          value = worker_info.config[key]
+          worker.send("#{key}=", value) if value
+        end
+        worker.master_logger = @logger
+        worker
+      end
+
+      def check_pid
+        pid = wait_pid
+        return unless pid
+        worker_info = @worker_infos.detect { |wi| wi.pid == pid }
+        return unless worker_info
+
+        case worker_info.config[:control]
+        when :static
+          fork_alt_worker(worker_info)
+        when :dynamic
+          @worker_infos.delete(worker_info)
+        end
+      end
+
+      def wait_pid
+        Process.waitpid(-1, Process::WNOHANG)
+      rescue Errno::ECHILD
+        nil
+      end
+
+      def check_dynamic_worker
+        @dynamic_worker_configs.each do |worker_config|
+          current_count = @worker_infos.count { |wi| wi.config[:queues] == worker_config[:queues] }
+          remaining_count = worker_config[:count] - current_count
+          if remaining_count > 0 && (job_count = count_job_for_worker(worker_config)) > 0
+            [remaining_count, job_count].min.times do
+              fork_dynamic_worker(worker_config)
+            end
+          end
+        end
+      end
+
+      def count_job_for_worker(worker_config)
+        Delayed::Master::JobCounter.count(worker_config)
+      end
+
+      def fork_dynamic_worker(worker_config)
+        worker_info = Delayed::Master::WorkerInfo.new(@worker_infos.size, worker_config)
+        @worker_infos << worker_info
+
+        @logger.info "forking dynamic worker..."
+        fork_worker(worker_info)
+        @logger.info "forked worker #{worker_info.pid}"
+
+        debug_worker_infos
+      end
+
+      def fork_alt_worker(worker_info)
+        @logger.info "worker #{worker_info.pid} seems to be killed, forking alternative worker..."
+        fork_worker(worker_info)
+        @logger.info "forked worker #{worker_info.pid}"
+
+        debug_worker_infos
+      end
+
+      def debug_worker_infos
+        @worker_infos.each do |worker_info|
+          @logger.debug "#{worker_info.pid}: #{worker_info.title}"
+        end
+      end
+    end
+  end
+end

data/lib/delayed/util/file_reopener.rb

@@ -0,0 +1,16 @@
+module Delayed
+  class Util
+    class FileReopener
+      class << self
+        def reopen
+          ObjectSpace.each_object(File) do |file|
+            next if file.closed? || !file.sync
+            file.reopen file.path 
+            file.sync = true
+            file.flush
+          end
+        end
+      end
+    end
+  end
+end

data/lib/delayed/worker/extension.rb

@@ -0,0 +1,9 @@
+require_relative 'plugins/memory_checker'
+require_relative 'plugins/signal_handler'
+require_relative 'plugins/status_notifier'
+
+module Delayed
+  class Worker
+    attr_accessor :master_logger, :max_memory
+  end
+end

data/lib/delayed/worker/plugins/memory_checker.rb

@@ -0,0 +1,16 @@
+require 'get_process_mem'
+
+class Delayed::Plugins::WorkerMemoryChecker < Delayed::Plugin
+  callbacks do |lifecycle|
+    lifecycle.after(:perform) do |worker, job|
+      next unless worker.max_memory
+      mem = GetProcessMem.new
+      if mem.mb > worker.max_memory
+        worker.master_logger.info "shutting down worker #{Process.pid} because it consumes large memory #{mem.mb.to_i} MB..."
+        worker.stop
+      end
+    end
+  end
+end
+
+Delayed::Worker.plugins << Delayed::Plugins::WorkerMemoryChecker

data/lib/delayed/worker/plugins/signal_handler.rb

@@ -0,0 +1,27 @@
+class Delayed::Plugins::SignalHandler < Delayed::Plugin
+  callbacks do |lifecycle|
+    lifecycle.before(:execute) do |worker|
+      worker.instance_eval do
+        trap('USR1') do
+          Thread.new do
+            master_logger.info "reopening files..."
+            Delayed::Util::FileReopener.reopen
+            master_logger.info "reopened"
+          end
+        end
+        trap('USR2') do
+          Thread.new do
+            $0 = "#{$0} [OLD]"
+            master_logger.info "shutting down worker #{Process.pid}..."
+            stop
+          end
+        end
+      end
+    end
+    lifecycle.after(:execute) do |worker|
+      worker.master_logger.info "shut down worker #{Process.pid}"
+    end
+  end
+end
+
+Delayed::Worker.plugins << Delayed::Plugins::SignalHandler

data/lib/delayed/worker/plugins/status_notifier.rb

@@ -0,0 +1,13 @@
+class Delayed::Plugins::StatusNotifier < Delayed::Plugin
+  callbacks do |lifecycle|
+    lifecycle.around(:perform) do |worker, job, &block|
+      title = $0
+      $0 = "#{title} [BUSY]"
+      ret = block.call
+      $0 = title
+      ret
+    end
+  end
+end
+
+Delayed::Worker.plugins << Delayed::Plugins::StatusNotifier

data/lib/generators/delayed_job_master_generator.rb

@@ -0,0 +1,14 @@
+require 'rails/generators'
+
+class DelayedJobMasterGenerator < Rails::Generators::Base
+  source_root File.join(File.dirname(__FILE__), 'templates')
+
+  def create_script_file
+    template 'script', 'bin/delayed_job_master'
+    chmod 'bin/delayed_job_master', 0o755
+  end
+
+  def create_config_file
+    template 'config.rb', 'config/delayed_job_master.rb'
+  end
+end

data/lib/generators/templates/config.rb

@@ -0,0 +1,52 @@
+# working directory
+working_directory Dir.pwd
+
+# monitor wait time in second
+monitor_wait 5
+
+# path to pid file
+pid_file "#{Dir.pwd}/tmp/pids/delayed_job_master.pid"
+
+# path to log file
+log_file "#{Dir.pwd}/log/delayed_job_master.log"
+
+# log level
+log_level :info
+
+# worker1
+add_worker do |worker|
+  # queue name for the worker
+  worker.queues %w(queue1)
+
+  # worker control (:static or :dynamic)
+  worker.control :static
+
+  # worker count
+  worker.count 1
+
+  # max memory in MB
+  worker.max_memory 300
+
+  # configs below are same as delayed_job, see https://github.com/collectiveidea/delayed_job
+  # worker.sleep_delay 5
+  # worker.read_ahead 5
+  # worker.max_attempts 25
+  # worker.max_run_time 4.hours
+  # worker.min_priority 1
+  # worker.max_priority 10
+end
+
+# worker2
+add_worker do |worker|
+  worker.queues %w(queue2)
+  worker.control :dynamic
+  worker.count 2
+end
+
+before_fork do |master, worker_info|
+  Delayed::Worker.before_fork if defined?(Delayed::Worker)
+end
+
+after_fork do |master, worker_info|
+  Delayed::Worker.after_fork if defined?(Delayed::Worker)
+end

data/lib/generators/templates/script

@@ -0,0 +1,5 @@
+#!/usr/bin/env ruby
+
+require 'bundler/setup'
+require 'delayed/master'
+Delayed::Master.new(ARGV).run

metadata

@@ -0,0 +1,200 @@
+--- !ruby/object:Gem::Specification
+name: delayed_job_master
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+platform: ruby
+authors:
+- Yoshikazu Kaneta
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2017-11-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: delayed_job
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.1'
+- !ruby/object:Gem::Dependency
+  name: get_process_mem
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec-rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: simplecov
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: delayed_job_mongoid
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '2.3'
+- !ruby/object:Gem::Dependency
+  name: delayed_job_active_record
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '4.1'
+description: A simple delayed_job master process to control multiple workers
+email:
+- kaneta@sitebridge.co.jp
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- CODE_OF_CONDUCT.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- delayed_job_master.gemspec
+- gemfiles/active_record.gemfile
+- gemfiles/mongoid.gemfile
+- lib/delayed/master.rb
+- lib/delayed/master/callback.rb
+- lib/delayed/master/command.rb
+- lib/delayed/master/dsl.rb
+- lib/delayed/master/job_counter.rb
+- lib/delayed/master/job_counter/active_record.rb
+- lib/delayed/master/job_counter/mongoid.rb
+- lib/delayed/master/signal_handler.rb
+- lib/delayed/master/version.rb
+- lib/delayed/master/worker_info.rb
+- lib/delayed/master/worker_pool.rb
+- lib/delayed/util/file_reopener.rb
+- lib/delayed/worker/extension.rb
+- lib/delayed/worker/plugins/memory_checker.rb
+- lib/delayed/worker/plugins/signal_handler.rb
+- lib/delayed/worker/plugins/status_notifier.rb
+- lib/generators/delayed_job_master_generator.rb
+- lib/generators/templates/config.rb
+- lib/generators/templates/script
+homepage: https://github.com/kanety/delayed_job_master
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '2.3'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.2.1
+signing_key: 
+specification_version: 4
+summary: A simple delayed_job master process to control multiple workers
+test_files: []