import FCGI-ProcManager 0.19 from CPAN
[catagits/FCGI-ProcManager.git] / ProcManager.pm
index d11835b..f8e0797 100644 (file)
@@ -5,20 +5,22 @@ package FCGI::ProcManager;
 # Public License, Version 2.1.  Please read the important licensing and
 # disclaimer information included below.
 
-# $Id: ProcManager.pm,v 1.15 2001/01/31 07:00:55 muaddib Exp $
+# $Id: ProcManager.pm,v 1.23 2001/04/23 16:10:11 muaddie Exp $
 
 use strict;
 use Exporter;
+use POSIX qw(:signal_h);
 
-use vars qw($VERSION @ISA @EXPORT_OK %EXPORT_TAGS $Q);
+use vars qw($VERSION @ISA @EXPORT_OK %EXPORT_TAGS $Q $SIG_CODEREF);
 BEGIN {
-  $VERSION = '0.15';
+  $VERSION = '0.19'; 
   @ISA = qw(Exporter);
-  @EXPORT_OK = qw(pm_manage pm_die pm_reap_server
+  @EXPORT_OK = qw(pm_manage pm_die pm_wait
                  pm_write_pid_file pm_remove_pid_file
                  pm_pre_dispatch pm_post_dispatch
                  pm_change_process_name pm_received_signal pm_parameter 
-                 pm_warn pm_notify pm_abort pm_exit);
+                 pm_warn pm_notify pm_abort pm_exit
+                 $SIG_CODEREF);
   $EXPORT_TAGS{all} = \@EXPORT_OK;
   $FCGI::ProcManager::Default = 'FCGI::ProcManager';
 }
