Added HTTP chunk support for hyperf/http-message (#4290)

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

Author: 3 people

src/Server/Chunk/Chunkable.php

@@ -0,0 +1,17 @@
+<?php
+
+declare(strict_types=1);
+/**
+ * This file is part of Hyperf.
+ *
+ * @link     https://www.hyperf.io
+ * @document https://hyperf.wiki
+ * @contact  [email protected]
+ * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
+ */
+namespace Hyperf\HttpMessage\Server\Chunk;
+
+interface Chunkable
+{
+    public function write(string $data): bool;
+}

src/Server/Chunk/HasChunk.php

@@ -0,0 +1,24 @@
+<?php
+
+declare(strict_types=1);
+/**
+ * This file is part of Hyperf.
+ *
+ * @link     https://www.hyperf.io
+ * @document https://hyperf.wiki
+ * @contact  [email protected]
+ * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
+ */
+namespace Hyperf\HttpMessage\Server\Chunk;
+
+trait HasChunk
+{
+    public function write(string $content): bool
+    {
+        if (isset($this->connection) && $this->connection instanceof Chunkable) {
+            return $this->connection->write($content);
+        }
+
+        return false;
+    }
+}

src/Server/Connection/SwooleConnection.php

@@ -0,0 +1,34 @@
+<?php
+
+declare(strict_types=1);
+/**
+ * This file is part of Hyperf.
+ *
+ * @link     https://www.hyperf.io
+ * @document https://hyperf.wiki
+ * @contact  [email protected]
+ * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
+ */
+namespace Hyperf\HttpMessage\Server\Connection;
+
+use Hyperf\HttpMessage\Server\Chunk\Chunkable;
+use Hyperf\HttpMessage\Server\ConnectionInterface;
+use Swoole\Http\Response;
+
+class SwooleConnection implements ConnectionInterface, Chunkable
+{
+    /**
+     * @var Response
+     */
+    protected $response;
+
+    public function __construct(Response $response)
+    {
+        $this->response = $response;
+    }
+
+    public function write(string $data): bool
+    {
+        return $this->response->write($data);
+    }
+}

src/Server/ConnectionInterface.php

@@ -0,0 +1,16 @@
+<?php
+
+declare(strict_types=1);
+/**
+ * This file is part of Hyperf.
+ *
+ * @link     https://www.hyperf.io
+ * @document https://hyperf.wiki
+ * @contact  [email protected]
+ * @license  https://github.com/hyperf/hyperf/blob/master/LICENSE
+ */
+namespace Hyperf\HttpMessage\Server;
+
+interface ConnectionInterface
+{
+}

src/Server/Response.php

@@ -12,10 +12,14 @@
 namespace Hyperf\HttpMessage\Server;
 
 use Hyperf\HttpMessage\Cookie\Cookie;
+use Hyperf\HttpMessage\Server\Chunk\Chunkable;
+use Hyperf\HttpMessage\Server\Chunk\HasChunk;
 use Hyperf\HttpMessage\Stream\SwooleStream;
 
-class Response extends \Hyperf\HttpMessage\Base\Response
+class Response extends \Hyperf\HttpMessage\Base\Response implements Chunkable
 {
+    use HasChunk;
+
     /**
      * @var array
      */
@@ -26,6 +30,11 @@ class Response extends \Hyperf\HttpMessage\Base\Response
      */
     protected $trailers = [];
 
+    /**
+     * @var null|ConnectionInterface
+     */
+    protected $connection;
+
     /**
      * Returns an instance with body content.
      */
@@ -80,4 +89,15 @@ public function getTrailers(): array
     {
         return $this->trailers;
     }
+
+    public function setConnection(ConnectionInterface $connection)
+    {
+        $this->connection = $connection;
+        return $this;
+    }
+
+    public function getConnection(): ?ConnectionInterface
+    {
+        return $this->connection;
+    }
 }

tests/ResponseProxyTest.php

@@ -40,6 +40,11 @@ public function testCookies()
         parent::testCookies();
     }
 
+    public function testWrite()
+    {
+        $this->markTestSkipped('Response proxy does not support chunk.');
+    }
+
     protected function newResponse()
     {
         $response = new ResponseStub();

tests/ResponseTest.php

@@ -12,8 +12,11 @@
 namespace HyperfTest\HttpMessage;
 
 use Hyperf\HttpMessage\Cookie\Cookie;
+use Hyperf\HttpMessage\Server\Connection\SwooleConnection;
 use Hyperf\HttpMessage\Server\Response;
+use Mockery;
 use PHPUnit\Framework\TestCase;
+use Swoole\Http\Response as SwooleResponse;
 
 /**
  * @internal
@@ -52,6 +55,18 @@ public function testCookies()
         $this->assertSame(['hyperf.io' => ['/' => ['test' => $cookie]]], $response->getCookies());
     }
 
+    public function testWrite()
+    {
+        $content = 'hello';
+        $swooleResponse = Mockery::mock(SwooleResponse::class);
+        $swooleResponse->shouldReceive('write')->with($content)->once()->andReturn(true);
+
+        $response = $this->newResponse();
+        $response->setConnection(new SwooleConnection($swooleResponse));
+        $status = $response->write($content);
+        $this->assertTrue($status);
+    }
+
     protected function newResponse()
     {
         return new Response();