keep_running 0.0.4

data/bin/keep_running

@@ -0,0 +1,7 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH << File.join( File.dirname(__FILE__), '../lib')
+
+require 'keep_running'
+
+KeepRunning.load_runner File.expand_path(ARGV[0])

data/lib/keep_running.rb

@@ -0,0 +1,115 @@
+require 'fileutils'
+require 'pony'
+
+class KeepRunning
+  class << self
+    def load_runner(runner)
+      load @runner_location = runner
+    end
+    
+    def runner_location
+      @runner_location ||= File.expand_path($0)
+    end
+    
+    def process_argv
+      ARGV.shift if File.basename($0) == 'keep_running'
+    end
+    
+    def run(&block)
+      runner = new
+      process_argv
+      runner.instance_eval &block
+      runner.run
+    end
+  end
+  
+  def pidfile(p)            ; @pidfile = p            ; end
+  def email(e)              ; @email_opts = e         ; end
+  def restart_threshold(t)  ; @restart_threshold = t  ; end
+  def restart_delay(d)      ; @restart_delay = d      ; end
+  def process(p)
+    @process = File.expand_path(File.join(File.dirname(self.class.runner_location), p))
+  end
+  
+  def run
+    write_pidfile
+    
+    output = nil
+    loop do
+      output = spawn { start_process }              # load the given script
+      send_mail subject(@process), output           # send the email
+      throttle_restart                              # don't restart too fast
+    end
+  end
+  
+  private
+    
+    def start_process
+      if File.extname(@process) == '.rb'
+        load @process
+      else
+        puts 'only spawning of ruby processes supported'
+      end
+    end
+    
+    def add_exit_hook(pid)
+      at_exit do
+        Process.kill 'QUIT', pid rescue nil         # ensure the child exits
+        sleep 1                                     # give it some time
+        Process.kill 'KILL', pid rescue nil         # really ensure the child exits
+      end
+    end
+    
+    def send_mail(subject, body)
+      Pony.mail @email_opts.update(:subject => subject(@process), :body => body)
+    end
+    
+    def write_pidfile
+      puts "wrote PID-file #{@pidfile}"
+      File.open(@pidfile, 'w'){|file| file << Process.pid }
+      at_exit{
+        if File.new(@pidfile, "r").gets.to_i == Process.pid
+          FileUtils.rm @pidfile rescue nil
+        end
+      }
+    end
+    
+    def spawn
+      readme, writeme = IO.pipe                     # open an pipe for the child to write in, the parent to read
+      pid = fork{ |variable|                        # fork the process
+        $stdout.reopen writeme                      # reopen the child end of the pipe
+        $stderr.reopen writeme                      # reopen the child end of the pipe
+        readme.close                                # close the parent end of the pipe in the child
+        yield                                       # YIELD the given block
+      }
+      writeme.close                                 # close the child end of the pipe in the parent
+      puts "forked child as PID #{pid}"
+      
+      add_exit_hook pid
+      
+      output = []
+      readme.each{ |line|
+        output = output[-1000,1000] || [] # truncate the output array
+        puts line
+        output << line            # collect the childs stdout
+      }
+      
+      Process.waitpid pid, 0                        # wait for the child to exit
+      return output.join
+    end
+    
+    def throttle_restart
+      if @last_exit && Time.now - @last_exit < @restart_threshold
+        puts "Last restart was less than #{@restart_threshold} seconds ago: #{@last_exit}."
+        puts "Sleeping some #{@restart_delay} seconds..."
+        sleep @restart_delay
+        puts "Let's try again."
+      end
+      @last_exit = Time.now
+    end
+    
+    def subject(script)
+      "#{script} ended (PID #{Process.pid} on #{`hostname`.strip})"
+    end
+  
+end

data/lib/keep_running/version.rb

@@ -0,0 +1,3 @@
+class KeepRunning
+  VERSION = '0.0.4'
+end

metadata

@@ -0,0 +1,66 @@
+--- !ruby/object:Gem::Specification 
+name: keep_running
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.0.4
+platform: ruby
+authors: 
+- Niko Dittmann
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-12-13 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: pony
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+description: a script to restart other scripts whenever they crash, stop, get killed. It sends emails, when restart occurs.
+email: mail+git@niko-dittmann.com
+executables: 
+- keep_running
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- lib/keep_running/version.rb
+- lib/keep_running.rb
+- bin/keep_running
+homepage: http://github.com/niko/keep_running
+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.11
+signing_key: 
+specification_version: 3
+summary: a script to restart other scripts whenever they crash, stop, get killed.
+test_files: []
+