forking 0.1

data/.gitignore

@@ -0,0 +1,8 @@
+log
+TODO
+locks
+*.gem
+*.log
+*.pid
+Gemfile.lock
+.bundle

data/Gemfile

@@ -0,0 +1,3 @@
+source :rubygems
+
+gemspec

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2012 Makarchev K
+
+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.md

@@ -0,0 +1,56 @@
+Forking
+=======
+
+Simple processes forking, and restarting. Master process starts as daemon.
+
+
+    $ gem install forking
+
+
+Example 1.rb (run 5 child processes and capturing logs):
+
+```ruby
+#!/usr/bin/env ruby
+require 'rubygems'
+require 'forking'
+
+f = Forking.new(:name => 'test', :working_dir => File.dirname(__FILE__),
+    :log_file => "spawner.log", :pid_file => "spawner.pid", :sync_log => true)
+
+f.before_fork do
+  puts "load env"
+end
+
+f.after_fork do
+  puts "restart connects"
+end
+
+2.times do |i|
+  f.spawn(:name => "test1", :log_file => "test1.log") do
+    loop do
+      puts "test1 #{i}"
+      sleep 1
+    end
+  end
+end
+
+3.times do |i|
+  f.spawn(:log_file => "test2.log", :sync_log => true) do
+    exec 'ruby', '2.rb', '--test'
+  end
+end
+
+f.run!
+```
+
+Usage:
+
+    $ ./1.rb start
+    $ ./1.rb status
+    $ ./1.rb stop
+    $ ./1.rb restart
+
+Respawn childs:
+
+    $ kill -HUP master_pid 
+    

data/Rakefile

@@ -0,0 +1,8 @@
+require 'rubygems'
+require 'bundler/setup'
+require 'bundler'
+Bundler::GemHelper.install_tasks
+
+require 'rspec/core/rake_task'
+task :default => :spec
+RSpec::Core::RakeTask.new(:spec)

data/example/1.rb

@@ -0,0 +1,36 @@
+#!/usr/bin/env ruby
+require 'rubygems'
+require 'bundler/setup'
+require 'forking.rb'
+
+f = Forking.new(:name => 'test',
+    :working_dir => File.dirname(__FILE__),
+    :log_file => "spawner.log", 
+    :pid_file => "spawner.pid",
+    :sync_log => true)
+
+f.before_fork do
+  puts "before fork"
+end   
+ 
+f.after_fork do
+  puts "after fork"
+end
+
+2.times do |i|
+  f.spawn(:name => "test1", :log_file => "test1.log") do
+    $0 = "test1 #{i}"
+    loop do
+      puts "test1 #{i}"
+      sleep 1
+    end
+  end
+end
+
+3.times do |i|
+  f.spawn(:log_file => "test2.log", :sync_log => true) do
+    exec 'ruby', '2.rb', '--test', 'some_value'
+  end
+end
+
+f.run!

data/example/2.rb

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+
+$0 = "test2"
+
+loop do
+
+  p ARGV
+  sleep 1
+
+end

