worker_pool.rb in daemonizer-0.5.0.beta.1

- old
+ new

@@ -1,20 +1,20 @@
 module Daemonizer
   class WorkerPool
     MONITOR_VALUE = [:vm_size, :private_dirty_rss, :rss]
-    
+
     attr_reader :name
     attr_reader :stats
 
     def initialize(name, pm, &blk)
       @name = name
       @pm = pm
       @worker_block = blk
       @workers = []
       @stats = ::SimpleStatistics::DataSet.new
     end
-    
+
     def find_worker_by_name(name)
       @workers.detect do |w|
         w.process_name.to_s == name.to_s
       end
     end
@@ -27,11 +27,11 @@
       Daemonizer.logger.debug "Creating #{number} workers for #{name} pool..."
       number.times do |i|
         worker = Worker.new(name, @pm, i+1, &@worker_block)
         @workers << worker
         @stats.add_data(worker.process_name)
-        Daemonizer.logger.info "Gathering data for #{worker.name}"    
+        Daemonizer.logger.info "Gathering data for #{worker.name}"
       end
     rescue Exception => e
       Daemonizer.logger.info "Result - #{e.inspect}"
     end
 
@@ -43,11 +43,11 @@
         MONITOR_VALUE.each do |value|
           @stats.tick(value)
           @stats[worker_name][value].add_probe(p.send(value))
         end
       end
-      
+
       @workers.each do |worker|
         unless worker.running? || worker.shutdown?
           Daemonizer.logger.warn "Worker #{worker.name} is not running. Restart!"
           @stats.add_data(worker.process_name)
           MONITOR_VALUE.each do |v|
@@ -72,9 +72,15 @@
     def stop_workers(force)
       Daemonizer.logger.debug "Stopping #{name} pool workers..."
       @workers.each do |worker|
         next unless worker.running?
         worker.stop(force)
+      end
+    end
+
+    def send_signal_to_workers(signal)
+      @workers.each do |worker|
+        worker.send_signal(signal)
       end
     end
   end
 end