sponges 0.5.0 → 0.5.0.1.pre

data/README.md

@@ -1,10 +1,5 @@
 # sponges
 
-[![Code
-Climate](https://codeclimate.com/badge.png)](https://codeclimate.com/github/AF83/sponges)
-[![Gem
-Version](https://fury-badge.herokuapp.com/rb/sponges.png)](http://badge.fury.io/rb/sponges)
-
 When I build workers, I want them to be like an army of spongebobs, always
 stressed and eager to work. `sponges` helps you build this army of sponges, to
 control them, and, well, to kill them gracefully. Making them stressed and eager

data/lib/sponges.rb

@@ -3,9 +3,7 @@ require 'boson/runner'
 require 'socket'
 require 'logger'
 require 'machine'
-require 'forwardable'
 require_relative 'sponges/configuration'
-require_relative 'sponges/handler'
 require_relative 'sponges/supervisor'
 require_relative 'sponges/runner'
 require_relative 'sponges/commander'
@@ -15,8 +13,7 @@ require_relative 'sponges/store/memory'
 require_relative 'sponges/store/redis'
 
 module Sponges
-  STOP_SIGNALS = [:INT, :QUIT, :TERM]
-  SIGNALS = STOP_SIGNALS + [:HUP, :TTIN, :TTOU, :CHLD]
+  SIGNALS = [:INT, :QUIT, :TERM]
 
   def configure(&block)
     Sponges::Configuration.configure &block

data/lib/sponges/commander.rb

@@ -106,7 +106,7 @@ module Sponges
     end
 
     def alive?(pid)
-      !Sys::ProcTable.ps.find {|f| f.pid == pid }.nil?
+      !!Sys::ProcTable.ps.find {|f| f.pid == pid }.nil?
     end
 
     def gracefully?

data/lib/sponges/runner.rb

@@ -50,6 +50,7 @@ module Sponges
 
     def fork_supervisor
       fork do
+        $PROGRAM_NAME = "#{@name}_supervisor"
         Supervisor.new(@name, @options, store, @block).start
       end
     end

data/lib/sponges/store/memory.rb

@@ -1,4 +1,6 @@
 # encoding: utf-8
+require 'forwardable'
+
 module Sponges
   class Store
     class Memory

data/lib/sponges/supervisor.rb

@@ -1,45 +1,111 @@
 # encoding: utf-8
 module Sponges
   class Supervisor
-    attr_reader :store, :name, :options, :handler
+    attr_reader :store, :name, :options
 
     def initialize(name, options, store, block)
       @name, @options, @store, @block = name, options, store, block
-      $PROGRAM_NAME = "#{@name}_supervisor"
       store.on_fork
       store.register Process.pid
       @children_seen = 0
-      @handler = Handler.new self
     end
 
     def start
       trap_signals
+      at_exit do
+        Sponges.logger.info "Supervisor exits."
+      end
       options[:size].times do
-        handler.call :TTIN
+        fork_children
       end
       Sponges.logger.info "Supervisor started, waiting for messages."
       sleep
     rescue Exception => exception
       Sponges.logger.error exception
-      handler.call :INT
+      kill_them_all(:INT)
       raise exception
     end
 
-    def call
-      @block.call
-    end
-
     private
 
+    def fork_children
+      name = children_name
+      pid = fork do
+        $PROGRAM_NAME = name
+        (Sponges::SIGNALS + [:HUP]).each{ |sig| trap(sig) { exit!(0) } }
+        Sponges::Hook.after_fork
+        @block.call
+      end
+      Sponges.logger.info "Supervisor create a child with #{pid} pid."
+      store.add_children pid
+    end
+
     def children_name
       "#{name}_child_#{@children_seen +=1}"
     end
 
     def trap_signals
-      Sponges::SIGNALS.each do |signal|
-        trap(signal) {|signal| handler.call signal }
+      (Sponges::SIGNALS + [:HUP]).each do |signal|
+        trap(signal) do
+          handle_signal signal
+        end
+      end
+      trap(:TTIN) do
+        Sponges.logger.warn "Supervisor increment child's pool by one."
+        fork_children
+      end
+      trap(:TTOU) do
+        Sponges.logger.warn "Supervisor decrement child's pool by one."
+        if store.children_pids.first
+          kill_one(store.children_pids.first, :HUP)
+          store.delete_children(store.children_pids.first)
+        else
+          Sponges.logger.warn "No more child to kill."
+        end
+      end
+      trap(:CHLD) do
+        store.children_pids.each do |pid|
+          begin
+            dead = Process.waitpid(pid.to_i, Process::WNOHANG)
+            if dead
+              Sponges.logger.warn "Child #{dead} died. Restarting a new one..."
+                store.delete_children dead
+              Sponges::Hook.on_chld
+              fork_children
+            end
+          rescue Errno::ECHILD => e
+            # Don't panic
+          end
+        end
+      end
+    end
+
+    def handle_signal(signal)
+      Sponges.logger.info "Supervisor received #{signal} signal."
+      kill_them_all(signal)
+      Process.waitall
+      Sponges.logger.info "Children shutdown complete."
+      Sponges.logger.info "Supervisor shutdown. Exiting..."
+      store.clear(name)
+      exit
+    rescue Errno::ESRCH, Errno::ECHILD, SignalException => e
+      # Don't panic
+    end
+
+    def kill_them_all(signal)
+      store.children_pids.each do |pid|
+        kill_one(pid.to_i, signal)
       end
     end
 
+    def kill_one(pid, signal)
+      begin
+        Process.kill signal, pid
+        Process.waitpid pid
+        Sponges.logger.info "Child #{pid} receive a #{signal} signal."
+      rescue Errno::ESRCH, Errno::ECHILD, SignalException => e
+        # Don't panic
+      end
+    end
   end
 end

data/lib/sponges/version.rb

@@ -1,4 +1,4 @@
 # encoding: utf-8
 module Sponges
-  VERSION = "0.5.0"
+  VERSION = "0.5.0.1.pre"
 end

metadata

@@ -1,15 +1,15 @@
 --- !ruby/object:Gem::Specification
 name: sponges
 version: !ruby/object:Gem::Version
-  version: 0.5.0
-  prerelease: 
+  version: 0.5.0.1.pre
+  prerelease: 8
 platform: ruby
 authors:
 - chatgris
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-02-06 00:00:00.000000000 Z
+date: 2013-01-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: boson
@@ -75,7 +75,6 @@ files:
 - lib/sponges/cli.rb
 - lib/sponges/commander.rb
 - lib/sponges/configuration.rb
-- lib/sponges/handler.rb
 - lib/sponges/runner.rb
 - lib/sponges/store.rb
 - lib/sponges/store/memory.rb
@@ -97,9 +96,9 @@ required_ruby_version: !ruby/object:Gem::Requirement
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
-  - - ! '>='
+  - - ! '>'
     - !ruby/object:Gem::Version
-      version: '0'
+      version: 1.3.1
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.24
@@ -107,4 +106,3 @@ signing_key:
 specification_version: 3
 summary: Turn any ruby object to a daemon controlling an army of sponges.
 test_files: []
-has_rdoc: 

data/lib/sponges/handler.rb

@@ -1,126 +0,0 @@
-# encoding: utf-8
-module Sponges
-  class Handler
-    extend Forwardable
-    attr_reader :supervisor
-
-    def initialize(supervisor)
-      @supervisor = supervisor
-      at_exit do
-        for_supervisor do
-          Sponges.logger.info "Supervisor exits."
-        end
-      end
-    end
-
-    def call(signal)
-      if Sponges::SIGNALS.include?(signal = find_signal(signal))
-        send "handler_#{signal.to_s.downcase}", signal
-      end
-    end
-
-    def_delegators :@supervisor, :store, :fork_children, :name, :children_name
-
-    private
-
-    def for_supervisor
-      yield if Process.pid == store.supervisor_pid
-    end
-
-    def find_signal(signal)
-      return signal if signal.is_a?(Symbol)
-      if signal = Signal.list.find {|k,v| v == signal }
-        signal.first.to_sym
-      end
-    end
-
-    def handler_ttin(signal)
-      for_supervisor do
-        Sponges.logger.warn "Supervisor increment child's pool by one."
-        fork_children
-      end
-    end
-
-    def handler_ttou(signal)
-      for_supervisor do
-        Sponges.logger.warn "Supervisor decrement child's pool by one."
-        if store.children_pids.first
-          kill_one(store.children_pids.first, :HUP)
-          store.delete_children(store.children_pids.first)
-        else
-          Sponges.logger.warn "No more child to kill."
-        end
-      end
-    end
-
-    def handler_chld(signal)
-      for_supervisor do
-        return if stopping?
-        store.children_pids.each do |pid|
-          begin
-            if dead = Process.waitpid(pid.to_i, Process::WNOHANG)
-              store.delete_children dead
-              Sponges.logger.warn "Child #{dead} died. Restarting a new one..."
-              Sponges::Hook.on_chld
-              fork_children
-            end
-          rescue Errno::ECHILD => e
-            # Don't panic
-          end
-        end
-      end
-    end
-
-    def handler_int(signal)
-      for_supervisor do
-        @stopping = true
-        Sponges.logger.info "Supervisor received #{signal} signal."
-        kill_them_all(signal) and shutdown
-      end
-    end
-
-    alias handler_quit handler_int
-    alias handler_term handler_int
-
-    def kill_them_all(signal)
-      store.children_pids.map do |pid|
-        Thread.new { kill_one(pid.to_i, signal) }
-      end.each &:join
-    end
-
-    def kill_one(pid, signal)
-      begin
-        Process.kill signal, pid
-        Process.waitpid pid
-        Sponges.logger.info "Child #{pid} receive a #{signal} signal."
-      rescue Errno::ESRCH, Errno::ECHILD, SignalException => e
-        # Don't panic
-      end
-    end
-
-    def shutdown
-      Process.waitall
-      Sponges.logger.info "Children shutdown complete.", "Supervisor shutdown. Exiting..."
-      store.clear(name)
-      exit
-    rescue Errno::ESRCH, Errno::ECHILD, SignalException => e
-      # Don't panic
-    end
-
-    def fork_children
-      name = children_name
-      pid = fork do
-        $PROGRAM_NAME = name
-        (Sponges::STOP_SIGNALS + [:HUP]).each{ |sig| trap(sig) { exit!(0) } }
-        Sponges::Hook.after_fork
-        supervisor.call
-      end
-      Sponges.logger.info "Supervisor create a child with #{pid} pid."
-      store.add_children pid
-    end
-
-    def stopping?
-      @stopping
-    end
-  end
-end