hamster-the-process-watcher 1.1.0

data/.document

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

data/.rvmrc

@@ -0,0 +1 @@
+rvm use ree@hampster --create

data/Gemfile

@@ -0,0 +1,5 @@
+# A sample Gemfile
+source "http://rubygems.org"
+gem 'wrong'
+gem 'jeweler'
+gem 'daemons'

data/LICENSE

@@ -0,0 +1,15 @@
+
+    Copyright (C) 2009,2010 Curtis Schofield
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.

data/README.rdoc

@@ -0,0 +1,21 @@
+= Hampster
+
+Simple process monitor - 
+ Used to check if a process is running - if it insn't execute some ruby code
+
+see ./bin/hampster --help for all the arguments that it takes
+
+= Example
+
+hampster --callback "%x{echo "Rails is Dead" | mail -s rails_is_dead me@ram9.cc}" --watch 'Rails:' --callback-message 'emailing you some fyi' --delay 1
+
+samples.  
+
+ Put this in a cronjob to run every minute and you have a very simple, robust way to detect missing jobs
+
+ This is not a replacement for more robust excellent froody things like Daemontools - but it is a simple way to
+ make sure a important background job is running in userland - without muking about with root
+
+= Fin
+
+Copyright (c) 2011 Curtis Schofield. See LICENSE for details.

data/Rakefile

@@ -0,0 +1,58 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "hamster-the-process-watcher"
+    gem.summary = %Q{Hamster : The process runner}
+    gem.description = %Q{Dedicated to hamsters running in wheels everywhere}
+    gem.email = "github.com@robotarmyma.de"
+    gem.homepage = "http://github.com/robotarmy/hamster"
+    gem.authors = ["Curtis Schofield"]
+    gem.add_development_dependency "bundler", ">= 0"
+    # 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 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/test_*.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/test_*.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+task :check_dependencies do
+  puts '..bundler'
+  `bundle check || bundle install`
+end
+
+task :test => :check_dependencies
+
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  version = File.exist?('VERSION') ? File.read('VERSION') : ""
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "weasel #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+1.1.0

data/bin/hamster

@@ -0,0 +1,39 @@
+#!/usr/bin/env ruby -rubygems
+$:.push File.join(File.dirname(__FILE__),'..','lib')
+require 'optparse'
+require 'hamster'
+options = {}
+OptionParser.new do |opts|
+  opts.banner = "Usage: #{__FILE__} [options]"
+   opts.on( '-h', '--help', 'Display this screen' ) do
+     puts opts
+     exit
+   end
+
+  opts.on("--command shellcode",String, "shell code to execute if process missing") do |v|
+    options[:command] = v
+  end
+
+  opts.on("--callback rubycode",String, "ruby code to execute if process missing") do |v|
+    options[:callback] = lambda {  # may be evil
+                          eval(v)
+                        }
+  end
+  opts.on("--callback_message string",String, "message to write when callback fired") do |v|
+    options[:callback_message] = v
+  end
+
+  opts.on("--watch S",String, "process identifying string") do |v|
+    options[:watch] = v
+  end
+
+  opts.on("--num_cycles N", "total executions before exit") do |v|
+    options[:num_cycles] = v.to_i
+  end
+
+  opts.on("--delay F",Float, "delay between cycles") do |v|
+    options[:delay] = v
+  end
+
+end.parse!
+ProcessWatch.new(options).cycle

data/bin/test_daemon

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+loop do
+  sleep 100
+end

data/hamster-the-process-watcher.gemspec

