Add psgix.harakiri support to FCGI::Engine::PSGI

lib/FCGI/Engine/Core.pm

@@ -192,16 +192,24 @@ sub run {
     }
 
     while ($request->Accept() >= 0) {
+        $self->process_request(
+            $self->prepare_environment($env, $proc_manager),
+            $request,
+            $proc_manager
+        );
+    }
+}
 
-        $proc_manager && $proc_manager->pre_dispatch;
+sub process_request {
+    my ($self, $env, $request, $proc_manager) = @_;
 
-        $self->handle_request(
-            $self->prepare_environment( $env ),
-            $request
-        );
+    $proc_manager && $proc_manager->pre_dispatch;
 
-        $proc_manager && $proc_manager->post_dispatch;
-    }
+    $self->handle_request($env, $request);
+
+    $request->Finish();
+
+    $proc_manager && $proc_manager->post_dispatch;
 }
 
 1;

lib/FCGI/Engine/PSGI.pm

@@ -21,7 +21,7 @@ has 'app' => (
 # - SL
 
 augment 'prepare_environment' => sub {
-    my ($self, $env) = @_;
+    my ($self, $env, $proc_manager) = @_;
     return +{
         %$env,
         'psgi.version'      => [1,0],
@@ -34,6 +34,8 @@ augment 'prepare_environment' => sub {
         'psgi.run_once'     => Plack::Util::FALSE,
         'psgi.streaming'    => Plack::Util::TRUE,
         'psgi.nonblocking'  => Plack::Util::FALSE,
+
+        'psgix.harakiri'    => defined $proc_manager,
     };
 };
 
@@ -83,6 +85,16 @@ sub _handle_response {
     }
 }
 
+around 'process_request' => sub {
+    my ($orig, $self, $env, $request, $proc_manager) = @_;
+
+    $self->$orig($env, $request);
+
+    if ($proc_manager && $env->{'psgix.harakiri.commit'}) {
+        $proc_manager->exit("safe exit with harakiri");
+    }
+};
+
 __PACKAGE__->meta->make_immutable;
 
 no Moose; 1;

t/201_fcgi_engine_psgi.t

@@ -0,0 +1,45 @@
+#!/usr/bin/perl
+
+use strict;
+use warnings;
+use FindBin;
+
+use Test::More;
+
+BEGIN {
+    {
+        local $@;
+        eval "use Plack 0.9910; use FCGI::Client 0.06; use MooseX::NonMoose 0.07; use IO::String; use Plack::App::FCGIDispatcher;";
+        plan skip_all => "Plack 0.9910, FCGI::Client 0.06, MooseX::NonMoose 0.07 and Plack::App::FCGIDispatcher are required for this test" if $@;
+    }
+}
+
+use FCGI::Engine::PSGI;
+use Test::TCP;
+use Plack::Test::Suite;
+use t::lib::FCGIUtils;
+
+my $http_port;
+my $fcgi_port;
+
+test_fcgi_standalone(
+   sub {
+       ($http_port, $fcgi_port) = @_;
+       Plack::Test::Suite->run_server_tests(\&run_server, $fcgi_port, $http_port);
+       done_testing();
+    }
+);
+
+sub run_server {
+    my($port, $app) = @_;
+
+    $| = 0; # Test::Builder autoflushes this. reset!
+
+    my $server = FCGI::Engine::PSGI->new(
+        listen      => "127.0.0.1:$port",
+        pidfile     => '/tmp/101_plack_server_fcgi_engine_client.pid',
+        app         => $app,
+    );
+    $server->run();
+}
+