preforker 0.1.0

data/.document

@@ -0,0 +1,5 @@
+README.rdoc
+lib/**/*.rb
+bin/*
+features/**/*.feature
+LICENSE

data/.gitignore

@@ -0,0 +1,24 @@
+## MAC OS
+.DS_Store
+
+## TEXTMATE
+*.tmproj
+tmtags
+
+## EMACS
+*~
+\#*
+.\#*
+
+## VIM
+*.swp
+
+## PROJECT::GENERAL
+coverage
+rdoc
+pkg
+
+## PROJECT::SPECIFIC
+*.log
+*.pid
+

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Daniel Cadenas
+
+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.rdoc

@@ -0,0 +1,77 @@
+= preforker
+
+A gem to easily create prefork servers.
+
+== Example
+
+Let's see an example using the Mac 'say' command.
+
+  require 'preforker'
+
+  #you can open some socket here or reserve any other resource you want to share with your workers, this is master space
+  Preforker.new(:timeout => 10) do |master|
+    #this block is only run from each worker (10 by default)
+
+    #here you should write the code that is needed to be ran each time a fork is created, initializations, etc.
+    `say hi`
+
+    #here you could IO.select a socket, run an EventMachine service (see example), or just run worker loop
+    #you need to ask master if it wants you alive periodically or else it will kill you after the timeout elapses. Respect your master!
+    while master.wants_me_alive? do
+      sleep 1
+      `say ping pong`
+    end
+
+    #here we can do whatever we want when exiting gracefully
+    `say bye`
+  end.start
+
+To kill the server just run:
+
+  kill -s QUIT `cat preforker.pid`
+
+See the examples directory and the specs for more examples.
+
+== Configuration options
+
+[:timeout] The timeout in seconds, 5 by default. If a worker takes more than this it will be killed and respawned.
+[:workers] Number of workers, 10 by default.
+[:stdout_path] Path to redirect stdout to. By default it's /dev/null
+[:stderr_path] Path to redirect stderr to. By default it's /dev/null
+[:app_name] The app name, 'preforker' by default. Used for some ps message, log messages messages and pid file name.
+[:pid_path] The path to the pid file for this server. By default it's './preforker.pid'.
+[:logger] This is Logger.new('./preforker.log') by default
+
+== Signals
+
+You can send some signals to master to control the way it handles the workers lifetime.
+
+[WINCH] Gracefully kill all workers but keep master alive
+[TTIN] Increase number of workers
+[TTOU] Decrease number of workers
+[QUIT] Kill workers and master in a graceful way
+[TERM, INT] Kill workers and master immediately
+
+== Acknowledgments
+
+Most of the preforking operating system tricks come from Unicorn. Check out its source code for a hands on tutorial on Unix internals!
+
+== To do list
+
+* More tests
+* Log rotation through the USR1 signal
+* Have something like min_spare_workers, max_workers
+
+== Note on Patches/Pull Requests
+
+* Fork the project.
+* Make your feature addition or bug fix.
+* Add tests for it. This is important so I don't break it in a
+  future version unintentionally.
+* Commit, do not mess with rakefile, version, or history.
+  (if you want to have your own version, that is fine but bump version in a commit by itself I can ignore when I pull)
+* Send me a pull request. Bonus points for topic branches.
+
+== Copyright
+
+Copyright (c) 2010 Daniel Cadenas. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,46 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "preforker"
+    gem.summary = %Q{A gem to easily create prefork servers.}
+    gem.description = %Q{A gem to easily create prefork servers.}
+    gem.email = "dcadenas@gmail.com"
+    gem.homepage = "http://github.com/dcadenas/preforker"
+    gem.authors = ["Daniel Cadenas"]
+    gem.add_development_dependency "rspec", ">= 1.3.0"
+    gem.add_development_dependency "filetesthelper", ">= 0.10.1"
+    # gem is a Gem::Specification... see http://www.rubygems.org/read/chapter/20 for additional settings
+  end
+  Jeweler::GemcutterTasks.new
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: gem install jeweler"
+end
+
+require 'spec/rake/spectask'
+Spec::Rake::SpecTask.new(:spec) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.spec_files = FileList['spec/**/*_spec.rb']
+end
+
+Spec::Rake::SpecTask.new(:rcov) do |spec|
+  spec.libs << 'lib' << 'spec'
+  spec.pattern = 'spec/**/*_spec.rb'
+  spec.rcov = true
+end
+
+task :spec => :check_dependencies
+
+task :default => :spec
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "preforker #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.1.0

