feat: enable http compression by sending data not with chunked encodi…

…ng - if possible
laravel · May 3, 2023 · 13b7fc1 · 13b7fc1
1 parent f3ae57e
commit 13b7fc1
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 5 deletions.
diff --git a/src/Swoole/SwooleClient.php b/src/Swoole/SwooleClient.php
@@ -260,12 +260,15 @@ protected function sendResponseContent(OctaneResponse $octaneResponse, SwooleRes
             return;
         }
 
+        // if the content fits into one chunk -> perform no chunked encoding
         if ($length <= $this->chunkSize) {
-            $swooleResponse->write($content);
-        } else {
-            for ($offset = 0; $offset < $length; $offset += $this->chunkSize) {
-                $swooleResponse->write(substr($content, $offset, $this->chunkSize));
-            }
+            $swooleResponse->end($content);
+
+            return;
+        }
+
+        for ($offset = 0; $offset < $length; $offset += $this->chunkSize) {
+            $swooleResponse->write(substr($content, $offset, $this->chunkSize));
         }
 
         $swooleResponse->end();

diff --git a/tests/SwooleClientTest.php b/tests/SwooleClientTest.php
@@ -119,6 +119,7 @@ public function test_static_file_can_be_served()
         $client->serveStaticFile($request, $context);
     }
 
+    /** @doesNotPerformAssertions @test */
     public function test_static_file_headers_can_be_sent()
     {
         $client = new SwooleClient;
@@ -297,4 +298,47 @@ public function test_error_method_sends_detailed_error_response_to_swoole_in_deb
         $swooleResponse->shouldHaveReceived('header')->with('Content-Type', 'text/plain');
         $swooleResponse->shouldHaveReceived('end')->with((string) $e);
     }
+
+    /** @doesNotPerformAssertions @test */
+    public function test_respond_method_send_not_chunked_response_to_swoole(): void
+    {
+        $client = new SwooleClient;
+
+        $swooleResponse = Mockery::mock('Swoole\Http\Response');
+
+        $swooleResponse->shouldReceive('status')->once()->with(200);
+        $swooleResponse->shouldReceive('header')->once()->with('Cache-Control', 'no-cache, private');
+        $swooleResponse->shouldReceive('header')->once()->with('Content-Type', 'text/html');
+        $swooleResponse->shouldReceive('header')->once()->with('Date', Mockery::type('string'));
+        $swooleResponse->shouldReceive('write')->never();
+        $swooleResponse->shouldReceive('end')->once()->with('Hello World');
+
+        $response = new Response('Hello World', 200, ['Content-Type' => 'text/html']);
+
+        $client->respond(new RequestContext([
+            'swooleResponse' => $swooleResponse,
+        ]), new OctaneResponse($response));
+    }
+
+    /** @doesNotPerformAssertions @test */
+    public function test_respond_method_send_chunked_response_to_swoole(): void
+    {
+        $client = new SwooleClient(6);
+
+        $swooleResponse = Mockery::mock('Swoole\Http\Response');
+
+        $swooleResponse->shouldReceive('status')->once()->with(200);
+        $swooleResponse->shouldReceive('header')->once()->with('Cache-Control', 'no-cache, private');
+        $swooleResponse->shouldReceive('header')->once()->with('Content-Type', 'text/html');
+        $swooleResponse->shouldReceive('header')->once()->with('Date', Mockery::type('string'));
+        $swooleResponse->shouldReceive('write')->once()->with('Hello ');
+        $swooleResponse->shouldReceive('write')->once()->with('World');
+        $swooleResponse->shouldReceive('end')->once();
+
+        $response = new Response('Hello World', 200, ['Content-Type' => 'text/html']);
+
+        $client->respond(new RequestContext([
+            'swooleResponse' => $swooleResponse,
+        ]), new OctaneResponse($response));
+    }
 }