boreman 0.1.27 → 0.1.28

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: e6335a3b3a005c92b43305c6a9a185597cb623ed
-  data.tar.gz: dc5ba754a8387c82cc795b5afdd71f091ab5440d
+  metadata.gz: b43032f6ccbd8241a33a650f841b0918cb1d33d4
+  data.tar.gz: 8b7fde40f727ed9f62e6fc1eec3edc9ba99f75cd
 SHA512:
-  metadata.gz: 280a78fd78cac5f2cd49a0e188154981618bc1b60e4b5dfc6ae5834cacc1f5a585f0d5b3f99be26c9d611d1bc794a3f23b470c7f2b1be665301e246f6e566f63
-  data.tar.gz: 951312a6f3b690d94969211b0af5a36c05b661ddd9ef647dfaa0a0750c7755d8eb35e1cc66663ead9157079c9225f0079eb06ec5d68829755928ad4aea2c7108
+  metadata.gz: a92f68270e25b7f0449c95aac5089b7f75d2298ce094385736d4253fe7547e7aaf962f41002e1e0bc98cb398294dcaf9017d820bfa6ddcfdcf9f73ad9d2cdda3
+  data.tar.gz: 61f18ed3c883abd891780d6bd2dbaf6c2b83ac7a17ad2c22066dd130fba7fe0cf399e976e66043bfc0a67804780bc087cd9281c7c4bdae038824d50cebbeed35

data/lib/boreman.rb

@@ -1,4 +1,5 @@
 require "boreman/version"
+require "boreman/daemonize"
 
 module Boreman
   def self.run(action, selector, opts = {})
@@ -93,32 +94,13 @@ module Boreman
     if cmd = procs[selector]
       cmd = prepare_command(cmd)
 
-      begin
-        r, w = IO.pipe
-
-        fpid = fork do
-          STDOUT.reopen(w)
-
-          r.close
-
-          pid = Process.spawn cmd, pgroup: true
-
-          puts pid.to_s
-
-          exit! 0
-        end
-
-        pid = r.gets.chomp.to_i
-
-        Process.detach(fpid)
-        Process.detach(pid)
-
-        write_pid selector, pid
-        puts "Started #{selector}, pid = #{pid}"
-      ensure
-        r.close rescue nil
-        w.close rescue nil
+      run = Proc.new do
+        Kernel.exec cmd
       end
+
+      pid = Daemonize.call_as_daemon run
+      write_pid selector, pid
+      puts "Started #{selector}, pid = #{pid}"
     else
       puts "Entry #{selector} not found in Procfile"
     end

data/lib/boreman/daemonize.rb

@@ -0,0 +1,141 @@
+module Daemonize
+  # Try to fork if at all possible retrying every 5 sec if the
+  # maximum process limit for the system has been reached
+  def self.safefork
+    tryagain = true
+
+    while tryagain
+      tryagain = false
+      begin
+        if pid = fork
+          return pid
+        end
+      rescue Errno::EWOULDBLOCK
+        sleep 5
+        tryagain = true
+      end
+    end
+  end
+
+  # Call a given block as a daemon
+  def self.call_as_daemon(block, logfile_name = nil, app_name = nil)
+    # we use a pipe to return the PID of the daemon
+    rd, wr = IO.pipe
+
+    if tmppid = safefork
+      # in the parent
+
+      wr.close
+      pid = rd.read.to_i
+      rd.close
+
+      Process.waitpid(tmppid)
+
+      return pid
+    else
+      # in the child
+
+      rd.close
+
+      # Detach from the controlling terminal
+      unless Process.setsid
+        fail Daemons.RuntimeException.new('cannot detach from controlling terminal')
+      end
+
+      # Prevent the possibility of acquiring a controlling terminal
+      trap 'SIGHUP', 'IGNORE'
+      exit if pid = safefork
+
+      wr.write Process.pid
+      wr.close
+
+      $0 = app_name if app_name
+
+      close_io
+
+      redirect_io(logfile_name)
+
+      # Split rand streams between spawning and daemonized process
+      srand
+
+      block.call
+
+      exit
+    end
+  end
+
+  # Transform the current process into a daemon
+  def self.daemonize(logfile_name = nil, app_name = nil)
+    # Fork and exit from the parent
+    safefork && exit
+
+    # Detach from the controlling terminal
+    unless sess_id = Process.setsid
+      fail Daemons.RuntimeException.new('cannot detach from controlling terminal')
+    end
+
+    # Prevent the possibility of acquiring a controlling terminal
+    trap 'SIGHUP', 'IGNORE'
+    exit if safefork
+
+    $0 = app_name if app_name
+
+    # Release old working directory
+    Dir.chdir '/'
+
+    close_io
+
+    redirect_io(logfile_name)
+
+    # Split rand streams between spawning and daemonized process
+    srand
+
+    sess_id
+  end
+
+  def self.close_io
+    # Make sure all input/output streams are closed
+    # Part I: close all IO objects (except for $stdin/$stdout/$stderr)
+    ObjectSpace.each_object(IO) do |io|
+      unless [$stdin, $stdout, $stderr].include?(io)
+        io.close rescue nil
+      end
+    end
+
+    # Make sure all input/output streams are closed
+    # Part II: close all file decriptors (except for $stdin/$stdout/$stderr)
+    3.upto(8192) do |i|
+      IO.for_fd(i).close rescue nil
+    end
+  end
+
+  # Free $stdin/$stdout/$stderr file descriptors and
+  # point them somewhere sensible
+  def self.redirect_io(logfile_name)
+    begin; $stdin.reopen '/dev/null'; rescue ::Exception; end
+
+    if logfile_name == 'SYSLOG'
+      # attempt to use syslog via syslogio
+      begin
+        require 'syslogio'
+        $stdout = ::Daemons::SyslogIO.new($0, :local0, :info, $stdout)
+        $stderr = ::Daemons::SyslogIO.new($0, :local0, :err, $stderr)
+        # error out early so we can fallback to null
+        $stdout.puts "no logfile provided, output redirected to syslog"
+      rescue ::Exception
+        # on unsupported platforms simply reopen /dev/null
+        begin; $stdout.reopen '/dev/null'; rescue ::Exception; end
+        begin; $stderr.reopen '/dev/null'; rescue ::Exception; end
+      end
+    elsif logfile_name
+      $stdout.reopen logfile_name, 'a'
+      File.chmod(0644, logfile_name)
+      $stdout.sync = true
+      begin; $stderr.reopen $stdout; rescue ::Exception; end
+      $stderr.sync = true
+    else
+      begin; $stdout.reopen '/dev/null'; rescue ::Exception; end
+      begin; $stderr.reopen '/dev/null'; rescue ::Exception; end
+    end
+  end
+end

data/lib/boreman/version.rb

@@ -1,3 +1,3 @@
 module Boreman
-  VERSION = "0.1.27"
+  VERSION = "0.1.28"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: boreman
 version: !ruby/object:Gem::Version
-  version: 0.1.27
+  version: 0.1.28
 platform: ruby
 authors:
 - Kyle Brett
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-03-25 00:00:00.000000000 Z
+date: 2017-03-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -58,6 +58,7 @@ files:
 - boreman.gemspec
 - exe/boreman
 - lib/boreman.rb
+- lib/boreman/daemonize.rb
 - lib/boreman/version.rb
 homepage: 
 licenses: []