daemonizer 0.4.18 → 0.5.0.beta.1

data/lib/daemonizer/process_manager.rb

@@ -21,13 +21,13 @@ module Daemonizer
         Daemonizer.logger.debug "Checking workers' health..."
         break if shutdown?
         @pool.check_workers
-        
+
         @config.on_poll.each do |on_poll|
           on_poll.call(@pool)
         end
 
         break if shutdown?
-        Daemonizer.logger.debug "Sleeping for #{@config.poll_period} seconds..." 
+        Daemonizer.logger.debug "Sleeping for #{@config.poll_period} seconds..."
         sleep(@config.poll_period)
       end
     rescue Exception => e
@@ -69,7 +69,7 @@ module Daemonizer
       return false
     end
 
-    def stop_workers(force = false)      
+    def stop_workers(force = false)
       # Set shutdown flag
       Daemonizer.logger.debug "Stopping workers#{force ? ' (forced)' : ''}..."
 
@@ -77,6 +77,10 @@ module Daemonizer
       @pool.stop_workers(force)
     end
 
+    def send_signal_to_workers(signal)
+      @pool.send_signal_to_workers(signal)
+    end
+
     def shutdown?
       @shutdown
     end

data/lib/daemonizer/version.rb

@@ -0,0 +1,3 @@
+module Daemonizer
+  VERSION = "0.5.0.beta.1"
+end

data/lib/daemonizer/worker.rb

@@ -28,11 +28,13 @@ module Daemonizer
         STDIN.reopen '/dev/null'
         STDOUT.reopen '/dev/null', 'a'
         STDERR.reopen STDOUT
-        
+
+        setup_signals
+
         @pid = Process.pid
         Daemonizer.reopen_log_file
         Daemonizer.logger_context = "#{@pid}/#{@worker_id}"
-        
+
         Daemonizer.logger.info "Log file reopened after fork"
         normal_exit = false
         begin
@@ -80,5 +82,17 @@ module Daemonizer
         Daemonizer.logger.error("Exception from kill: #{e} at #{e.backtrace.first}")
       end
     end
+
+    def send_signal(signal)
+      Process.kill(signal, @pid)
+    end
+
+    private
+
+    def setup_signals
+      trap('SIGHUP') do
+        Daemonizer.reopen_log_file
+      end
+    end
   end
 end

data/lib/daemonizer/worker_pool.rb

@@ -1,7 +1,7 @@
 module Daemonizer
   class WorkerPool
     MONITOR_VALUE = [:vm_size, :private_dirty_rss, :rss]
-    
+
     attr_reader :name
     attr_reader :stats
 
@@ -12,7 +12,7 @@ module Daemonizer
       @workers = []
       @stats = ::SimpleStatistics::DataSet.new
     end
-    
+
     def find_worker_by_name(name)
       @workers.detect do |w|
         w.process_name.to_s == name.to_s
@@ -29,7 +29,7 @@ module Daemonizer
         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}"
@@ -45,7 +45,7 @@ module Daemonizer
           @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!"
@@ -76,5 +76,11 @@ module Daemonizer
         worker.stop(force)
       end
     end
+
+    def send_signal_to_workers(signal)
+      @workers.each do |worker|
+        worker.send_signal(signal)
+      end
+    end
   end
 end

data/spec/cli/debugging_spec.rb

@@ -0,0 +1,45 @@
+require "spec_helper"
+
+describe "with Daemonfile and daemonzier is in 'debug' mode" do
+
+  before :each do
+    @pid_files = simple_daemonfile(
+             :name => :test1,
+             :pid_file =>"#{tmp_dir}/test1.pid",
+             :on_prepare => "Daemonizer.logger.info \"test1: executed prepare\"",
+             :on_start => "Daemonizer.logger.info \"test1: executed start\"")
+  end
+
+  describe "with all pools" do
+
+    before(:each) do
+      daemonizer :debug
+    end
+
+    it "should return valid text" do
+      @out.should match(/You should supply pool_name to debug/)
+    end
+
+    it "should not return anything to stderr" do
+      @err.should == ''
+    end
+  end
+
+  describe "with specific pool" do
+
+    before(:each) do
+      daemonizer "debug test1"
+    end
+
+    it "should return valid text" do
+      @out.should match(/test1: executed prepare/)
+      @out.should match(/test1: executed start/)
+    end
+
+    it "should not return anything to stderr" do
+      @err.should == ''
+    end
+  end
+
+
+end

