sponges 0.0.4 → 0.0.5

data/README.md

@@ -103,6 +103,16 @@ ruby example.rb stop -g
 In this case, you will have to trap the `HUP` signal, and handle a clean stop
 from each workers. The point is to wait for a task to be done before quitting.
 
+Increment worker's pool size :
+``` bash
+ruby example.rb increment # will add a worker to the pool.
+```
+
+Decrement worker's pool size :
+``` bash
+ruby example.rb decrement # will remove a worker to the pool.
+```
+
 Show a list of workers and their children.
 ``` bash
 ruby example.rb list
@@ -110,8 +120,7 @@ ruby example.rb list
 
 ## TODO
 
-* Specing.
-* Increment / decrement workers pool size with `TTIN` and `TTOUT`.
+* More specs.
 * Check on OSX.
 
 ## Copyright

data/lib/sponges/cli.rb

@@ -2,7 +2,7 @@
 module Sponges
   # This class concern is to expose a nice CLI interface.
   #
-  class Cli  < Boson::Runner
+  class Cli < Boson::Runner
     option :daemonize,  type: :boolean
     option :size,       type: :numeric
     desc "Start workers"
@@ -32,6 +32,18 @@ module Sponges
       start(options)
     end
 
+    desc "Increment workers pool size"
+    def increment(options = {})
+      Sponges::Commander.new(Sponges::Configuration.worker_name, options).
+        increment
+    end
+
+    desc "Decrement workers pool size"
+    def decrement(options = {})
+      Sponges::Commander.new(Sponges::Configuration.worker_name, options).
+        decrement
+    end
+
     desc "Show running processes"
     def list
       redis = Nest.new('sponges')

data/lib/sponges/commander.rb

@@ -27,6 +27,32 @@ module Sponges
       end
     end
 
+    def increment
+      Sponges.logger.info "Runner #{@name} increment message received."
+      if pid = @redis[:worker][@name][:supervisor].get
+        begin
+          Process.kill :TTIN, pid.to_i
+        rescue Errno::ESRCH => e
+          Sponges.logger.error e
+        end
+      else
+        Sponges.logger.info "No supervisor found."
+      end
+    end
+
+    def decrement
+      Sponges.logger.info "Runner #{@name} decrement message received."
+      if pid = @redis[:worker][@name][:supervisor].get
+        begin
+          Process.kill :TTOU, pid.to_i
+        rescue Errno::ESRCH => e
+          Sponges.logger.error e
+        end
+      else
+        Sponges.logger.info "No supervisor found."
+      end
+    end
+
     private
 
     def alive?(pid)

data/lib/sponges/supervisor.rb

@@ -53,6 +53,18 @@ module Sponges
           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 pids.first
+          kill_one(pids.first, :HUP)
+        else
+          Sponges.logger.warn "No more child to kill."
+        end
+      end
       trap(:CHLD) do
         pids.each do |pid|
           begin
@@ -83,13 +95,17 @@ module Sponges
 
     def kill_them_all(signal)
       pids.each do |pid|
-        begin
-          Process.kill signal, pid.to_i
-          @pids.srem pid
-          Sponges.logger.info "Child #{pid} receive a #{signal} signal."
-        rescue Errno::ESRCH => e
-          # Don't panic
-        end
+        kill_one(pid, signal)
+      end
+    end
+
+    def kill_one(pid, signal)
+      begin
+        Process.kill signal, pid.to_i
+        @pids.srem pid
+        Sponges.logger.info "Child #{pid} receive a #{signal} signal."
+      rescue Errno::ESRCH => e
+        # Don't panic
       end
     end
 

data/lib/sponges/version.rb

@@ -1,4 +1,4 @@
 # encoding: utf-8
 module Sponges
-  VERSION = "0.0.4"
+  VERSION = "0.0.5"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sponges
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-10-12 00:00:00.000000000 Z
+date: 2012-10-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: boson