Support to set headers for websocket-client.

Co-authored-by: 李铭昕 <[email protected]>

src/Client.php

@@ -22,7 +22,7 @@ class Client
 {
     protected Coroutine\Http\Client $client;
 
-    public function __construct(protected UriInterface $uri)
+    public function __construct(protected UriInterface $uri, array $headers = [])
     {
         $host = $uri->getHost();
         $port = $uri->getPort();
@@ -33,7 +33,7 @@ public function __construct(protected UriInterface $uri)
         }
 
         $this->client = new Coroutine\Http\Client($host, $port, $ssl);
-
+        $headers && $this->client->setHeaders($headers);
         parse_str($this->uri->getQuery(), $query);
 
         $query = http_build_query($query);

src/ClientFactory.php

@@ -20,12 +20,12 @@
 
 class ClientFactory
 {
-    public function create(string $uri, bool $autoClose = true): Client
+    public function create(string $uri, bool $autoClose = true, array $headers = []): Client
     {
         if (! Str::startsWith($uri, ['ws://', 'wss://'])) {
             $uri = 'ws://' . $uri;
         }
-        $client = make(Client::class, ['uri' => new Uri($uri)]);
+        $client = make(Client::class, ['uri' => new Uri($uri), 'headers' => $headers]);
         if ($autoClose) {
             defer(function () use ($client) {
                 $client->close();

tests/ClientTest.php

@@ -12,6 +12,7 @@
 
 namespace HyperfTest\WebSocketClient;
 
+use Hyperf\Codec\Json;
 use Hyperf\HttpMessage\Uri\Uri;
 use Hyperf\WebSocketClient\Client;
 use Hyperf\WebSocketClient\Exception\ConnectException;
@@ -31,4 +32,29 @@ public function testClientConnectFailed()
 
         new Client(new Uri('ws://172.168.1.1:9522'));
     }
+
+    public function testClientConnected()
+    {
+        $client = new Client(new Uri('ws://127.0.0.1:10002/ws'));
+
+        $client->push('ping');
+
+        $this->assertSame('pong', $client->recv(1)->data);
+
+        $client->close();
+    }
+
+    public function testClientHeaders()
+    {
+        $client = new Client(new Uri('ws://127.0.0.1:10002/ws'), ['x-token' => $token = uniqid()]);
+
+        $client->push('headers');
+
+        $data = $client->recv(1);
+        $headers = Json::decode($data->data);
+
+        $this->assertSame($token, $headers['x-token']);
+
+        $client->close();
+    }
 }