Add Gzip support on chunk

Author: Zachary He

src/Server/Manager.php

@@ -225,7 +225,7 @@ public function onRequest($swooleRequest, $swooleResponse)
             $illuminateResponse = $sandbox->run($illuminateRequest);
 
             // send response
-            Response::make($illuminateResponse, $swooleResponse)->send();
+            Response::make($illuminateResponse, $swooleResponse, $swooleRequest)->send();
         } catch (Throwable $e) {
             try {
                 $exceptionResponse = $this->app
@@ -234,7 +234,7 @@ public function onRequest($swooleRequest, $swooleResponse)
                         $illuminateRequest,
                         $this->normalizeException($e)
                     );
-                Response::make($exceptionResponse, $swooleResponse)->send();
+                Response::make($exceptionResponse, $swooleResponse, $swooleRequest)->send();
             } catch (Throwable $e) {
                 $this->logServerError($e);
             }

src/Transformers/Response.php

@@ -4,6 +4,7 @@
 
 use Illuminate\Http\Response as IlluminateResponse;
 use Swoole\Http\Response as SwooleResponse;
+use Swoole\Http\Request as SwooleRequest;
 use Symfony\Component\HttpFoundation\BinaryFileResponse;
 use Symfony\Component\HttpFoundation\Response as SymfonyResponse;
 use Symfony\Component\HttpFoundation\StreamedResponse;
@@ -17,6 +18,11 @@ class Response
      */
     protected $swooleResponse;
 
+    /**
+     * @var \Swoole\Http\Request
+     */
+    protected $swooleRequest;
+
     /**
      * @var \Illuminate\Http\Response
      */
@@ -27,24 +33,27 @@ class Response
      *
      * @param $illuminateResponse
      * @param \Swoole\Http\Response $swooleResponse
+     * @param \Swoole\Http\Request $swooleRequest
      *
      * @return \SwooleTW\Http\Transformers\Response
      */
-    public static function make($illuminateResponse, SwooleResponse $swooleResponse)
+    public static function make($illuminateResponse, SwooleResponse $swooleResponse, SwooleRequest $swooleRequest)
     {
-        return new static($illuminateResponse, $swooleResponse);
+        return new static($illuminateResponse, $swooleResponse, $swooleRequest);
     }
 
     /**
      * Response constructor.
      *
      * @param mixed $illuminateResponse
      * @param \Swoole\Http\Response $swooleResponse
+     * @param \Swoole\Http\Request $swooleRequest
      */
-    public function __construct($illuminateResponse, SwooleResponse $swooleResponse)
+    public function __construct($illuminateResponse, SwooleResponse $swooleResponse, SwooleRequest $swooleRequest)
     {
         $this->setIlluminateResponse($illuminateResponse);
         $this->setSwooleResponse($swooleResponse);
+        $this->setSwooleRequest($swooleRequest);
     }
 
     /**
@@ -133,6 +142,12 @@ protected function sendInChunk($content)
             return;
         }
 
+        // Swoole Chunk mode does not support compress by default, this patch only supports gzip
+        if ($chunkGzip) {
+            $this->swooleResponse->header('Content-Encoding', 'gzip');
+            $content = gzencode($content, config('swoole_http.server.options.http_compression_level', 3));
+        }
+
         foreach (str_split($content, static::CHUNK_SIZE) as $chunk) {
             $this->swooleResponse->write($chunk);
         }
@@ -184,4 +199,37 @@ public function getIlluminateResponse()
     {
         return $this->illuminateResponse;
     }
+
+    /**
+     * @param \Swoole\Http\Request $swooleRequest
+     *
+     * @return \SwooleTW\Http\Transformers\Response
+     */
+    protected function setSwooleRequest(SwooleRequest $swooleRequest)
+    {
+        $this->swooleRequest = $swooleRequest;
+
+        return $this;
+    }
+
+    /**
+     * @return \Swoole\Http\Request
+     */
+    public function getSwooleRequest()
+    {
+        return $this->swooleRequest;
+    }
+
+    /**
+     * @param string $responseContentEncoding
+     * @return bool
+     */
+    protected function canGzipContent($responseContentEncoding)
+    {
+        return empty($responseContentEncoding) &&
+            config('swoole_http.server.options.http_compression', true) &&
+            !empty($this->swooleRequest->header['accept-encoding']) &&
+            strpos($this->swooleRequest->header['accept-encoding'], 'gzip') !== false &&
+            function_exists('gzencode');
+    }
 }

tests/Transformers/ResponseTest.php

@@ -4,29 +4,30 @@
 
 use Illuminate\Http\Response as IlluminateResponse;
 use Swoole\Http\Response as SwooleResponse;
+use Swoole\Http\Request as SwooleRequest;
 use SwooleTW\Http\Tests\TestCase;
 use SwooleTW\Http\Transformers\Response;
 
 class ResponseTest extends TestCase
 {
     public function testMake()
     {
-        $response = Response::make(new IlluminateResponse, new SwooleResponse);
+        $response = Response::make(new IlluminateResponse, new SwooleResponse, new SwooleRequest);
 
         $this->assertInstanceOf(Response::class, $response);
     }
 
     public function testGetIlluminateResponse()
     {
-        $response = Response::make(new IlluminateResponse, new SwooleResponse);
+        $response = Response::make(new IlluminateResponse, new SwooleResponse, new SwooleRequest);
         $illuminateResponse = $response->getIlluminateResponse();
 
         $this->assertInstanceOf(IlluminateResponse::class, $illuminateResponse);
     }
 
     public function testGetSwooleResponse()
     {
-        $response = Response::make(new IlluminateResponse, new SwooleResponse);
+        $response = Response::make(new IlluminateResponse, new SwooleResponse, new SwooleRequest);
         $swooleResponse = $response->getSwooleResponse();
 
         $this->assertInstanceOf(SwooleResponse::class, $swooleResponse);