Add profiling option argument

Author: prophet777

Command/ServerCommand.php

@@ -4,6 +4,7 @@
 
 use Symfony\Component\Console\Input\InputArgument;
 use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Input\InputOption;
 use Symfony\Component\Console\Output\OutputInterface;
 
 /**
@@ -18,7 +19,8 @@ protected function configure()
         $this
             ->setName('gos:server')
             ->setDescription('Starts the web socket servers')
-            ->addArgument('name', InputArgument::OPTIONAL, 'Server name');
+            ->addArgument('name', InputArgument::OPTIONAL, 'Server name')
+            ->addOption('profile', 'p', InputOption::VALUE_NONE, 'Profiling server');
     }
 
     /**

Command/WebsocketServerCommand.php

@@ -8,6 +8,7 @@
 use Symfony\Component\Console\Command\Command;
 use Symfony\Component\Console\Input\InputArgument;
 use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Input\InputOption;
 use Symfony\Component\Console\Output\OutputInterface;
 
 class WebsocketServerCommand extends Command
@@ -39,7 +40,8 @@ protected function configure()
         $this
             ->setName('gos:websocket:server')
             ->setDescription('Starts the web socket servers')
-            ->addArgument('name', InputArgument::OPTIONAL, 'Server name');
+            ->addArgument('name', InputArgument::OPTIONAL, 'Server name')
+            ->addOption('profile', 'p', InputOption::VALUE_NONE, 'Profiling server');
     }
 
     /**
@@ -48,6 +50,6 @@ protected function configure()
      */
     protected function execute(InputInterface $input, OutputInterface $output)
     {
-        $this->entryPoint->launch($input->getArgument('name'));
+        $this->entryPoint->launch($input->getArgument('name'), $input->getOption('profile'));
     }
 }

DependencyInjection/CompilerPass/DevCompilerPass.php

@@ -2,7 +2,6 @@
 
 namespace Gos\Bundle\WebSocketBundle;
 
-use Gos\Bundle\WebSocketBundle\DependencyInjection\CompilerPass\DevCompilerPass;
 use Gos\Bundle\WebSocketBundle\DependencyInjection\CompilerPass\PeriodicCompilerPass;
 use Gos\Bundle\WebSocketBundle\DependencyInjection\CompilerPass\PingableDriverCompilerPass;
 use Gos\Bundle\WebSocketBundle\DependencyInjection\CompilerPass\RpcCompilerPass;
@@ -26,6 +25,5 @@ public function build(ContainerBuilder $container)
         $container->addCompilerPass(new TopicCompilerPass());
         $container->addCompilerPass(new PeriodicCompilerPass());
         $container->addCompilerPass(new PingableDriverCompilerPass());
-        $container->addCompilerPass(new DevCompilerPass());
     }
 }

GosWebSocketBundle.php

@@ -25,7 +25,7 @@ public function __construct(LoggerInterface $logger)
      */
     public function tick()
     {
-        $this->logger->debug('Memory usage : ' . round((memory_get_usage() / (1024 * 1024)), 2) . 'Mo');
+        $this->logger->info('Memory usage : ' . round((memory_get_usage() / (1024 * 1024)), 4) . 'Mo');
     }
 
     /**

Periodic/PeriodicMemoryUsage.php

@@ -24,9 +24,10 @@ public function __construct(ServerRegistry $serverRegistry)
     }
 
     /**
-     * Launches the relevant servers needed by Gos WebSocket.
+     * @param string $serverName
+     * @param bool   $profile
      */
-    public function launch($serverName)
+    public function launch($serverName, $profile)
     {
         $servers = $this->serverRegistry->getServers();
 
@@ -45,6 +46,6 @@ public function launch($serverName)
             $server = $servers[$serverName];
         }
 
-        $server->launch();
+        $server->launch($profile);
     }
 }

Server/EntryPoint.php

@@ -7,7 +7,7 @@ interface ServerInterface
     /**
      * Launches the server loop.
      */
-    public function launch();
+    public function launch($profile);
 
     /**
      * Returns a string of the host:port for debugging / display purposes.

Server/Type/ServerInterface.php

@@ -5,6 +5,7 @@
 use Gos\Bundle\WebSocketBundle\Event\Events;
 use Gos\Bundle\WebSocketBundle\Event\ServerEvent;
 use Gos\Bundle\WebSocketBundle\Periodic\PeriodicInterface;
+use Gos\Bundle\WebSocketBundle\Periodic\PeriodicMemoryUsage;
 use Gos\Bundle\WebSocketBundle\Server\App\Registry\OriginRegistry;
 use Gos\Bundle\WebSocketBundle\Server\App\Registry\PeriodicRegistry;
 use Gos\Bundle\WebSocketBundle\Server\App\WampApplication;
@@ -107,7 +108,7 @@ public function setSessionHandler(\SessionHandlerInterface $sessionHandler)
         $this->sessionHandler = $sessionHandler;
     }
 
-    public function launch()
+    public function launch($profile)
     {
         $this->logger->info('Starting web socket');
 
@@ -119,6 +120,11 @@ public function launch()
         $server = new Server($loop);
         $server->listen($this->port, $this->host);
 
+        if (true === $profile) {
+            $memoryUsagePeriodicTimer = new PeriodicMemoryUsage($this->logger);
+            $this->periodicRegistry->addPeriodic($memoryUsagePeriodicTimer);
+        }
+
         /** @var PeriodicInterface $periodic */
         foreach ($this->periodicRegistry->getPeriodics() as $periodic) {
             $loop->addPeriodicTimer($periodic->getTimeout(), [$periodic, 'tick']);