data/spec/cli/not_started_spec.rb

@@ -0,0 +1,127 @@
+require "spec_helper"
+
+describe "with Daemonfile and daemonzier is not started " do
+
+  before :each do
+    @pid_files = simple_daemonfile({:name => :test1, :pid_file =>"#{tmp_dir}/test1.pid"}, {:name => :test2, :pid_file => "#{tmp_dir}/test2.pid"})
+  end
+
+  describe "on call 'stats'" do
+    before(:each) do
+      daemonizer :stats
+    end
+
+    it "should return valid text" do
+      @out.should match(/It seems like pool 'test1' is not running/)
+      @out.should match(/It seems like pool 'test2' is not running/)
+    end
+
+    it "should not return anything to stderr" do
+      @err.should == ''
+    end
+  end
+
+  describe "on call 'logrotate'" do
+    before(:each) do
+      daemonizer :logrotate
+    end
+
+    it "should return valid text" do
+      @out.should match(/test1: not started/)
+      @out.should match(/test2: not started/)
+    end
+
+    it "should not return anything to stderr" do
+      @err.should == ''
+    end
+  end
+
+  describe "on call 'start'" do
+    describe "with all pools" do
+      before(:each) do
+        daemonizer :start
+      end
+
+      after(:each) do
+        daemonizer :stop
+      end
+
+      it "should print info about starting 2 pools" do
+        @out.should match(/test1: Starting pool/)
+        @out.should match(/test1: successfully started/)
+        @out.should match(/test2: Starting pool/)
+        @out.should match(/test2: successfully started/)
+      end
+
+      it "should not return anything to stderr" do
+        @err.should == ''
+      end
+
+      it "should run daemonizer processes" do
+        daemonizer_runned?(@pid_files[0]).should be_true
+        daemonizer_runned?(@pid_files[1]).should be_true
+      end
+    end
+
+    describe "with specific pool" do
+      before(:each) do
+        daemonizer "start test1"
+      end
+
+      after(:each) do
+        daemonizer "stop test1"
+      end
+
+      it "should print info about starting 2 pools" do
+        @out.should match(/test1: Starting pool/)
+        @out.should match(/test1: successfully started/)
+        @out.should_not match(/test2: Starting pool/)
+        @out.should_not match(/test2: successfully started/)
+      end
+
+      it "should not return anything to stderr" do
+        @err.should == ''
+      end
+
+      it "should run daemonizer processes" do
+        daemonizer_runned?(@pid_files[0]).should be_true
+        daemonizer_runned?(@pid_files[1]).should be_false
+      end
+    end
+  end
+
+  describe "on call 'stop'" do
+    describe "with all pools" do
+
+      before(:each) do
+        daemonizer :stop
+      end
+
+      it "should return valid text" do
+        @out.should match(/test1: No pid file or a stale pid file!/)
+        @out.should match(/test2: No pid file or a stale pid file!/)
+      end
+
+      it "should not return anything to stderr" do
+        @err.should == ''
+      end
+    end
+
+    describe "with specific pool" do
+
+      before(:each) do
+        daemonizer "stop test1"
+      end
+
+      it "should return valid text" do
+        @out.should match(/test1: No pid file or a stale pid file!/)
+        @out.should_not match(/test2: No pid file or a stale pid file!/)
+      end
+
+      it "should not return anything to stderr" do
+        @err.should == ''
+      end
+    end
+
+  end
+end

data/spec/cli/started_spec.rb

