Integrated shutdown exception to be triggered following SIGINT and after the connection pool had been shutdown

Author: AndrewCarterUK

src/Daemon.php

@@ -4,6 +4,7 @@
 
 use PHPFastCGI\FastCGIDaemon\Connection\ConnectionPoolInterface;
 use PHPFastCGI\FastCGIDaemon\ConnectionHandler\ConnectionHandlerFactoryInterface;
+use PHPFastCGI\FastCGIDaemon\Exception\ShutdownException;
 use Psr\Log\LoggerAwareInterface;
 use Psr\Log\LoggerAwareTrait;
 use Psr\Log\LoggerInterface;
@@ -55,18 +56,24 @@ public function __construct(ConnectionPoolInterface $connectionPool, ConnectionH
      */
     public function run()
     {
+        declare(ticks = 1);
+
+        pcntl_signal(SIGINT, function () {
+            $this->connectionPool->shutdown();
+            throw new ShutdownException('Received SIGINT, shutting down...');
+        });
+
         try {
             while (1) {
                 $this->connectionPool->operate($this->connectionHandlerFactory, 5);
                 // @codeCoverageIgnoreStart
             }
             // @codeCoverageIgnoreEnd
+        } catch (ShutdownException $exception) {
+            $this->logger->notice($exception->getMessage());
         } catch (\RuntimeException $exception) {
             $this->logger->emergency($exception->getMessage());
             throw $exception;
         }
-
-        // @codeCoverageIgnoreStart
     }
-    // @codeCoverageIgnoreEnd
 }

test/DaemonTest.php

@@ -30,7 +30,6 @@ public function testRun()
 
             throw new \RuntimeException('foo');
         };
-
         $mockConnectionPool = new CallableConnectionPool($connectionPoolCallback);
 
         // Create daemon
@@ -51,4 +50,31 @@ public function testRun()
             $this->assertEquals('foo',       $messages[0]['message']);
         }
     }
+
+    /**
+     * Tests that the daemon shuts down cleanly after a SIGINT
+     */
+    public function testInterruptSignal()
+    {
+        $logger = new InMemoryLogger;
+
+        // Create mock connection handler factory
+        $mockConnectionHandlerFactory = new CallableConnectionHandlerFactory(function () { });
+
+        // Create mock connection pool that sends a SIGINT to itself immediately
+        $connectionPoolCallback = function () {
+            posix_kill(posix_getpid(), SIGINT);
+        };
+        $mockConnectionPool = new CallableConnectionPool($connectionPoolCallback);
+
+        // Create daemon
+        $daemon = new Daemon($mockConnectionPool, $mockConnectionHandlerFactory, $logger);
+
+        $daemon->run();
+
+        // Check that the daemon exits and the correct message is logged
+        $messages = $logger->getMessages();
+        $this->assertEquals('notice',                            $messages[0]['level']);
+        $this->assertEquals('Received SIGINT, shutting down...', $messages[0]['message']);
+    }
 }