data/examples/amqp.rb

@@ -0,0 +1,24 @@
+require 'rubygems'
+require 'preforker'
+require 'mq'
+
+#be sure to run your AMQP server first
+
+Preforker.new(:timeout => 5, :workers => 4, :app_name => "Amqp example") do |master|
+  AMQP.start(:host => 'dcadenas-laptop.local') do
+    EM.add_periodic_timer(1) do
+      AMQP.stop{ EM.stop } unless master.wants_me_alive?
+    end
+
+    MQ.prefetch(1)
+    channel = MQ.new
+    test_exchange = channel.fanout('test_exchange')
+    channel.queue('test_queue').bind(test_exchange).subscribe(:ack => true) do |h, msg|
+      $stdout.puts "#{$$} received #{msg.inspect}"
+      h.ack
+    end
+  end
+end.start
+
+#run examples/amqp_client.rb to see the output
+#kill server with: kill -s TERM `cat 'amqp example.pid'`

data/examples/amqp_client.rb

@@ -0,0 +1,11 @@
+require 'rubygems'
+require 'bunny'
+
+Bunny.run(:host => 'dcadenas-laptop.local', :port => 5672, :user => 'guest', :password => 'guest') do |b|
+  e = b.exchange('test_exchange', :type => :fanout)
+
+  25.times do |i|
+    e.publish("number #{i}")
+  end
+end
+

data/examples/ping_pong.rb

@@ -0,0 +1,21 @@
+require 'preforker'
+
+#you can open some socket here or reserve any other resource you want to share with your workers, this is master space
+Preforker.new(:timeout => 10) do |master|
+  #this block is only run from each worker (10 by default)
+
+  #here you should write the code that is needed to be ran each time a fork is created, initializations, etc.
+  `say hi`
+
+  #here you could IO.select a socket, run an EventMachine service (see example), or just run worker loop
+  #you need to ask master if it wants you alive periodically or else it will kill you after the timeout elapses. Respect your master!
+  while master.wants_me_alive? do
+    sleep 1
+    `say ping pong`
+  end
+
+  #here we can do whatever we want when exiting gracefully
+  `say bye`
+end.start
+
+#to kill the server just run: kill -s QUIT `cat preforker.pid`

data/lib/preforker/pid_manager.rb

@@ -0,0 +1,59 @@
+class Preforker
+  class  PidManager
+    attr_reader :pid_path, :pid
+
+    def initialize(pid_path)
+      set_pid_path(pid_path, $$)
+    end
+
+    def set_pid_path(new_pid_path, new_pid)
+      if new_pid_path
+        if read_pid = read_path_pid(new_pid_path)
+          return new_pid_path if @pid_path && new_pid_path == @pid_path && read_pid == @pid
+          raise ArgumentError, "#{$$} Already running on PID:#{read_pid} (or #{new_pid_path} is stale)"
+        end
+      end
+      unlink_pid_safe(@pid_path) if @pid_path
+
+      if new_pid_path
+        fp = begin
+               tmp = "#{File.dirname(new_pid_path)}/#{rand}.#{pid}"
+               File.open(tmp, File::RDWR|File::CREAT|File::EXCL, 0644)
+             rescue Errno::EEXIST
+               retry
+             end
+        fp.syswrite("#{new_pid}\n")
+        File.rename(fp.path, new_pid_path)
+        fp.close
+      end
+
+      @pid = new_pid
+      @pid_path = new_pid_path
+    end
+
+    # unlinks a PID file at given +path+ if it contains the current PID
+    # still potentially racy without locking the directory (which is
+    # non-portable and may interact badly with other programs), but the
+    # window for hitting the race condition is small
+    def unlink
+      File.unlink(@pid_path) if @pid_path && File.read(@pid_path).to_i == @pid
+    rescue
+    end
+
+    private
+
+    # returns a PID if a given path contains a non-stale PID file,
+    # nil otherwise.
+    def read_path_pid(path)
+      pid = File.read(path).to_i
+      return nil if pid <= 0
+      begin
+        Process.kill(0, pid)
+        pid
+      rescue Errno::ESRCH
+        # don't unlink stale pid files, racy without non-portable locking...
+      end
+    rescue Errno::ENOENT
+    end
+  end
+end