data/forking.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+require File.dirname(__FILE__) + "/lib/version"
+
+Gem::Specification.new do |s|
+  s.name = %q{forking}
+  s.version = Forking::VERSION
+
+  s.authors = ["Makarchev Konstantin"]
+ 
+  s.description = %q{Simple processes forking, and restarting. Master process starts as daemon.}
+  s.summary = %q{Simple processes forking, and restarting. Master process starts as daemon.}
+
+  s.email = %q{kostya27@gmail.com}
+  s.homepage = %q{http://github.com/kostya/forking}
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_dependency 'daemon-spawn'
+  s.add_development_dependency "rspec"
+  s.add_development_dependency "rake"
+  
+end

data/lib/daemon_spawn_base.rb

@@ -0,0 +1,69 @@
+require 'daemon_spawn' # gem
+
+class DaemonSpawnBase < DaemonSpawn::Base
+
+  def initialize(args)
+    GC.copy_on_write_friendly = true if GC.respond_to?(:copy_on_write_friendly=)
+    @args = args
+    @procs = @args[:procs]
+    @restart = true        
+
+    super
+  end
+
+  def start(_)
+    raise "empty @procs" if !@procs || @procs.empty?
+
+    # HUP is restart all childs
+    trap("HUP") do
+      stop_all
+    end
+    
+    @args[:before_fork].call if @args[:before_fork]
+
+    start_all
+
+    while @restart
+      pid = nil
+      
+      begin  
+        pid = Process.wait
+      rescue Errno::ECHILD
+      end
+      
+      process = @procs.detect{|p| p.pid == pid }
+      puts "#{process.name} had just died!"
+
+      process.start if @restart
+    end
+    
+  rescue
+    stop_all
+  end
+
+  def stop
+    @restart = false
+    stop_all
+  end
+
+private
+
+  def start_all
+    @procs.each do |process|
+      process.start
+    end
+
+    puts "started all (#{@procs.size})!"
+  end
+
+  def stop_all
+    return unless @procs
+
+    @procs.each do |process|
+      process.stop
+    end
+
+    puts "stopped all (#{@procs.size})!"
+  end
+
+end

data/lib/forking.rb

@@ -0,0 +1,36 @@
+require File.join(File.dirname(__FILE__), 'daemon_spawn_base')
+require File.join(File.dirname(__FILE__), 'mini_process')
+
+class Forking
+
+  # opts:
+  #   :name
+  #   :sync_log
+  #   :log_file
+  #   :pid_file
+  #   :working_dir *
+
+  def initialize(opts = {})
+    @opts = opts
+    @merge_opts = opts.update(:sync_log => nil) # sync_log should specify for each spawn
+    @procs = []    
+  end  
+
+  def run!
+    $0 = "#{@opts[:name]} spawner" unless @opts[:dont_touch_pl]
+    DaemonSpawnBase.spawn!(@opts.merge(:procs => @procs, :before_fork => @before_fork, :application => @opts[:name]))
+  end
+
+  def spawn(opts = {}, &block)
+    @procs << MiniProcess.new(@merge_opts.merge(opts).merge(:after_fork => @after_fork), &block)
+  end
+  
+  def before_fork(&block)
+    @before_fork = block
+  end
+  
+  def after_fork(&block)
+    @after_fork = block
+  end
+
+end

data/lib/mini_process.rb

@@ -0,0 +1,56 @@
+class MiniProcess
+  attr_accessor :pid
+
+  DEFAULT_SLEEP = 0.2
+
+  # options
+  #   :name         -
+  #   :sleep        - timeout between spawns
+  #   :log_file     -
+  #   :sync_log     - 
+  #   :working_dir  - for log
+  #   :after_fork
+
+  def initialize(opts = {}, &block)
+    @opts = opts
+    @block = block
+    @opts[:sync_log] ||= true
+
+    raise "block should be" unless block
+  end
+
+  def start    
+    GC.copy_on_write_friendly = true if GC.respond_to?(:copy_on_write_friendly=)
+
+    @pid = fork do
+      STDIN.reopen("/dev/null")
+      log_name = @opts[:log_file] || "/dev/null"
+      log_name = File.expand_path(log_name)
+      STDOUT.reopen(log_name, "a")
+      STDOUT.sync = @opts[:sync_log]
+      STDERR.reopen(log_name, "a")
+      STDERR.sync = @opts[:sync_log]
+      @opts[:after_fork].call if @opts[:after_fork]
+      @block.call
+    end
+
+    puts "start process #{name}"
+    pause
+    @pid
+  end
+
+  def stop
+    return unless @pid
+    Process.kill("KILL", @pid) rescue nil
+    puts "stop #{name}"
+  end
+
+  def pause
+    sleep (@opts[:sleep] || DEFAULT_SLEEP).to_f
+  end
+
+  def name
+    "#{@opts[:name]}[#{@pid}]"
+  end  
+
+end

data/lib/version.rb

@@ -0,0 +1,3 @@
+class Forking
+  VERSION = "0.1"
+end

metadata

@@ -0,0 +1,116 @@
+--- !ruby/object:Gem::Specification 
+name: forking
+version: !ruby/object:Gem::Version 
+  hash: 9
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  version: "0.1"
+platform: ruby
+authors: 
+- Makarchev Konstantin
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2012-06-30 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: daemon-spawn
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rspec
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: rake
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :development
+  version_requirements: *id003
+description: Simple processes forking, and restarting. Master process starts as daemon.
+email: kostya27@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- LICENSE
+- README.md
+- Rakefile
+- example/1.rb
+- example/2.rb
+- forking.gemspec
+- lib/daemon_spawn_base.rb
+- lib/forking.rb
+- lib/mini_process.rb
+- lib/version.rb
+homepage: http://github.com/kostya/forking
+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.8.24
+signing_key: 
+specification_version: 3
+summary: Simple processes forking, and restarting. Master process starts as daemon.
+test_files: []
+