forkr 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 76c883a35bda24bd878cfa72a2a45f506d67cd2c
+  data.tar.gz: e378330ad4969a24d3554216a0a7a0521aa0df0e
+SHA512:
+  metadata.gz: bf8932667a70eb7b31b6d0ce367f17faa166267d77c0a49eb3317b25e33454d8439c6edcb97f1df0dbc6e35f729a92fc364713f4067977244127b53e6348c5a1
+  data.tar.gz: 5da12ece3a39c67f62e41295a9226cb736c33d666bf79d081e6d02e4b7fe77e87c7b9c8d7823706ede81a604250bb2e7c3dc2878cef25f704c2aa5defaed9135

data/lib/forkr.rb

@@ -0,0 +1,155 @@
+class Forkr
+  attr_reader :master_pid, :children, :inbound, :outbound, :child_count
+
+  def initialize(forklet, num_kids = 1)
+    @worker_client = forklet 
+    @master_pid = $$
+    @children = []
+    @child_count = num_kids
+    @in_shutdown = false
+  end
+
+  def run
+    @inbound, @outbound = IO.pipe
+    Signal.trap('CHLD') { dead_child }
+    Signal.trap('INT') { interrupt }
+    Signal.trap('TERM') { shutdown }
+    Signal.trap('QUIT') { core_dump_quit }
+    Signal.trap('TTIN') { add_worker }
+    Signal.trap('TTOU') { remove_worker }
+    master_loop
+  end
+
+  def send_wake_notice(notice)
+    return(nil) if $$ != master_pid
+    return(nil) if @in_shutdown
+    @outbound.write(notice)
+  end
+
+  def core_dump_quit
+    send_wake_notice("Q")
+  end
+
+  def add_worker
+    send_wake_notice("+")
+  end
+
+  def remove_worker
+    send_wake_notice("-")
+  end
+
+  def interrupt
+    send_wake_notice("I")
+  end
+
+  def shutdown
+    send_wake_notice("T")
+  end
+
+  def dead_child
+    send_wake_notice("D")
+  end
+
+  def spawn_worker
+    if new_pid = fork
+      @children << new_pid
+    else
+      worker_loop
+    end
+  end
+
+  def increment_workers
+    @child_count = @child_count + 1
+  end
+
+  def decrement_workers
+    if @child_count > 1
+      @child_count = @child_count - 1
+    end
+  end
+
+  def shutdown_using(sig)
+    @in_shutdown = true
+    signal_all_workers(sig)
+    raise StopIteration.new
+  end
+
+  def master_loop
+    ensure_right_worker_count
+    loop do
+      fds = IO.select([@inbound],nil,nil,2)
+      unless fds.nil?
+        data_read = fds.first.first.read(1)
+        if data_read == "I"
+          shutdown_using(:INT)
+        elsif data_read == "T"
+          shutdown_using(:TERM)
+        elsif data_read == "Q"
+          shutdown_using(:QUIT)
+        elsif data_read == "+"
+          increment_workers
+        elsif data_read == "-"
+          decrement_workers
+        end
+      end
+      prune_workers
+      ensure_right_worker_count
+    end
+    reap_all_workers
+    @outbound.close
+    @inbound.close
+  end
+
+  def reap_all_workers
+    begin
+      wpid, status = Process.waitpid2(-1, Process::WNOHANG)
+    rescue Errno::ECHILD
+      break
+    end while true
+  end
+
+  def ensure_right_worker_count
+    existing_workers = @children.length
+    off_by = @child_count - @children.length
+    if off_by > 0
+      off_by.times do
+        spawn_worker
+      end
+    elsif off_by < 0
+      @children.take(off_by.abs).each do |kid|
+        signal_worker(kid, :TERM)
+      end
+    end
+  end
+
+  def signal_all_workers(sig)
+    @children.each { |c| signal_worker(c, sig) }
+  end
+
+  def signal_worker(wpid, signal)
+    begin
+      Process.kill(signal, wpid)
+    rescue Errno::ESRCH
+    end
+  end
+
+  def prune_workers
+    @children = @children.reject { |pid| child_dead?(pid) }
+  end
+
+  def worker_loop
+    @worker_client.after_fork if @worker_client.respond_to?(:after_fork)
+    @inbound.close
+    @outbound.close
+    $stderr.puts "Worker spawned as #{$$}!"
+    @worker_client.run
+  end
+
+  def child_dead?(pid)
+    status = Process.waitpid(pid, Process::WNOHANG)
+    unless status.nil?
+      $stderr.puts "Process #{pid} dead: #{status}"
+    end
+    !status.nil?
+  end
+end

metadata

@@ -0,0 +1,44 @@
+--- !ruby/object:Gem::Specification
+name: forkr
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Trey Evans
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2014-12-01 00:00:00.000000000 Z
+dependencies: []
+description: A pre-forking worker host.
+email: lewis.r.evans@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/forkr.rb
+homepage: http://rubygems.org/gems/hola
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.2.2
+signing_key: 
+specification_version: 4
+summary: A pre-forking worker host - shamelessly inspired by unicorn.
+test_files: []