@@ -0,0 +1,121 @@
+require "spec_helper"
+
+describe "with Daemonfile and daemonzier is already started " do
+
+  before :each do
+    @pid_files = simple_daemonfile({:name => :test1, :pid_file =>"#{tmp_dir}/test1.pid"}, {:name => :test2, :pid_file => "#{tmp_dir}/test2.pid"})
+    daemonizer "start"
+  end
+
+  after :each do
+    daemonizer "stop"
+  end
+
+  describe "on call 'stats'" do
+    before(:each) do
+      daemonizer :stats
+    end
+
+    it "should return valid text" do
+      @out.should match(/test1 processes/)
+      @out.should match(/test2 processes/)
+    end
+
+    it "should not return anything to stderr" do
+      @err.should == ''
+    end
+  end
+
+  describe "on call 'logrotate'" do
+    before(:each) do
+      daemonizer :logrotate
+    end
+
+    it "should return valid text" do
+      @out.should match(/test1: log file reopened/)
+      @out.should match(/test2: log file reopened/)
+    end
+
+    it "should not return anything to stderr" do
+      @err.should == ''
+    end
+  end
+
+  describe "on call 'start'" do
+    describe "with all pools" do
+      before(:each) do
+        daemonizer 'start'
+      end
+
+      it "should print info about starting 2 pools" do
+        @out.should match(/test1: Can't start, another process exists!/)
+        @out.should match(/test2: Can't start, another process exists!/)
+      end
+
+      it "should not return anything to stderr" do
+        @err.should == ''
+      end
+
+      it "should run daemonizer processes" do
+        daemonizer_runned?(@pid_files[0]).should be_true
+        daemonizer_runned?(@pid_files[1]).should be_true
+      end
+    end
+
+    describe "with specific pool" do
+      before(:each) do
+        daemonizer "start test1"
+      end
+
+      it "should print info about starting 2 pools" do
+        @out.should match(/test1: Can't start, another process exists!/)
+        @out.should_not match(/test2/)
+      end
+
+      it "should not return anything to stderr" do
+        @err.should == ''
+      end
+
+      it "should run daemonizer processes" do
+        daemonizer_runned?(@pid_files[0]).should be_true
+        daemonizer_runned?(@pid_files[1]).should be_true
+      end
+    end
+
+  end
+
+  describe "on call 'stop'" do
+    describe "with all pools" do
+
+      before(:each) do
+        daemonizer "stop"
+      end
+
+      it "should return valid text" do
+        @out.should match(/test1: Killing the process/)
+        @out.should match(/test2: Killing the process/)
+      end
+
+      it "should not return anything to stderr" do
+        @err.should == ''
+      end
+    end
+
+    describe "with specific pool" do
+
+      before(:each) do
+        daemonizer "stop test1"
+      end
+
+      it "should return valid text" do
+        @out.should match(/test1: Killing the process/)
+        @out.should_not match(/test2: Killing the process/)
+      end
+
+      it "should not return anything to stderr" do
+        @err.should == ''
+      end
+    end
+
+  end
+end

data/spec/common/forking_spec.rb

@@ -0,0 +1,35 @@
+require "spec_helper"
+
+describe "daemonzier after start" do
+
+  before :each do
+    @pid_files = simple_daemonfile(
+             :name => :test1,
+             :pid_file =>"#{tmp_dir}/test1.pid",
+             :on_start => "loop { sleep 1 }",
+             :workers => 3,
+             :poll_period => 1)
+    daemonizer :start
+    sleep 8
+  end
+
+  after :each do
+    daemonizer :stop
+  end
+
+  it "should create 3 forks" do
+    children_count(pid(@pid_files[0])).should == 3
+  end
+
+  describe "after one worker died" do
+    before :each do
+      Process.kill("KILL", children_pids(pid(@pid_files[0])).first.to_i)
+      sleep 5
+    end
+
+    it "should restore it" do
+      children_count(pid(@pid_files[0])).should == 3
+    end
+  end
+
+end