@@ -0,0 +1,62 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{hamster-the-process-watcher}
+  s.version = "1.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Curtis Schofield"]
+  s.date = %q{2011-07-07}
+  s.description = %q{Dedicated to hamsters running in wheels everywhere}
+  s.email = %q{github.com@robotarmyma.de}
+  s.executables = ["test_daemon", "hamster"]
+  s.extra_rdoc_files = [
+    "LICENSE",
+    "README.rdoc"
+  ]
+  s.files = [
+    ".document",
+    ".rvmrc",
+    "Gemfile",
+    "LICENSE",
+    "README.rdoc",
+    "Rakefile",
+    "VERSION",
+    "bin/hamster",
+    "bin/test_daemon",
+    "hamster-the-process-watcher.gemspec",
+    "lib/hamster.rb",
+    "lib/process_watch.rb",
+    "test/helper.rb",
+    "test/test_process_watch.rb"
+  ]
+  s.homepage = %q{http://github.com/robotarmy/hamster}
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.6.1}
+  s.summary = %q{Hamster : The process runner}
+
+  if s.respond_to? :specification_version then
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
+      s.add_runtime_dependency(%q<wrong>, [">= 0"])
+      s.add_runtime_dependency(%q<jeweler>, [">= 0"])
+      s.add_runtime_dependency(%q<daemons>, [">= 0"])
+      s.add_development_dependency(%q<bundler>, [">= 0"])
+    else
+      s.add_dependency(%q<wrong>, [">= 0"])
+      s.add_dependency(%q<jeweler>, [">= 0"])
+      s.add_dependency(%q<daemons>, [">= 0"])
+      s.add_dependency(%q<bundler>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<wrong>, [">= 0"])
+    s.add_dependency(%q<jeweler>, [">= 0"])
+    s.add_dependency(%q<daemons>, [">= 0"])
+    s.add_dependency(%q<bundler>, [">= 0"])
+  end
+end
+

data/lib/hamster.rb

@@ -0,0 +1,2 @@
+require 'daemons'
+require 'process_watch'

data/lib/process_watch.rb

