Added flag shutdown functionality to command

Author: AndrewCarterUK

src/Command/DaemonRunCommand.php

@@ -27,6 +27,11 @@ class DaemonRunCommand extends Command
      */
     private $driverContainer;
 
+    /**
+     * @var DaemonInterface
+     */
+    private $daemon;
+
     /**
      * Constructor.
      *
@@ -39,6 +44,7 @@ public function __construct(KernelInterface $kernel, DriverContainerInterface $d
     {
         $this->kernel          = $kernel;
         $this->driverContainer = $driverContainer;
+        $this->daemon          = null;
 
         $name        = $name        ?: self::DEFAULT_NAME;
         $description = $description ?: self::DEFAULT_DESCRIPTION;
@@ -97,15 +103,27 @@ protected function execute(InputInterface $input, OutputInterface $output)
 
         if (null !== $port) {
             // If we have the port, create a TCP daemon
-            $daemon = $daemonFactory->createTcpDaemon($this->kernel, $daemonOptions, $host ?: 'localhost', $port);
+            $this->daemon = $daemonFactory->createTcpDaemon($this->kernel, $daemonOptions, $host ?: 'localhost', $port);
         } elseif (null !== $host) {
             // If we have the host but not the port, we cant create a TCP daemon - throw exception
             throw new \InvalidArgumentException('TCP port option must be set if host option is set');
         } else {
             // With no host or port, listen on FCGI_LISTENSOCK_FILENO (default)
-            $daemon = $daemonFactory->createDaemon($this->kernel, $daemonOptions, $fd);
+            $this->daemon = $daemonFactory->createDaemon($this->kernel, $daemonOptions, $fd);
+        }
+
+        $this->daemon->run();
+    }
+
+    /**
+     * Flag the daemon for shutdown.
+     */
+    public function flagShutdown()
+    {
+        if (null === $this->daemon) {
+            throw new \RuntimeException('There is no daemon running');
         }
 
-        $daemon->run();
+        $this->daemon->flagShutdown();
     }
 }

test/Command/DaemonRunCommandTest.php

@@ -129,6 +129,35 @@ public function testPortOptionOnly()
         $this->assertEquals('getFactory',      $context['driverContainer']->getDelegatedCalls()[0][0]);
     }
 
+    /**
+     * Test flag shutdown.
+     */
+    public function testShutdown()
+    {
+        $expectations = ['driver' => 'userland'];
+
+        $context = $this->createTestingContext($expectations);
+
+        $context['command']->run(new ArrayInput([]), new NullOutput());
+        $context['command']->flagShutdown();
+
+        $this->assertEquals('run',          $context['daemon']->getDelegatedCalls()[0][0]);
+        $this->assertEquals('flagShutdown', $context['daemon']->getDelegatedCalls()[1][0]);
+        $this->assertEquals('createDaemon', $context['daemonFactory']->getDelegatedCalls()[0][0]);
+        $this->assertEquals('getFactory',   $context['driverContainer']->getDelegatedCalls()[0][0]);
+    }
+
+    /**
+     * Test flag shutdown with no daemon.
+     * 
+     * @expectedException \RuntimeException
+     */
+    public function testShutdownNoDaemon()
+    {
+        $command = new DaemonRunCommand(new MockKernel(), new MockDriverContainer());
+        $command->flagShutdown();
+    }
+
     /**
      * Construct mock objects set to test for expected values of different
      * parameters when specified.
@@ -146,7 +175,7 @@ private function createTestingContext(array $expectations)
         };
 
         $mockKernel = new MockKernel();
-        $mockDaemon = new MockDaemon(['run' => false]);
+        $mockDaemon = new MockDaemon(['run' => false, 'flagShutdown' => false]);
 
         $mockDaemonFactory = new MockDaemonFactory([
             'createTcpDaemon' => function ($kernel, $options, $host, $port) use ($assertExpected, $mockKernel, $mockDaemon) {

test/Helper/Mocker/MockDaemon.php

@@ -15,5 +15,6 @@ public function run()
 
     public function flagShutdown()
     {
+        return $this->delegateCall('flagShutdown', func_get_args());
     }
 }