data/lib/preforker/signal_processor.rb

@@ -0,0 +1,87 @@
+class Preforker
+  class SignalProcessor
+    attr_reader :interesting_signals, :signal_queue
+    def initialize(master)
+      @read_pipe, @write_pipe = IO.pipe
+      @master = master
+      @signal_queue = []
+      @interesting_signals = [:WINCH, :QUIT, :INT, :TERM, :USR1, :USR2, :HUP, :TTIN, :TTOU]
+      @interesting_signals.each { |sig| trap_deferred(sig) }
+    end
+
+    def trap_deferred(signal)
+      trap(signal) do
+        if @signal_queue.size < 5
+          @signal_queue << signal
+          wake_up_master
+        else
+          @master.logger.error "ignoring SIG#{signal}, queue=#{@signal_queue.inspect}"
+        end
+      end
+    end
+
+    def start_signal_loop
+      last_check = Time.now
+      begin
+        loop do
+          @master.reap_all_workers
+          case @signal_queue.shift
+            when nil
+              # avoid murdering workers after our master process (or the
+              # machine) comes out of suspend/hibernation
+              @master.murder_lazy_workers if (last_check + @master.timeout) >= (last_check = Time.now)
+              @master.maintain_worker_count
+              sleep_master
+            when :WINCH
+              @master.logger.info "Gracefully stopping all workers"
+              @master.number_of_workers = 0
+            when :TTIN
+              @master.number_of_workers += 1
+            when :TTOU
+              @master.number_of_workers -= 1 if @master.number_of_workers > 0
+            when :QUIT # graceful shutdown
+              break
+            when :TERM, :INT # immediate shutdown
+              @master.stop(false)
+              break
+          end
+        end
+      rescue Errno::EINTR
+        retry
+      rescue => e
+        @master.logger.error "Unhandled master loop exception #{e.inspect}.\n#{e.backtrace.join("\n")}"
+        retry
+      ensure
+        @master.logger.info "Master quitting"
+        @master.quit
+      end
+    end
+
+    # wait for a signal hander to wake us up and then consume the pipe
+    # Wake up every second anyways to run murder_lazy_workers
+    def sleep_master
+      begin
+        maximum_sleep = @master.timeout > 1 ? 1 : @master.timeout / 2
+        ready = IO.select([@read_pipe], nil, nil, maximum_sleep) or return
+        ready.first && ready.first.first or return
+        chunk_size = 16 * 1024
+        loop { @read_pipe.read_nonblock(chunk_size) }
+      rescue Errno::EAGAIN, Errno::EINTR
+      end
+    end
+
+    def wake_up_master
+      begin
+        @write_pipe.write_nonblock('.') # wakeup master process from select
+      rescue Errno::EAGAIN, Errno::EINTR
+        # pipe is full, master should wake up anyways
+        retry
+      end
+    end
+
+    def reset
+      @interesting_signals.each { |sig| trap(sig, nil) }
+      @signal_queue.clear
+    end
+  end
+end

data/lib/preforker/util.rb