@@ -0,0 +1,56 @@
+class ProcessWatch
+  attr_accessor :watch_string,
+    :delay,
+    :num_cycles,
+    :command,
+    :callback,
+    :callback_message
+
+  def initialize(options)
+    self.command          = options[:command]              || nil
+    self.callback         = options[:callback]             || 
+                             lambda { p 'default callback does nothing'}
+    self.callback_message = (options[:callback_message]    || 
+                             "Callback Triggered").strip
+    self.watch_string     = (options[:watch]               || 
+                             "this poem is a pomme").strip
+    self.delay            = options[:delay]                || 60
+    self.num_cycles       = options[:num_cycles]           || 3
+  end
+
+  class << self 
+    
+    def find(watch_string,ps_args = "")
+      cmd = %%ps -x#{ps_args} |grep '#{watch_string}' |grep -v grep | grep -v 'watch #{watch_string}' | awk '{print $1}'%
+      pids = %x{#{cmd}}.split()
+    end
+  end
+
+  public
+  def cycle
+    self.num_cycles.times do 
+      _check
+      sleep self.delay
+    end
+  end
+  private
+  def _trigger_callback
+      _fork
+    ensure
+      self.callback.call
+      puts self.callback_message
+    end
+  def _check
+    if self.class.find(self.watch_string).empty?
+      _trigger_callback
+    end
+  end
+  def _fork
+    Daemonize.call_as_daemon(lambda {
+      p ENV['PWD']
+      Dir.chdir(ENV['PWD'])
+      ::Process.exec(self.command)
+    },"hamster.daemon.log",self.command)
+  end
+end
+

data/test/helper.rb

@@ -0,0 +1,11 @@
+require 'rubygems'
+require 'test/unit'
+require "bundler/setup"
+require 'wrong/adapters/test_unit'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'hamster'
+
+class Test::Unit::TestCase
+end

data/test/test_process_watch.rb

@@ -0,0 +1,92 @@
+require 'helper'
+class TestProcessWatch < Test::Unit::TestCase
+  def test_find_running_daemon
+    cmd = %%./bin/test_daemon ID=1%
+    Process.detach(d_pid = fork {
+      Process.exec cmd
+    })
+    assert {
+      ProcessWatch.find('test_daemon ID=1').size == 1
+    }
+    %x{kill -9 #{d_pid}}
+  end
+  def test_pid_of_new_process_is_sibling
+    cmd = %%ruby ./bin/hamster --watch 'test_daemon ID=1' --command "./bin/test_daemon ID=1" --callback_message o_o__poot --delay 4 --num_cycles 45 %
+    Process.detach(h_pid = fork { 
+      Process.exec cmd 
+    })
+    sleep 3 # let it start
+    pid = ProcessWatch.find('test_daemon ID=1').first
+
+    assert {
+      lookup_parent_pid_via_ps('ruby ./bin/test_daemon ID=1') ==  lookup_parent_pid_via_ps('o_o__poot')
+    }
+    # clean up hamster and the test_daemon
+    %x{kill -9 #{pid}}
+    %x{kill -9 #{h_pid}}
+  end
+  def test_running_already_exit_without_action
+    cmd = %%./bin/test_daemon ID=1%
+    Process.detach(d_pid = fork { 
+      Process.exec cmd
+    })
+    cmd = %%ruby ./bin/hamster --watch 'test_daemon ID=1' --command "./bin/test_daemon ID=1" --callback_message poot --delay 1 --num_cycles 2 %
+    Process.detach(h_pid = fork { 
+      %x{#{cmd}}
+    })
+    pid, status = Process.waitpid2(h_pid)
+    assert {
+      status.exitstatus == 0 
+    }
+    %x{kill -9 #{d_pid}}
+  end
+
+  def test_watches_self
+    callback_called = false
+    pw = ProcessWatch.new(:watch => "nothing", :delay => 1, :cycle => 1,
+                          :callback => lambda {callback_called = true})
+    pw.cycle
+    assert { 
+      callback_called
+    }
+  end
+  def test_watches_but_not_the_watch_string_exits
+    cmd = %%./bin/hamster --watch hamster --callback "puts 'toop'" --callback_message poot --delay 1 --num_cycles 2 %
+    out = %x{#{cmd}}
+    assert {
+        out =~ /poot/ && out =~ /toop/
+    }
+  end
+  def test_runs_test_daemon_command_if_not_present
+    cmd = %%ruby ./bin/hamster --watch 'test_daemon ID=1' --command "./bin/test_daemon ID=1" --callback_message poot --delay 1 --num_cycles 2 %
+    Process.detach(pid = fork { 
+      Process.exec cmd
+    })
+    sleep 3 # let it start
+    assert {
+      ProcessWatch.find('test_daemon ID=1').size == 1
+    }
+    %x{kill -9 #{ProcessWatch.find('test_daemon ID=1').first}}
+  end
+
+  def test_daemon_keeps_running
+    cmd = %%ruby ./bin/hamster --watch 'test_daemon ID=1' --command "./bin/test_daemon ID=1" --callback_message poot --delay 5 --num_cycles 10 %
+    Process.detach(pid = fork { 
+      Process.exec cmd
+    })
+    sleep 3 # let it start
+    %x{kill -9 #{pid}}
+    #child should live on if hampster is killed
+    sleep 3 # wait..for it
+    assert {
+      ProcessWatch.find('test_daemon ID=1').size == 1
+    }
+    %x{kill -9 #{ProcessWatch.find('test_daemon ID=1').first}}
+  end
+
+  def lookup_parent_pid_via_ps(match)
+    ProcessWatch.find(match,"f").first
+  end
+end
+
+

metadata

@@ -0,0 +1,137 @@
+--- !ruby/object:Gem::Specification 
+name: hamster-the-process-watcher
+version: !ruby/object:Gem::Version 
+  hash: 19
+  prerelease: 
+  segments: 
+  - 1
+  - 1
+  - 0
+  version: 1.1.0
+platform: ruby
+authors: 
+- Curtis Schofield
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-07-07 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  version_requirements: *id001
+  name: wrong
+  prerelease: false
+  type: :runtime
+- !ruby/object:Gem::Dependency 
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  version_requirements: *id002
+  name: jeweler
+  prerelease: false
+  type: :runtime
+- !ruby/object:Gem::Dependency 
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  version_requirements: *id003
+  name: daemons
+  prerelease: false
+  type: :runtime
+- !ruby/object:Gem::Dependency 
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  version_requirements: *id004
+  name: bundler
+  prerelease: false
+  type: :development
+description: Dedicated to hamsters running in wheels everywhere
+email: github.com@robotarmyma.de
+executables: 
+- test_daemon
+- hamster
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.rdoc
+files: 
+- .document
+- .rvmrc
+- Gemfile
+- LICENSE
+- README.rdoc
+- Rakefile
+- VERSION
+- bin/hamster
+- bin/test_daemon
+- hamster-the-process-watcher.gemspec
+- lib/hamster.rb
+- lib/process_watch.rb
+- test/helper.rb
+- test/test_process_watch.rb
+has_rdoc: true
+homepage: http://github.com/robotarmy/hamster
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.6.1
+signing_key: 
+specification_version: 3
+summary: "Hamster : The process runner"
+test_files: []
+