Merge pull request #24 from psadaic/master

enable ws server keep alive

src/Console/Commands/RatchetServerCommand.php

@@ -61,6 +61,13 @@ class RatchetServerCommand extends Command
      */
     protected $driver;
 
+    /**
+     * Keep alive interval.
+     *
+     * @var int
+     */
+    protected $keepAlive;
+
     /**
      * The ReactPHP event loop.
      *
@@ -80,6 +87,11 @@ class RatchetServerCommand extends Command
      */
     protected $ratchetServer;
 
+    /**
+     * WebSocket server instance.
+     */
+    protected $wsServerInstance;
+
     /**
      * Get the console command options.
      *
@@ -93,6 +105,7 @@ protected function getOptions()
             ['class', null, InputOption::VALUE_OPTIONAL, 'Class that implements MessageComponentInterface.', config('ratchet.class')],
             ['driver', null, InputOption::VALUE_OPTIONAL, 'Ratchet connection driver [IoServer|WsServer|WampServer]', 'WampServer'],
             ['zmq', 'z', null, 'Bind server to a ZeroMQ socket (always on for WampServer)'],
+            ['keepAlive', null, InputOption::VALUE_OPTIONAL, 'Option to enable WebSocket server keep alive [interval in seconds]', config('ratchet.keepAlive', 0)],
         ];
     }
 
@@ -109,6 +122,8 @@ public function handle()
 
         $this->driver = $this->option('driver');
 
+        $this->keepAlive = $this->option('keepAlive');
+
         $this->startServer();
     }
 
@@ -165,10 +180,16 @@ private function bootWebSocketServer($withZmq = false)
             $this->bootZmqConnection();
         }
 
+        $this->wsServerInstance = new WsServer($this->serverInstance);
+
         $this->serverInstance = new HttpServer(
-            new WsServer($this->serverInstance)
+            $this->wsServerInstance
         );
 
+        if($this->keepAlive > 0){
+            $this->wsServerInstance->enableKeepAlive($this->getEventLoop(), $this->keepAlive);
+        }
+
         return $this->bootIoServer();
     }