@@ -33,7 +35,9 @@ BEGIN {
  # In Object-oriented style.
  use CGI::Fast;
  use FCGI::ProcManager;
- my $proc_manager = FCGI::ProcManager->new({ n_processes => 10 });
+ my $proc_manager = FCGI::ProcManager->new({
+       n_processes => 10 
+ });
  $proc_manager->pm_manage();
  while (my $cgi = CGI::Fast->new()) {
    $proc_manager->pm_pre_dispatch();
@@ -43,7 +47,8 @@ BEGIN {
 
  # This style is also supported:
  use CGI::Fast;
- use FCGI::ProcManager qw(pm_manage pm_pre_dispatch pm_post_dispatch);
+ use FCGI::ProcManager qw(pm_manage pm_pre_dispatch 
+                         pm_post_dispatch);
  pm_manage( n_processes => 10 );
  while (my $cgi = CGI::Fast->new()) {
    pm_pre_dispatch();
@@ -72,27 +77,95 @@ It is necessary for the caller, when implementing its request loop, to
 insert a call to C<pm_pre_dispatch> at the top of the loop, and then
 7C<pm_post_dispatch> at the end of the loop.
 
+=head2 Signal Handling
+
+FCGI::ProcManager attempts to do the right thing for proper shutdowns now.
+
+When it receives a SIGHUP, it sends a SIGTERM to each of its children, and
+then resumes its normal operations.   
+
+When it receives a SIGTERM, it sends a SIGTERM to each of its children, sets
+an alarm(3) "die timeout" handler, and waits for each of its children to
+die.  If all children die before this timeout, process manager exits with
+return status 0.  If all children do not die by the time the "die timeout"
+occurs, the process manager sends a SIGKILL to each of the remaining
+children, and exists with return status 1.
+
+In order to get FastCGI servers to exit upon receiving a signal, it is
+necessary to use its FAIL_ACCEPT_ON_INTR.  See FCGI.pm's description of
+FAIL_ACCEPT_ON_INTR.  Unfortunately, if you want/need to use CGI::Fast, it
+appears currently necessary to modify your installation of FCGI.pm, with
+something like the following:
+
+ -*- patch -*-
+ --- FCGI.pm     2001/03/09 01:44:00     1.1.1.3
+ +++ FCGI.pm     2001/03/09 01:47:32     1.2
+ @@ -24,7 +24,7 @@
+  *FAIL_ACCEPT_ON_INTR = sub() { 1 };
+  
+  sub Request(;***$$$) {
+ -    my @defaults = (\*STDIN, \*STDOUT, \*STDERR, \%ENV, 0, 0);
+ +    my @defaults = (\*STDIN, \*STDOUT, \*STDERR, \%ENV, 0, FAIL_ACCEPT_ON_INTR());
+      splice @defaults,0,@_,@_;
+      RequestX(@defaults);
+  }   
+ -*- end patch -*-
+
+Otherwise, if you don't, there is a loop around accept(2) which prevents
+os_unix.c OS_Accept() from returning the necessary error when FastCGI
+servers blocking on accept(2) receive the SIGTERM or SIGHUP.
+
+FCGI::ProcManager uses POSIX::sigaction() to override the default SA_RESTART
+policy used for perl's %SIG behavior.  Specifically, the process manager
+never uses SA_RESTART, while the child FastCGI servers turn off SA_RESTART
+around the accept(2) loop, but re-enstate it otherwise.
+
+The desired (and implemented) effect is to give a request as big a chance as
+possible to succeed and to delay their exits until after their request,
+while allowing the FastCGI servers waiting for new requests to die right
+away. 
 
 =head1 METHODS
 
 =head2 new
 
+ class or instance
+ (ProcManager) new([hash parameters])
+
+Constructs a new process manager.  Takes an option has of initial parameter
+values, and assigns these to the constructed object HASH, overriding any
+default values.  The default parameter values currently are:
+
+ role         => manager
+ start_delay  => 0
+ die_timeout  => 60
+ pm_title => 'perl-fcgi-pm'
+
 =cut
 
 sub new {
   my ($proto,$init) = @_;
+  $init ||= {};
 
   my $this = { 
              role => "manager",
              start_delay => 0,
-             die_timeout => 60
+             die_timeout => 60,
+        pm_title => 'perl-fcgi-pm',
+             %$init
             };
-  $init and %$this = %$init;
-
   bless $this, ref($proto)||$proto;
 
   $this->{PIDS} = {};
 
+  # initialize signal constructions.
+  unless ($this->no_signals()) {
+    $this->{sigaction_no_sa_restart} =
+       POSIX::SigAction->new('FCGI::ProcManager::sig_sub');
+    $this->{sigaction_sa_restart} =
+       POSIX::SigAction->new('FCGI::ProcManager::sig_sub',undef,POSIX::SA_RESTART);
+  }
+
   return $this;
 }
 
@@ -100,12 +173,22 @@ sub new {
 
 =head2 pm_manage
 
- global
- () pm_manage(int processes_to_spawn)
+ instance or export
+ (int) pm_manage([hash parameters])
 
 DESCRIPTION:
 
-When this is called by a FastCGI script to manage application servers.
+When this is called by a FastCGI script to manage application servers.  It
+defines a sequence of instructions for a process to enter this method and
+begin forking off and managing those handlers, and it defines a sequence of
+instructions to intialize those handlers.
+
+If n_processes < 1, the managing section is subverted, and only the
+handling sequence is executed.
+
+Either returns the return value of pm_die() and/or pm_abort() (which will
+not ever return in general), or returns 1 to the calling script to begin
+handling requests.
 
 =cut
 
@@ -114,7 +197,7 @@ sub pm_manage {
   map { $this->pm_parameter($_,$values{$_}) } keys %values;
 
   # skip to handling now if we won't be managing any processes.
-  $this->n_processes() or goto HANDLING;
+  $this->n_processes() or return;
 
   # call the (possibly overloaded) management initialization hook.
   $this->role("manager");
@@ -125,10 +208,6 @@ sub pm_manage {
 
  MANAGING_LOOP: while (1) {
 
-    # if the calling process goes away, perform cleanup.
-    getppid() == 1 and
-      return $this->pm_die("calling process has died");
-
     $this->n_processes() > 0 or
       return $this->pm_die();
 
@@ -144,7 +223,6 @@ sub pm_manage {
        return $this->pm_abort("fork: $!");
 
       } else {
-       $this->role("server");
        $this->{MANAGER_PID} = $manager_pid;
        # the server exits the managing loop.
        last MANAGING_LOOP;
@@ -154,12 +232,15 @@ sub pm_manage {
     }
 
     # this should block until the next server dies.
-    $this->pm_reap_server();
+    $this->pm_wait();
 
   }# while 1
 
 HANDLING:
 
+  # forget any children we had been collecting.
+  delete $this->{PIDS};
+
   # call the (possibly overloaded) handling init hook
   $this->role("server");
   $this->handling_init();
@@ -171,30 +252,57 @@ HANDLING:
 
 =head2 managing_init
 
+ instance
+ () managing_init()
+
+DESCRIPTION:
+
+Overrideable method which initializes a process manager.  In order to
+handle signals, manage the PID file, and change the process name properly,
+any method which overrides this should call SUPER::managing_init().
+
 =cut
 
 sub managing_init {
-  my ($this) = self_or_default(@_);
+  my ($this) = @_;
 
   # begin to handle signals.
-  $SIG{TERM} = sub { $this->sig_manager(@_) };
-  $SIG{HUP}  = sub { $this->sig_manager(@_) };
+  # We do NOT want SA_RESTART in the process manager.
+  # -- we want start the shutdown sequence immediately upon SIGTERM.
+  unless ($this->no_signals()) {
+    sigaction(SIGTERM, $this->{sigaction_no_sa_restart}) or
+       $this->pm_warn("sigaction: SIGTERM: $!");
+    sigaction(SIGHUP,  $this->{sigaction_no_sa_restart}) or
+       $this->pm_warn("sigaction: SIGHUP: $!");
+    $SIG_CODEREF = sub { $this->sig_manager(@_) };
+  }
 
   # change the name of this process as it appears in ps(1) output.
-  $this->pm_change_process_name("perl-fcgi-pm");
+  $this->pm_change_process_name($this->pm_parameter('pm_title'));
 
   $this->pm_write_pid_file();
 }
 
-
 =head2 pm_die
 
+ instance or export
+ () pm_die(string msg[, int exit_status])
+
+DESCRIPTION:
+
+This method is called when a process manager receives a notification to
+shut itself down.  pm_die() attempts to shutdown the process manager
+gently, sending a SIGTERM to each managed process, waiting die_timeout()
+seconds to reap each process, and then exit gracefully once all children
+are reaped, or to abort if all children are not reaped.
+
 =cut
 
 sub pm_die {
   my ($this,$msg,$n) = self_or_default(@_);
 
   # stop handling signals.
+  undef $SIG_CODEREF;
   $SIG{HUP}  = 'DEFAULT';
   $SIG{TERM} = 'DEFAULT';
 
@@ -202,27 +310,40 @@ sub pm_die {
 
   # prepare to die no matter what.
   if (defined $this->die_timeout()) {
-    $SIG{ARLM} = sub { $this->pm_abort("reap timeout") };
+    $SIG{ALRM} = sub { $this->pm_abort("wait timeout") };
     alarm $this->die_timeout();
   }
 
   # send a TERM to each of the servers.
-  kill "TERM", keys %{$this->{PIDS}};
+  if (my @pids = keys %{$this->{PIDS}}) {
+    $this->pm_notify("sending TERM to PIDs, @pids");
+    kill "TERM", @pids;
+  }
 
   # wait for the servers to die.
   while (%{$this->{PIDS}}) {
-    $this->pm_reap_server();
+    $this->pm_wait();
   }
 
   # die already.
   $this->pm_exit("dying: ".$msg,$n);
 }
 
-=head2 pm_reap_server
+=head2 pm_wait
+
+ instance or export
+ (int pid) pm_wait()
+
+DESCRIPTION:
+
+This calls wait() which suspends execution until a child has exited.
+If the process ID returned by wait corresponds to a managed process,
+pm_notify() is called with the exit status of that process.
+pm_wait() returns with the return value of wait().
 
 =cut
 
-sub pm_reap_server {
+sub pm_wait {
   my ($this) = self_or_default(@_);
 
   # wait for the next server to die.
@@ -231,10 +352,20 @@ sub pm_reap_server {
   # notify when one of our servers have died.
   delete $this->{PIDS}->{$pid} and
     $this->pm_notify("server (pid $pid) exited with status $?");
+
+  return $pid;
 }
 
 =head2 pm_write_pid_file
 
+ instance or export
+ () pm_write_pid_file([string filename])
+
+DESCRIPTION:
+
+Writes current process ID to optionally specified file.  If no filename is
+specified, it uses the value of the C<pid_fname> parameter.
+
 =cut
 
 sub pm_write_pid_file {
@@ -250,6 +381,14 @@ sub pm_write_pid_file {
 
 =head2 pm_remove_pid_file
 
+ instance or export
+ () pm_remove_pid_file()
+
+DESCRIPTION:
+
+Removes optionally specified file.  If no filename is specified, it uses
+the value of the C<pid_fname> parameter.
+
 =cut
 
 sub pm_remove_pid_file {
@@ -259,14 +398,46 @@ sub pm_remove_pid_file {
   return $ret;
 }
 
+=head2 sig_sub
+
+ instance
+ () sig_sub(string name)
+
+DESCRIPTION:
+
+The name of this method is passed to POSIX::sigaction(), and handles signals
+for the process manager.  If $SIG_CODEREF is set, then the input arguments
+to this are passed to a call to that.
+
+=cut
+
+sub sig_sub {
+  $SIG_CODEREF->(@_) if ref $SIG_CODEREF;
+}
+
 =head2 sig_manager
 
+ instance
+ () sig_manager(string name)
+
+DESCRIPTION:
+
+Handles signals of the process manager.  Takes as input the name of signal
+being handled.
+
 =cut
 
 sub sig_manager {
   my ($this,$name) = @_;
-  if ($name eq "TERM" or $name eq "HUP") {
-    $this->pm_die("received signal $name");
+  if ($name eq "TERM") {
+    $this->pm_notify("received signal $name");
+    $this->pm_die("safe exit from signal $name");
+  } elsif ($name eq "HUP") {
+    # send a TERM to each of the servers, and pretend like nothing happened..
+    if (my @pids = keys %{$this->{PIDS}}) {
+      $this->pm_notify("sending TERM to PIDs, @pids");
+      kill "TERM", @pids;
+    }
   } else {
     $this->pm_notify("ignoring signal $name");
   }
@@ -276,14 +447,23 @@ sub sig_manager {
 
 =head2 handling_init
 
+ instance or export
+ () handling_init()
+
+DESCRIPTION:
+
 =cut
 
 sub handling_init {
-  my ($this) = self_or_default(@_);
+  my ($this) = @_;
 
   # begin to handle signals.
-  $SIG{TERM} = sub { $this->sig_handler(@_) };
-  $SIG{HUP}  = sub { $this->sig_handler(@_) };
+  # We'll want accept(2) to return -1(EINTR) on caught signal..
+  unless ($this->no_signals()) {
+    sigaction(SIGTERM, $this->{sigaction_no_sa_restart}) or $this->pm_warn("sigaction: SIGTERM: $!");
+    sigaction(SIGHUP,  $this->{sigaction_no_sa_restart}) or $this->pm_warn("sigaction: SIGHUP: $!");
+    $SIG_CODEREF = sub { $this->sig_handler(@_) };
+  }
 
   # change the name of this process as it appears in ps(1) output.
   $this->pm_change_process_name("perl-fcgi");
@@ -291,14 +471,30 @@ sub handling_init {
 
 =head2 pm_pre_dispatch
 
+ instance or export
+ () pm_pre_dispatch()
+
+DESCRIPTION:
+
 =cut
 
 sub pm_pre_dispatch {
   my ($this) = self_or_default(@_);
+
+  # Now, we want the request to continue unhindered..
+  unless ($this->no_signals()) {
+    sigaction(SIGTERM, $this->{sigaction_sa_restart}) or $this->pm_warn("sigaction: SIGTERM: $!");
+    sigaction(SIGHUP,  $this->{sigaction_sa_restart}) or $this->pm_warn("sigaction: SIGHUP: $!");
+  }
 }
 
 =head2 pm_post_dispatch
 
+ instance or export
+ () pm_post_dispatch()
+
+DESCRIPTION:
+
 =cut
 
 sub pm_post_dispatch {
@@ -309,13 +505,23 @@ sub pm_post_dispatch {
   if ($this->pm_received_signal("HUP")) {
     $this->pm_exit("safe exit after SIGHUP");
   }
-  if (getppid() != $this->{MANAGER_PID}) {
+  if ($this->{MANAGER_PID} and getppid() != $this->{MANAGER_PID}) {
     $this->pm_exit("safe exit: manager has died");
   }
+  # We'll want accept(2) to return -1(EINTR) on caught signal..
+  unless ($this->no_signals()) {
+    sigaction(SIGTERM, $this->{sigaction_no_sa_restart}) or $this->pm_warn("sigaction: SIGTERM: $!");
+    sigaction(SIGHUP,  $this->{sigaction_no_sa_restart}) or $this->pm_warn("sigaction: SIGHUP: $!");
+  }
 }
 
 =head2 sig_handler
 
+ instance or export
+ () sig_handler()
+
+DESCRIPTION:
+
 =cut
 
 sub sig_handler {
@@ -349,6 +555,11 @@ sub self_or_default {
 
 =head2 pm_change_process_name
 
+ instance or export
+ () pm_change_process_name()
+
+DESCRIPTION:
+
 =cut
 
 sub pm_change_process_name {
@@ -358,6 +569,11 @@ sub pm_change_process_name {
 
 =head2 pm_received_signal
 
+ instance or export
+ () pm_received signal()
+
+DESCRIPTION:
+
 =cut
 
 sub pm_received_signal {
@@ -367,8 +583,15 @@ sub pm_received_signal {
   return $this->{SIG_RECEIVED}->{$sig};
 }
 
+=head1 parameters
+
 =head2 pm_parameter
 
+ instance or export
+ () pm_parameter()
+
+DESCRIPTION:
+
 =cut
 
 sub pm_parameter {
@@ -389,6 +612,8 @@ sub pm_parameter {
 
 =head2 start_delay
 
+DESCRIPTION:
+
 =cut
 
 sub n_processes     { shift->pm_parameter("n_processes",     @_); }
@@ -398,8 +623,15 @@ sub die_timeout     { shift->pm_parameter("die_timeout",     @_); }
 sub role            { shift->pm_parameter("role",            @_); }
 sub start_delay     { shift->pm_parameter("start_delay",     @_); }
 
+=head1 notification and death
+
 =head2 pm_warn
 
+ instance or export
+ () pm_warn()
+
+DESCRIPTION:
+
 =cut
 
 sub pm_warn {
@@ -409,6 +641,11 @@ sub pm_warn {
 
 =head2 pm_notify
 
+ instance or export
+ () pm_notify()
+
+DESCRIPTION:
+
 =cut
 
 sub pm_notify {
@@ -417,13 +654,23 @@ sub pm_notify {
   print STDERR "FastCGI: ".$this->role()." (pid $$): ".$msg;
 }
 
-=head2 exit
+=head2 pm_exit
+
+ instance or export
+ () pm_exit(string msg[, int exit_status])
+
+DESCRIPTION:
 
 =cut
 
 sub pm_exit {
   my ($this,$msg,$n) = self_or_default(@_);
   $n ||= 0;
+
+  # if we still have children at this point, something went wrong.
+  # SIGKILL them now.
+  kill "KILL", keys %{$this->{PIDS}} if $this->{PIDS};
+
   $this->pm_warn($msg);
   $@ = $msg;
   exit $n;
@@ -431,6 +678,11 @@ sub pm_exit {
 
 =head2 pm_abort
 
+ instance or export
+ () pm_abort(string msg[, int exit_status])
+
+DESCRIPTION:
+
 =cut
 
 sub pm_abort {
@@ -450,6 +702,14 @@ No known bugs, but this does not mean no bugs exist.
 
 L<FCGI>.
 
+=head1 MAINTAINER
+
+Gareth Kirwan <gbjk@thermeon.com>
+
+=head1 AUTHOR
+
+James E Jurach Jr.
+
 =head1 COPYRIGHT
 
  FCGI-ProcManager - A Perl FCGI Process Manager