Add --connections-limit option

ostrolucky · ostrolucky · commit 9c7cb8f0f913 · 2019-01-28T21:29:53.000+01:00
diff --git a/bin/stdinho b/bin/stdinho
@@ -22,6 +22,7 @@ if (function_exists('pcntl_signal')) {
 }
 
 $application = new Application(APP_NAME, PrettyVersions::getVersion('ostrolucky/'.APP_NAME));
+$application->setAutoExit(false);
 $application->add($command = new Command(APP_NAME));
 $application
     ->setDefaultCommand($command->getName(), true)
diff --git a/src/Command.php b/src/Command.php
@@ -44,6 +44,13 @@ protected function configure(): void
                 'Custom HTTP headers to append to response (in JSON format). Eg. --http-headers=\'["Content-encoding: gzip"]\'',
                 '[]'
             )
+            ->addOption(
+                'connections-limit',
+                null,
+                InputOption::VALUE_REQUIRED,
+                'Determines after how many client connections should program shut down',
+                INF
+            )
             ->setDescription('Turn any STDIN/STDOUT into HTTP server')
         ;
     }
@@ -81,26 +88,24 @@ protected function initialize(InputInterface $input, OutputInterface $output): v
      */
     protected function execute(InputInterface $input, OutputInterface $output): int
     {
-        $addressPort = $input->getArgument('addressPort');
-        $filePath = $input->getOption('file');
-
         $logger = new ConsoleLogger($firstSection = $output->section());
 
         $bufferer = $this->hasStdin ?
-            new PipeBufferer($logger, STDIN, $filePath, $output->section()) :
-            new ResolvedBufferer($filePath)
+            new PipeBufferer($logger, STDIN, $input->getOption('file'), $output->section()) :
+            new ResolvedBufferer($input->getOption('file'))
         ;
 
         $bufferHandler = asyncCoroutine($bufferer);
         $clientHandler = asyncCoroutine(new Responder($logger, $bufferer, $output, $this->customHttpHeaders));
 
-        Loop::run(function () use ($addressPort, $clientHandler, $logger, $firstSection, $bufferHandler) {
+        Loop::run(function () use ($input, $clientHandler, $logger, $firstSection, $bufferHandler) {
             $bufferHandler();
-            $server = listen($addressPort);
+            $server = listen($input->getArgument('addressPort'));
             $firstSection->writeln(
                 "<info>Connection opened at http://{$server->getAddress()}\nPress CTRL+C to exit.</info>\n"
             );
-            while ($socket = yield $server->accept()) {
+            $connectionsLimit = $input->getOption('connections-limit');
+            while ($connectionsLimit-- && ($socket = yield $server->accept())) {
                 $clientHandler($socket);
             }
         });

Original file line number	Diff line number	Diff line change
`@@ -22,6 +22,7 @@ if (function_exists('pcntl_signal')) {`
`22`	`22`	`}`
`23`	`23`
`24`	`24`	`$application = new Application(APP_NAME, PrettyVersions::getVersion('ostrolucky/'.APP_NAME));`
	`25`	`+$application->setAutoExit(false);`
`25`	`26`	`$application->add($command = new Command(APP_NAME));`
`26`	`27`	`$application`
`27`	`28`	`->setDefaultCommand($command->getName(), true)`