Allow websocket URL to be defined during instantiation in WebsocketRe…

…source
polygon-io · Nov 29, 2023 · dd9b57c · dd9b57c
1 parent 577dfdf
commit dd9b57c
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/src/Websockets/WebsocketResource.php b/src/Websockets/WebsocketResource.php
@@ -18,16 +18,30 @@ class WebsocketResource
      */
     protected $topic;
 
+    /**
+     * @var string
+     */
+    protected $socketUrl;
+
     /**
      * WebsocketResource constructor.
      *
      * @param  string  $topic
      * @param  string  $apiKey
      */
-    public function __construct(string $topic, string $apiKey)
+    public function __construct(string $topic, string $apiKey, string $socketUrl = self::SOCKET_URI)
     {
         $this->apiKey = $apiKey;
         $this->topic = $topic;
+        $this->socketUrl = $socketUrl;
+    }
+
+    /**
+     * @param  string  $socketUrl
+     */
+    public function setSocketUrl(string $socketUrl)
+    {
+        $this->socketUrl = $socketUrl;
     }
 
     /**
@@ -41,7 +55,7 @@ function () use ($onMessageCallback, $subscriptions) {
                 /**
                 * @var Websocket\Connection $connection
                 */
-                $connection = yield Websocket\connect(self::SOCKET_URI . '/' . $this->topic);
+                $connection = yield Websocket\connect($this->socketUrl . '/' . $this->topic);
                 yield $connection->send('{"action":"auth", "params":"' . $this->apiKey . '"}');
                 yield $connection->send(
                     json_encode(