feat: add experimental ArrowData type and arrow_data field within AppendRowsRequest

Google APIs · copybara-github · commit b49a98382080 · 2024-10-11T11:27:46.000-07:00
PiperOrigin-RevId: 684900949
diff --git a/google/cloud/bigquery/storage/v1/storage.proto b/google/cloud/bigquery/storage/v1/storage.proto
@@ -422,6 +422,17 @@ message CreateWriteStreamRequest {
 // The size of a single AppendRowsRequest must be less than 10 MB in size.
 // Requests larger than this return an error, typically `INVALID_ARGUMENT`.
 message AppendRowsRequest {
+  // Arrow schema and data.
+  // Arrow format is an experimental feature only selected for allowlisted
+  // customers.
+  message ArrowData {
+    // Optional. Arrow Schema used to serialize the data.
+    ArrowSchema writer_schema = 1;
+
+    // Required. Serialized row data in Arrow format.
+    ArrowRecordBatch rows = 2;
+  }
+
   // ProtoData contains the data rows and schema when constructing append
   // requests.
   message ProtoData {
@@ -510,6 +521,10 @@ message AppendRowsRequest {
   oneof rows {
     // Rows in proto format.
     ProtoData proto_rows = 4;
+
+    // Rows in arrow format. This is an experimental feature only selected for
+    // allowlisted customers.
+    ArrowData arrow_rows = 5;
   }
 
   // Id set by client to annotate its identity. Only initial request setting is
