daemonizr 0.1.0

data/CHANGELOG

@@ -0,0 +1 @@
+v0.1.0. Initial release

data/LICENSE

@@ -0,0 +1,21 @@
+The MIT License
+
+Copyright (c) 2009 Carl Mercier
+
+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/Manifest

@@ -0,0 +1,7 @@
+CHANGELOG
+LICENSE
+Manifest
+README
+Rakefile
+daemonizr.rb
+lib/daemonizr.rb

data/README

@@ -0,0 +1,7 @@
+= Daemonizr - Process forking and monitoring for mere mortals
+
+An example is worth a thousand words...
+
+d = Daemonizr.new("Daemonizr")
+d.start_cluster("MyServer", 3, lambda {loop{ sleep 2; File.open("/tmp/daemonizr.log", "a") { |f| f.puts "#{Process.pid}: #{Time.now}" }  }})
+d.monitor_cluster! # Will hang here until the process is terminated with SIGTERM (kill 5). If a fork dies, it will be restarted.

data/Rakefile

@@ -0,0 +1,14 @@
+require 'rubygems'  
+require 'rake'  
+require 'echoe'  
+  
+Echoe.new('daemonizr') do |p|  
+  p.description     = "Process forking and monitoring for mere mortals"
+  p.url             = "http://github.com/cmer/daemonizr"  
+  p.author          = "Carl Mercier"  
+  p.email           = "carl@carlmercier.com"  
+  p.ignore_pattern  = ["tmp/*", "script/*", "pkg/*"]  
+  p.development_dependencies = []  
+end  
+  
+Dir["#{File.dirname(__FILE__)}/tasks/*.rake"].sort.each { |ext| load ext }

data/daemonizr.gemspec

