feat: add ability to request compressed ReadRowsResponse rows (#6955)

This change allows the client to request raw lz4 compression of the ReadRowsResponse rows data for both ArrowRecordBatches and Avro rows. PiperOrigin-RevId: 597000088 Source-Link: googleapis/googleapis@341d70f Source-Link: googleapis/googleapis-gen@01713f3 Copy-Tag: eyJwIjoiQmlnUXVlcnlTdG9yYWdlLy5Pd2xCb3QueWFtbCIsImgiOiIwMTcxM2YzZjU1MzRhY2M3OGYwNGQ1OWUxM2MwNjY4YzgxMjliZjAzIn0=
googleapis · Jan 12, 2024 · 8ec090e · 8ec090e
1 parent 53c8a0d
commit 8ec090e
Show file tree

Hide file tree

Showing 7 changed files with 219 additions and 0 deletions.
diff --git a/BigQueryStorage/metadata/V1/Storage.php b/BigQueryStorage/metadata/V1/Storage.php
diff --git a/BigQueryStorage/metadata/V1/Stream.php b/BigQueryStorage/metadata/V1/Stream.php
diff --git a/BigQueryStorage/src/V1/ReadRowsResponse.php b/BigQueryStorage/src/V1/ReadRowsResponse.php
diff --git a/BigQueryStorage/src/V1/ReadSession/TableReadOptions.php b/BigQueryStorage/src/V1/ReadSession/TableReadOptions.php
diff --git a/BigQueryStorage/src/V1/ReadSession/TableReadOptions/ResponseCompressionCodec.php b/BigQueryStorage/src/V1/ReadSession/TableReadOptions/ResponseCompressionCodec.php
diff --git a/BigQueryStorage/tests/Unit/V1/BigQueryReadClientTest.php b/BigQueryStorage/tests/Unit/V1/BigQueryReadClientTest.php
@@ -146,16 +146,22 @@ public function readRowsTest()
         $this->assertTrue($transport->isExhausted());
         // Mock response
         $rowCount = 1340416618;
+        $uncompressedByteSize = 2094376525;
         $expectedResponse = new ReadRowsResponse();
         $expectedResponse->setRowCount($rowCount);
+        $expectedResponse->setUncompressedByteSize($uncompressedByteSize);
         $transport->addResponse($expectedResponse);
         $rowCount2 = 349815907;
+        $uncompressedByteSize2 = 1643824294;
         $expectedResponse2 = new ReadRowsResponse();
         $expectedResponse2->setRowCount($rowCount2);
+        $expectedResponse2->setUncompressedByteSize($uncompressedByteSize2);
         $transport->addResponse($expectedResponse2);
         $rowCount3 = 349815906;
+        $uncompressedByteSize3 = 1643824295;
         $expectedResponse3 = new ReadRowsResponse();
         $expectedResponse3->setRowCount($rowCount3);
+        $expectedResponse3->setUncompressedByteSize($uncompressedByteSize3);
         $transport->addResponse($expectedResponse3);
         // Mock request
         $formattedReadStream = $gapicClient->readStreamName('[PROJECT]', '[LOCATION]', '[SESSION]', '[STREAM]');

diff --git a/BigQueryStorage/tests/Unit/V1/Client/BigQueryReadClientTest.php b/BigQueryStorage/tests/Unit/V1/Client/BigQueryReadClientTest.php
@@ -155,16 +155,22 @@ public function readRowsTest()
         $this->assertTrue($transport->isExhausted());
         // Mock response
         $rowCount = 1340416618;
+        $uncompressedByteSize = 2094376525;
         $expectedResponse = new ReadRowsResponse();
         $expectedResponse->setRowCount($rowCount);
+        $expectedResponse->setUncompressedByteSize($uncompressedByteSize);
         $transport->addResponse($expectedResponse);
         $rowCount2 = 349815907;
+        $uncompressedByteSize2 = 1643824294;
         $expectedResponse2 = new ReadRowsResponse();
         $expectedResponse2->setRowCount($rowCount2);
+        $expectedResponse2->setUncompressedByteSize($uncompressedByteSize2);
         $transport->addResponse($expectedResponse2);
         $rowCount3 = 349815906;
+        $uncompressedByteSize3 = 1643824295;
         $expectedResponse3 = new ReadRowsResponse();
         $expectedResponse3->setRowCount($rowCount3);
+        $expectedResponse3->setUncompressedByteSize($uncompressedByteSize3);
         $transport->addResponse($expectedResponse3);
         // Mock request
         $formattedReadStream = $gapicClient->readStreamName('[PROJECT]', '[LOCATION]', '[SESSION]', '[STREAM]');