From 744facc2333246bc91293c81cde86fa65bcdd65a Mon Sep 17 00:00:00 2001
From: freek <freek@spatie.be>
Date: Tue, 27 Nov 2018 21:53:50 +0100
Subject: [PATCH] commit

---
 src/Console/StartWebSocketServer.php  |  4 +++-
 src/Server/WebSocketServerFactory.php | 16 +++++++---------
 2 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/src/Console/StartWebSocketServer.php b/src/Console/StartWebSocketServer.php
index da065f1..4338b02 100644
--- a/src/Console/StartWebSocketServer.php
+++ b/src/Console/StartWebSocketServer.php
@@ -74,10 +74,12 @@ class StartWebSocketServer extends Command
         $routes = WebSocketRouter::getRoutes();
 
         /** 🎩 Start the magic 🎩 */
-        (new WebSocketServerFactory($routes))
+        (new WebSocketServerFactory())
+            ->useRoutes($routes)
             ->setHost($this->option('host'))
             ->setPort($this->option('port'))
             ->setConsoleOutput($this->output)
+            ->createServer()
             ->run();
     }
 }
diff --git a/src/Server/WebSocketServerFactory.php b/src/Server/WebSocketServerFactory.php
index 122a941..8348256 100644
--- a/src/Server/WebSocketServerFactory.php
+++ b/src/Server/WebSocketServerFactory.php
@@ -31,11 +31,16 @@ class WebSocketServerFactory
     /** @var Symfony\Component\Console\Output\OutputInterface */
     protected $consoleOutput;
 
-    public function __construct(RouteCollection $routes)
+    public function __construct()
     {
         $this->loop = LoopFactory::create();
+    }
 
+    public function useRoutes(RouteCollection $routes)
+    {
         $this->routes = $routes;
+
+        return $this;
     }
 
     public function setHost(string $host)
@@ -66,14 +71,7 @@ class WebSocketServerFactory
         return $this;
     }
 
-    public function run()
-    {
-        $server = $this->createServer();
-
-        $server->run();
-    }
-
-    protected function createServer(): IoServer
+    public function createServer(): IoServer
     {
         $socket = new Server("{$this->host}:{$this->port}", $this->loop);