@@ -0,0 +1,42 @@
+# -*- encoding: binary -*-
+
+require 'fcntl'
+require 'tmpdir'
+
+class Preforker
+  class TmpIO < ::File
+
+    # for easier env["rack.input"] compatibility
+    def size
+      # flush if sync
+      stat.size
+    end
+  end
+
+  class Util
+    def self.is_log?(fp)
+      append_flags = File::WRONLY | File::APPEND
+
+      ! fp.closed? &&
+        fp.sync &&
+        fp.path[0] == ?/ &&
+        (fp.fcntl(Fcntl::F_GETFL) & append_flags) == append_flags
+    end
+
+    # creates and returns a new File object.  The File is unlinked
+    # immediately, switched to binary mode, and userspace output
+    # buffering is disabled
+    def self.tmpio
+      fp = begin
+             TmpIO.open("#{Dir::tmpdir}/#{rand}",
+                        File::RDWR|File::CREAT|File::EXCL, 0600)
+           rescue Errno::EEXIST
+             retry
+           end
+      File.unlink(fp.path)
+      fp.binmode
+      fp.sync = true
+      fp
+    end
+  end
+end

data/lib/preforker/worker.rb

@@ -0,0 +1,86 @@
+#all this code is called inside a forked block
+require 'preforker/util'
+
+class Preforker
+  class Worker
+    class WorkerAPI
+      attr_reader :logger
+
+      def initialize(worker, master)
+        @master = master
+        @logger = master.logger
+        @worker = worker
+        @alive = @worker.tmp
+        @last_file_bit = 0
+      end
+
+      def wants_me_alive?
+        if @alive
+          #we are alive, let's be thankful and tell master we are alive and happy
+          @last_file_bit = 0 == @last_file_bit ? 1 : 0
+          @alive.chmod(@last_file_bit)
+        end
+        @alive
+      end
+
+      def kill
+        if @alive
+          @worker.log "Exiting gracefully"
+          @alive = false
+        end
+      end
+    end
+
+    attr_reader :tmp
+    attr_accessor :pid
+
+    def initialize(worker_block, master)
+      @worker_block = worker_block
+      @master = master
+      @tmp = Util.tmpio
+    end
+
+    def init_self_pipe!
+      @read_pipe, @write_pipe = IO.pipe
+      [@read_pipe, @write_pipe].each { |io| io.fcntl(Fcntl::F_SETFD, Fcntl::FD_CLOEXEC) }
+    end
+
+    def work
+      log "Created"
+      init
+
+      @worker_block.call(@master_api) if @master_api.wants_me_alive?
+    end
+
+    def init
+      init_self_pipe!
+      handle_signals
+      tmp.fcntl(Fcntl::F_SETFD, Fcntl::FD_CLOEXEC)
+      @master_api = WorkerAPI.new(self, @master)
+    end
+
+    def handle_signals
+      [:TERM, :INT].each do |sig|
+        trap(sig) { exit!(0) }
+      end
+
+      [:EXIT, :QUIT].each do |sig|
+         trap(sig){ @master_api.kill }
+      end
+
+      #what usr1 does
+      trap(:USR1) { @read_pipe.close rescue nil }
+      trap(:CHLD, 'DEFAULT')
+    end
+
+    def proc_message(message)
+      full_message = ["#{@master.app_name} Child #{$$}", message].join(": ")
+      $0 = full_message
+    end
+
+    def log(message)
+      full_message = proc_message(message)
+      @master.logger.info full_message
+    end
+  end
+end

data/lib/preforker.rb