@@ -0,0 +1,30 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{daemonizr}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 1.2") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Carl Mercier"]
+  s.date = %q{2009-11-05}
+  s.description = %q{Process forking and monitoring for mere mortals}
+  s.email = %q{carl@carlmercier.com}
+  s.extra_rdoc_files = ["CHANGELOG", "LICENSE", "README", "lib/daemonizr.rb"]
+  s.files = ["CHANGELOG", "LICENSE", "Manifest", "README", "Rakefile", "daemonizr.rb", "lib/daemonizr.rb", "daemonizr.gemspec"]
+  s.homepage = %q{http://github.com/cmer/daemonizr}
+  s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Daemonizr", "--main", "README"]
+  s.require_paths = ["lib"]
+  s.rubyforge_project = %q{daemonizr}
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{Process forking and monitoring for mere mortals}
+
+  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
+    else
+    end
+  else
+  end
+end

data/daemonizr.rb

@@ -0,0 +1 @@
+require 'lib/daemonizr'

data/lib/daemonizr.rb

@@ -0,0 +1,159 @@
+require 'rubygems'
+require 'daemons'
+require 'logger'
+
+class Daemonizr
+  CLUSTERS  = {}
+
+  def initialize(instance_name, options = {})
+    @name = instance_name
+    @pid_path = options[:pid_path]
+    @log_to_stdout = true
+    
+    if options[:log_file].is_a?(String)
+      raise ArgumentError.new("Invalid log_level value.") unless [:debug, :info, :warn, :error, :fatal, nil].include?(options[:log_level])
+      @logger = Logger.new(options[:log_file]) 
+      @logger.level = eval("Logger::#{(options[:log_level] || :info).to_s.upcase}")
+    end
+  end
+  
+  def start_cluster(name, count, prok)
+    log :info, "Starting new cluster named '#{name}' with #{count} processes..."
+    raise "Can't start another cluster with the name #{name}. Already running." if cluster_exists?(name)
+    CLUSTERS[name] = { :count => count, :proc => prok, :processes => {} }
+    (1..count).each do |id|
+      log :debug, "Asking to start process #{id}/#{count} for cluster '#{name}'..."
+      start_daemon(name, id)
+    end
+    true
+  end
+
+  def stop_all_clusters
+    log :debug, "Stopping all clusters..."
+    CLUSTERS.each_key do |k|
+      stop_cluster(k)
+    end
+  end
+
+  def monitor_cluster!(interval = 5)
+    log :info, "Monitoring of clusters starting..."
+    while "ruby" > "java" do
+      CLUSTERS.each_key do |cluster_name|
+        CLUSTERS[cluster_name][:processes].each_key do |process_id|
+          unless CLUSTERS[cluster_name][:processes][process_id].running?
+            log :warn, "Process ##{process_id} of cluster '#{cluster_name}' is not running. Restarting..."
+            start_daemon(cluster_name, process_id)
+          end
+        end
+      end
+      
+      begin
+        sleep interval
+      rescue Interrupt
+        # Process was killed. Stop clusters.
+        puts "\n"
+        log :info, "Received SIGTERM. Stopping all clusters and stop monitoring."
+        stop_all_clusters
+        log :info, "Monitoring stopped."
+        return true
+      end
+    end
+  end
+  
+
+  protected
+  def start_daemon(name, id)
+    if CLUSTERS[name][:processes][id] && CLUSTERS[name][:processes][id].running?
+      log :error, "Process ##{id} for cluster '#{name}' is already running. Can't start it again."
+      raise ClusterAlreadyRunningException
+    else
+      # remove_stale_pid_file(name, id)
+      process_title = "#{@name}: #{name} #{id}/#{CLUSTERS[name][:count]}"
+      log :info, "Starting process '#{process_title}'..."
+      prok = CLUSTERS[name][:proc]
+      
+      CLUSTERS[name][:processes][id] = Daemons.call(:multiple=>true) do
+        # TODO: figure out why it gets truncated
+        $PROGRAM_NAME = process_title
+        # write_pid_file(name, id)
+        prok.call
+      end
+      
+      log :info, "Process '#{process_title}' started with PID #{CLUSTERS[name][:processes][id].pid.pid}."
+      return CLUSTERS[name][:processes][id]
+    end
+  end
+
+  def stop_cluster(name)
+    log :info, "Stopping cluster '#{name}'..."
+    raise ClusterDoesNotExistException unless cluster_exists?(name)
+    
+    CLUSTERS[name][:processes].each_key do |k|
+      log :debug, "Stopping process ##{k} of cluster '#{name}'..."
+      CLUSTERS[name][:processes][k].stop
+      # remove_stale_pid_file(name, k)
+    end
+    
+    CLUSTERS.delete(name); true
+  end
+
+  def cluster_exists?(name)
+    return CLUSTERS.keys.include?(name)
+  end
+  
+  def log(severity, msg)
+    msg = format_log_message(msg)
+    @logger.send(severity, msg) if @logger
+    puts msg if @log_to_stdout || !@logger
+  end
+  
+  def format_log_message(msg)
+    msg
+  end
+  
+  # def pid_file(cluster, id)
+  #   if @pid_path
+  #     f = @pid_path.strip
+  #     f += File::SEPARATOR unless f[-1,1] == File::SEPARATOR
+  #     f += "#{@name}-#{cluster}-#{id}.pid"
+  #   else
+  #     nil
+  #   end
+  # end
+  # 
+  # def write_pid_file(name, id)
+  #   pf = pid_file(name, id)
+  #   if pf
+  #     log :debug, "Writing PID to #{pf}"
+  #     FileUtils.mkdir_p File.dirname(pf)
+  #     open(pf,"w") { |f| f.write(Process.pid) }
+  #     File.chmod(0644, pf)
+  #   end
+  # end
+  # 
+  # def remove_stale_pid_file(name, id)
+  #   pf = pid_file(name, id)
+  #   if File.exist?(pf)
+  #     pid = File.open(pf).read
+  #     File.delete(pf) unless pid && process_running?(pid)
+  #   end
+  # end
+  # 
+  # def process_running?(pid)
+  #   begin
+  #     Process.kill(0, pid.to_i)
+  #     true
+  #   rescue Errno::ESRCH
+  #     false
+  #   end
+  # end
+  
+  class ClusterAlreadyRunningException < Exception; end
+  class ClusterDoesNotExistException < Exception; end
+end
+
+
+
+# d = Daemonizr.new("Daemonizr", :pid_path => "/tmp/")
+# d.start_cluster("MyServer", 3, lambda {loop{ sleep 2; File.open("/tmp/daemonizr.log", "a") { |f| f.puts "#{Process.pid}: #{Time.now}" }  }})
+# d.monitor_cluster! # Will hang here until the process is terminated with SIGTERM (kill 5).

metadata

@@ -0,0 +1,70 @@
+--- !ruby/object:Gem::Specification 
+name: daemonizr
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Carl Mercier
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-11-05 00:00:00 -05:00
+default_executable: 
+dependencies: []
+
+description: Process forking and monitoring for mere mortals
+email: carl@carlmercier.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- CHANGELOG
+- LICENSE
+- README
+- lib/daemonizr.rb
+files: 
+- CHANGELOG
+- LICENSE
+- Manifest
+- README
+- Rakefile
+- daemonizr.rb
+- lib/daemonizr.rb
+- daemonizr.gemspec
+has_rdoc: true
+homepage: http://github.com/cmer/daemonizr
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --line-numbers
+- --inline-source
+- --title
+- Daemonizr
+- --main
+- README
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "1.2"
+  version: 
+requirements: []
+
+rubyforge_project: daemonizr
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Process forking and monitoring for mere mortals
+test_files: []
+