light-daemon 0.9.0 → 0.9.5

data/lib/light_daemon.rb

@@ -1,49 +1,98 @@
 module LightDaemon
   class Daemon
-    def initialize(obj, options={})
-      @obj = obj
-      @options = options
-      log("before daemonize")
+    DEFAULT_PID_FILE = "/tmp/light_daemon.pid"
+
+    class << self
+      def start(obj, options={})
+        @options = self.make_options(options)
+	@pid_file = @options[:pid_file]
+        if self.get_pid
+          raise "The pid file \"#{@pid_file}\" existed already. You need to stop the daemon first."
+        end
+        self.new(obj, @options).start
+      end
+  
+      def stop(pid_file=DEFAULT_PID_FILE)
+        @pid_file = pid_file
+        if(pid = self.get_pid)
+	  begin
+	    Process.kill("TERM", pid)
+	  rescue Errno::ESRCH   
+	    # no such process. do nothing
+          end
+	  self.clear_pid
+        end
+      end
+
+      def get_pid
+        return nil unless File.exist?(@pid_file)
+	begin 
+	  File.open(@pid_file) {|f| f.read}.to_i
+        rescue
+	  nil
+        end
+      end
+
+      def set_pid(pid)
+        File.open(@pid_file, 'w') {|f| f.write(pid.to_s)}
+      end
+
+      def clear_pid
+        File.unlink(@pid_file)
+      end
+
+      def make_options(options)
+        options[:children] ||= 1
+        options[:monitor_cycle] ||= 10    # seconds
+        options[:pid_file] ||= DEFAULT_PID_FILE
+	options
+      end
+
+      # for debugging 
+#      def log(msg)
+#        File.open("/tmp/yi.txt", "a") do |f|
+#          f.puts msg
+#        end
+#      end
+    end
+
+    def initialize(obj, options)
       daemonize
-      log("after daemonize")
+      self.class.set_pid(Process.pid)
+
+      @options = options
+      @obj = obj
       @processes = [] 
+      @is_child = false
       signal_handling
     end
 
-    def self.start(obj, options={})
-      self.new(obj, options).start
-    end
-
     def signal_handling
-      $_simple_daemon_running = true
+      $keep_running = true
       Signal.trap "TERM" do
-        log("Daemon received TERM !!!!!!!!!!!!!!!!!")
-        @processes.each do |pid|
-          Process.kill("TERM", pid)
-        end
-	  
-        sleep(3)
-        3.times do 
-	  @processes.size.times do
-	    pid = @processes.shift
-	    if process_alive?(pid)
-	      @processes << pid
-	      sleep(3)
-            end
-	  end  
-        end
-	@processes.each do |pid|
-	  Process.kill(9, pid)
+        unless @is_child
+          @processes.each do |pid|
+            Process.kill("TERM", pid)
+          end
+  	  
+          sleep(3)
+          options[:children].times do 
+    	    @processes.size.times do
+      	      pid = @processes.shift
+      	      if process_alive?(pid)
+      	        @processes << pid
+      	        sleep(3)
+              end
+    	    end  
+          end
+  	  @processes.each do |pid|
+  	    Process.kill(9, pid)
+          end
         end
-        $_simple_daemon_running = false
+        $keep_running = false
       end
     end
 
-    def log(msg)
-      File.open("/tmp/yi.txt", "a") do |f|
-        f.puts msg
-      end
-    end
 
     def process_alive?(pid)
       return false unless pid
@@ -57,33 +106,45 @@ module LightDaemon
       end
     end
 
-
     def start
-      5.times do 
+      @options[:children].times do 
         if pid = fork 
           @processes << pid
 	  Process.detach(pid)
         else
-          return @obj.send(:call)
+	  set_child
+          while($keep_running)
+	    break unless @obj.send(:call)
+	  end
+	  return
         end	
       end
 
-      while($_simple_daemon_running)
-        sleep(5)
+      while($keep_running)
+        sleep(@options[:monitor_cycle])
+
 	@processes.each_with_index do |pid, index|
           unless process_alive?(pid)
             if pid = fork
 	      @processes[index] = pid
 	      Process.detach(pid)
+	      sleep(2)
 	    else 
-              return @obj.send(:call)
+	      set_child 
+              while($keep_running)
+	        break unless @obj.send(:call)
+	      end
+              return
 	    end  
+          else
           end
 	  sleep(1)
         end
       end
+      self.class.clear_pid(@options[:pid_file])
     end
 
+    # some code in this method is learned from gem daemons
     def daemonize
       fork && exit
       unless Process.setsid
@@ -93,22 +154,14 @@ module LightDaemon
       trap 'SIGHUP', 'IGNORE'
       exit if fork
 
-      ObjectSpace.each_object(IO) do |io|
-        unless [STDIN, STDOUT, STDERR].include?(io)
-          begin
-	    unless io.closed?
-	      io.close
-	    end
-	  rescue   ::Exception
-	  end
-        end
-      end
-
       begin; STDIN.reopen "/dev/null"; rescue ::Exception; end
       begin; STDOUT.reopen "/dev/null"; rescue ::Exception; end
       begin; STDERR.reopen STDOUT; rescue ::Exception; end
       STDERR.sync = true
+    end
 
+    def set_child 
+      @is_child = true
     end
   end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: light-daemon
 version: !ruby/object:Gem::Version 
-  hash: 59
+  hash: 49
   prerelease: 
   segments: 
   - 0
   - 9
-  - 0
-  version: 0.9.0
+  - 5
+  version: 0.9.5
 platform: ruby
 authors: 
 - Yi Zhang
@@ -15,11 +15,11 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-03-08 00:00:00 -08:00
+date: 2012-03-10 00:00:00 -08:00
 default_executable: 
 dependencies: []
 
-description: A simple and light daemon
+description: A very simple and light ruby daemon API
 email: yzhang.wa@gmail.com
 executables: []