@@ -0,0 +1,189 @@
+require 'socket'
+require 'fcntl'
+require 'preforker/pid_manager'
+require 'preforker/worker'
+require 'preforker/signal_processor'
+require 'logger'
+
+class Preforker
+  attr_reader :timeout, :app_name, :logger
+  attr_accessor :number_of_workers
+
+  def initialize(options = {}, &worker_block)
+    @app_name = options[:app_name] || "Preforker"
+    default_log_file = "#{@app_name.downcase}.log"
+    @options = {
+      :timeout => 5,
+      :workers => 10,
+      :app_name => "Preforker",
+      :stderr_path => default_log_file,
+      :stderr_path => default_log_file
+    }.merge(options)
+
+    @logger = @options[:logger] || Logger.new(default_log_file)
+
+    @timeout = @options[:timeout]
+    @number_of_workers = @options[:workers]
+    @worker_block = worker_block || lambda {}
+
+    @workers = {}
+    $0 = "#@app_name Master"
+  end
+
+  def start
+    launch do |ready_write|
+      $stdin.reopen("/dev/null")
+      set_stdout_path(@options[:stdout_path])
+      set_stderr_path(@options[:stderr_path])
+
+      logger.info "Master started"
+
+      pid_path = @options[:pid_path] || "./#{@app_name.downcase}.pid"
+      @pid_manager = PidManager.new(pid_path)
+      @signal_processor = SignalProcessor.new(self)
+
+      spawn_missing_workers do
+        ready_write.close
+      end
+
+      #tell parent we are ready
+      ready_write.syswrite($$.to_s)
+      ready_write.close rescue nil
+
+      @signal_processor.start_signal_loop
+    end
+  end
+
+  def launch(&block)
+     puts "Starting server"
+
+     ready_read, ready_write = IO.pipe
+     [ready_read, ready_write].each { |io| io.fcntl(Fcntl::F_SETFD, Fcntl::FD_CLOEXEC) }
+
+     fork do
+       ready_read.close
+
+       Process.setsid
+       fork do
+         block.call(ready_write)
+       end
+     end
+
+     ready_write.close
+     master_pid = (ready_read.readpartial(16) rescue nil).to_i
+     ready_read.close
+     if master_pid <= 1
+       warn "Master failed to start, check stderr log for details"
+       exit!(1)
+     else
+       puts "Server started successfuly"
+       exit(0)
+     end
+  end
+
+  def close_resources_worker_wont_use
+    @signal_processor.reset
+    @workers.values.each { |other| other.tmp.close rescue nil }
+    @workers.clear
+  end
+
+  # Terminates all workers, but does not exit master process
+  def stop(graceful = true)
+    limit = Time.now + @timeout
+    until @workers.empty? || Time.now > limit
+      signal_each_worker(graceful ? :QUIT : :TERM)
+      sleep(0.1)
+      reap_all_workers
+    end
+    signal_each_worker(:KILL)
+  end
+
+  def quit(graceful = true)
+    stop(graceful)
+    @pid_manager.unlink
+  end
+
+  def reap_all_workers
+    begin
+      loop do
+        worker_pid, status = Process.waitpid2(-1, Process::WNOHANG)
+        break unless worker_pid
+        worker = @workers.delete(worker_pid) and worker.tmp.close rescue nil
+        logger.info "reaped #{status.inspect}"
+      end
+    rescue Errno::ECHILD
+    end
+  end
+
+  def spawn_missing_workers(new_workers_count = @number_of_workers, &init_block)
+    new_workers_count.times do
+      worker = Worker.new(@worker_block, self)
+      worker_pid = fork do
+        close_resources_worker_wont_use
+        init_block.call if init_block
+        worker.work
+      end
+
+      worker.pid = worker_pid
+      @workers[worker_pid] = worker
+    end
+  end
+
+  def maintain_worker_count
+    number_of_missing_workers = @number_of_workers - @workers.size
+    return if number_of_missing_workers == 0
+    return spawn_missing_workers(number_of_missing_workers) if number_of_missing_workers > 0
+    @workers.values[0..(-number_of_missing_workers - 1)].each do |unneeded_worker|
+      signal_worker(:QUIT, unneeded_worker.pid) rescue nil
+    end
+  end
+
+  # forcibly terminate all workers that haven't checked in in timeout
+  # seconds.  The timeout is implemented using an unlinked File
+  # shared between the parent process and each worker.  The worker
+  # runs File#chmod to modify the ctime of the File.  If the ctime
+  # is stale for >timeout seconds, then we'll kill the corresponding
+  # worker.
+  def murder_lazy_workers
+    @workers.dup.each_pair do |worker_pid, worker|
+      stat = worker.tmp.stat
+      # skip workers that disable fchmod or have never fchmod-ed
+      next if stat.mode == 0100600
+      next if (diff = (Time.now - stat.ctime)) <= @timeout
+      logger.error "Worker=#{worker_pid} timeout (#{diff}s > #{@timeout}s), killing"
+      signal_worker(:KILL, worker_pid) # take no prisoners for timeout violations
+    end
+  end
+
+  # delivers a signal to a worker and fails gracefully if the worker
+  # is no longer running.
+  def signal_worker(signal, worker_pid)
+    begin
+      Process.kill(signal, worker_pid)
+    rescue Errno::ESRCH
+      worker = @workers.delete(worker_pid) and worker.tmp.close rescue nil
+    end
+  end
+
+  # delivers a signal to each worker
+  def signal_each_worker(signal)
+    @workers.keys.each { |worker_pid| signal_worker(signal, worker_pid) }
+  end
+
+  def signal_quit
+    signal_worker(:QUIT, @pid_manager.pid)
+  end
+
+  def set_stdout_path(path)
+    redirect_io($stdout, path)
+  end
+
+  def set_stderr_path(path)
+    redirect_io($stderr, path)
+  end
+
+  def redirect_io(io, path)
+    File.open(path, 'ab') { |fp| io.reopen(fp) } if path
+    io.sync = true
+  end
+end

data/preforker.gemspec

@@ -0,0 +1,74 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{preforker}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Daniel Cadenas"]
+  s.date = %q{2010-04-02}
+  s.description = %q{A gem to easily create prefork servers.}
+  s.email = %q{dcadenas@gmail.com}
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+     ".gitignore",
+     "LICENSE",
+     "README.rdoc",
+     "Rakefile",
+     "VERSION",
+     "examples/amqp.rb",
+     "examples/amqp_client.rb",
+     "examples/ping_pong.rb",
+     "lib/preforker.rb",
+     "lib/preforker/pid_manager.rb",
+     "lib/preforker/signal_processor.rb",
+     "lib/preforker/util.rb",
+     "lib/preforker/worker.rb",
+     "preforker.gemspec",
+     "spec/integration/logging_spec.rb",
+     "spec/integration/timeout_spec.rb",
+     "spec/integration/worker_control_spec.rb",
+     "spec/spec.opts",
+     "spec/spec_helper.rb",
+     "spec/support/integration.rb"
+  ]
+  s.homepage = %q{http://github.com/dcadenas/preforker}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.6}
+  s.summary = %q{A gem to easily create prefork servers.}
+  s.test_files = [
+    "spec/integration/logging_spec.rb",
+     "spec/integration/timeout_spec.rb",
+     "spec/integration/worker_control_spec.rb",
+     "spec/spec_helper.rb",
+     "spec/support/integration.rb",
+     "examples/amqp.rb",
+     "examples/amqp_client.rb",
+     "examples/ping_pong.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<rspec>, [">= 1.3.0"])
+      s.add_development_dependency(%q<filetesthelper>, [">= 0.10.1"])
+    else
+      s.add_dependency(%q<rspec>, [">= 1.3.0"])
+      s.add_dependency(%q<filetesthelper>, [">= 0.10.1"])
+    end
+  else
+    s.add_dependency(%q<rspec>, [">= 1.3.0"])
+    s.add_dependency(%q<filetesthelper>, [">= 0.10.1"])
+  end
+end
+

data/spec/integration/logging_spec.rb

@@ -0,0 +1,67 @@
+require 'spec_helper'
+
+include FileTestHelper
+describe "Preforker" do
+  sandboxed_it "should redirect stdout" do
+    run_preforker <<-CODE
+      Preforker.new(:workers => 1, :stdout_path => 'test.log') do
+        puts "hello"
+        sleep 0.3 while master.wants_me_alive?
+      end.start
+    CODE
+
+    term_server
+    File.read("test.log").should == "hello\n"
+  end
+
+  sandboxed_it "should redirect stdout to the null device" do
+    run_preforker <<-CODE
+      Preforker.new(:workers => 1, :stdout_path => '/dev/null') do
+        puts "hello"
+        sleep 0.3 while master.wants_me_alive?
+      end.start
+    CODE
+
+    term_server
+    File.exists?("test.log").should == false
+  end
+
+  sandboxed_it "should redirect stderr" do
+    run_preforker <<-CODE
+      Preforker.new(:workers => 1, :stderr_path => 'test.log') do |master|
+        warn "hello"
+        sleep 0.3 while master.wants_me_alive?
+      end.start
+    CODE
+
+    term_server
+    File.read("test.log").should == "hello\n"
+  end
+
+  sandboxed_it "should have a default logger file" do
+    run_preforker <<-CODE
+      Preforker.new(:workers => 1) do
+        sleep 0.3 while master.wants_me_alive?
+      end.start
+    CODE
+
+    term_server
+    File.read("preforker.log").should =~ /Logfile created on/
+  end
+
+  sandboxed_it "should be possible to use the same file for logging, stdout and stderr" do
+    run_preforker <<-CODE
+      Preforker.new(:workers => 1, :stdout_path => 'test.log', :stderr_path => 'test.log', :logger => Logger.new('test.log')) do
+        puts "stdout string"
+        warn "stderr string"
+        sleep 0.3 while master.wants_me_alive?
+      end.start
+    CODE
+
+    term_server
+    log = File.read("test.log")
+    log.should =~ /stdout string/
+    log.should =~ /stderr string/
+    log.should =~ /Logfile created on/
+  end
+end

data/spec/integration/timeout_spec.rb

@@ -0,0 +1,32 @@
+require 'spec_helper'
+
+include FileTestHelper
+describe "Preforker" do
+  sandboxed_it "should not respawn workers when there's not a timeout" do
+    run_preforker <<-CODE
+      Preforker.new(:timeout => 2, :workers => 1) do |master|
+        sleep 0.1 while master.wants_me_alive?
+      end.start
+    CODE
+
+    sleep 1
+    term_server
+    log = File.read("preforker.log")
+    log.should_not =~ /ERROR.*timeout/
+    log.scan(/Child.*Created/).size.should == 1
+  end
+
+  sandboxed_it "should respawn workers when there's a timeout (master checks once a second max)" do
+    run_preforker <<-CODE
+      Preforker.new(:timeout => 1, :workers => 1) do
+        sleep 1000
+      end.start
+    CODE
+
+    sleep 1
+    term_server
+    log = File.read("preforker.log")
+    log.should =~ /ERROR.*timeout/
+    log.scan(/Child.*Created/).size.should > 1
+  end
+end

data/spec/integration/worker_control_spec.rb

@@ -0,0 +1,96 @@
+require 'spec_helper'
+
+include FileTestHelper
+describe "Preforker" do
+  sandboxed_it "should quit gracefully" do
+    run_preforker <<-CODE
+      Preforker.new(:workers => 1) do |master|
+        sleep 0.1 while master.wants_me_alive?
+
+        master.logger.info("Main loop ended. Dying")
+      end.start
+    CODE
+
+    quit_server
+    log = File.read("preforker.log")
+    log.should =~ /Main loop ended. Dying/
+  end
+
+  sandboxed_it "shouldn't quit gracefully on term signal" do
+    run_preforker <<-CODE
+      Preforker.new(:workers => 1) do |master|
+        sleep 0.1 while master.wants_me_alive?
+
+        master.logger.info("Main loop ended. Dying")
+      end.start
+    CODE
+
+    term_server
+    log = File.read("preforker.log")
+    log.should_not =~ /Main loop ended. Dying/
+  end
+
+  sandboxed_it "shouldn't quit gracefully on int signal" do
+    run_preforker <<-CODE
+      Preforker.new(:workers => 1) do |master|
+        sleep 0.1 while master.wants_me_alive?
+
+        master.logger.info("Main loop ended. Dying")
+      end.start
+    CODE
+
+    int_server
+    log = File.read("preforker.log")
+    log.should_not =~ /Main loop ended. Dying/
+  end
+
+  sandboxed_it "should add a worker on ttin" do
+    run_preforker <<-CODE
+      Preforker.new(:workers => 2) do |master|
+        sleep 0.1 while master.wants_me_alive?
+      end.start
+    CODE
+
+    signal_server(:TTIN)
+    sleep 0.5
+    log = File.read("preforker.log")
+    log.scan(/Child.*Created/).size.should == 3
+  end
+
+  sandboxed_it "should remove a worker on ttou" do
+    run_preforker <<-CODE
+      Preforker.new(:workers => 2) do |master|
+        sleep 0.1 while master.wants_me_alive?
+      end.start
+    CODE
+
+    signal_server(:TTOU)
+    sleep 0.5
+    log = File.read("preforker.log")
+    log.scan(/Child.*Exiting/).size.should == 1
+  end
+
+  sandboxed_it "should remove all workers on winch" do
+    run_preforker <<-CODE
+      Preforker.new(:workers => 2) do |master|
+        sleep 0.1 while master.wants_me_alive?
+      end.start
+    CODE
+
+    signal_server(:WINCH)
+    sleep 0.5
+    log = File.read("preforker.log")
+    log.scan(/Child.*Exiting/).size.should == 2
+  end
+
+  sandboxed_it "should keep creating workers when they die" do
+    run_preforker <<-CODE
+      Preforker.new(:workers => 1, :timeout => 0.2) do |master|
+      end.start
+    CODE
+
+    sleep 0.3
+    log = File.read("preforker.log")
+    log.scan(/Child.*Created/).size.should > 1
+  end
+end

data/spec/spec.opts

@@ -0,0 +1 @@
+--color

data/spec/spec_helper.rb

@@ -0,0 +1,12 @@
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+require 'preforker'
+require 'spec'
+require 'spec/autorun'
+require 'rubygems'
+require 'filetesthelper'
+Dir[File.dirname(__FILE__) + "/support/**/*.rb"].each {|f| require f}
+
+include Integration
+Spec::Runner.configure do |config|
+end

data/spec/support/integration.rb

@@ -0,0 +1,52 @@
+module Integration
+  PREFORKER_LIB_PATH = File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'lib'))
+  def run_preforker(code)
+    File.open("test.rb", 'w') do |f|
+      f.write <<-TOP
+        $LOAD_PATH.unshift('#{PREFORKER_LIB_PATH}')
+        require 'preforker'
+      TOP
+      f.write(code)
+    end
+
+    system("ruby test.rb")
+  end
+
+  def sandboxed_it(desc)
+    it desc do
+      with_files do
+        yield
+        term_server
+      end
+    end
+  end
+
+  def signal_server(signal)
+    Dir['*.pid'].each do |pid_file_path|
+      Process.kill(signal, File.read(pid_file_path).to_i)
+    end
+  end
+
+  def term_server
+    signal_server(:TERM)
+    wait_till_server_ends
+  end
+
+  def quit_server
+    signal_server(:QUIT)
+    wait_till_server_ends
+  end
+
+  def int_server
+    signal_server(:INT)
+    wait_till_server_ends
+  end
+
+  def wait_till_server_ends
+    Dir['*.pid'].each do |pid_file_path|
+      while File.exists?(pid_file_path) do
+        sleep 0.2
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,117 @@
+--- !ruby/object:Gem::Specification 
+name: preforker
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Daniel Cadenas
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-04-02 00:00:00 -03:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 1
+        - 3
+        - 0
+        version: 1.3.0
+  type: :development
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: filetesthelper
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        - 10
+        - 1
+        version: 0.10.1
+  type: :development
+  version_requirements: *id002
+description: A gem to easily create prefork servers.
+email: dcadenas@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .gitignore
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- examples/amqp.rb
+- examples/amqp_client.rb
+- examples/ping_pong.rb
+- lib/preforker.rb
+- lib/preforker/pid_manager.rb
+- lib/preforker/signal_processor.rb
+- lib/preforker/util.rb
+- lib/preforker/worker.rb
+- preforker.gemspec
+- spec/integration/logging_spec.rb
+- spec/integration/timeout_spec.rb
+- spec/integration/worker_control_spec.rb
+- spec/spec.opts
+- spec/spec_helper.rb
+- spec/support/integration.rb
+has_rdoc: true
+homepage: http://github.com/dcadenas/preforker
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.6
+signing_key: 
+specification_version: 3
+summary: A gem to easily create prefork servers.
+test_files: 
+- spec/integration/logging_spec.rb
+- spec/integration/timeout_spec.rb
+- spec/integration/worker_control_spec.rb
+- spec/spec_helper.rb
+- spec/support/integration.rb
+- examples/amqp.rb
+- examples/amqp_client.rb
+- examples/ping_pong.rb