| 1312 | } |
| 1313 | |
| 1314 | Status WriteRecordBatchMessage( |
| 1315 | int64_t length, int64_t body_length, |
| 1316 | const std::shared_ptr<const KeyValueMetadata>& custom_metadata, |
| 1317 | const std::vector<FieldMetadata>& nodes, const std::vector<BufferMetadata>& buffers, |
| 1318 | const std::vector<int64_t>& variadic_buffer_counts, const IpcWriteOptions& options, |
| 1319 | std::shared_ptr<Buffer>* out) { |
| 1320 | FBB fbb; |
| 1321 | RecordBatchOffset record_batch; |
| 1322 | RETURN_NOT_OK(MakeRecordBatch(fbb, length, body_length, nodes, buffers, |
| 1323 | variadic_buffer_counts, options, &record_batch)); |
| 1324 | return WriteFBMessage(fbb, flatbuf::MessageHeader::MessageHeader_RecordBatch, |
| 1325 | record_batch.Union(), body_length, options.metadata_version, |
| 1326 | custom_metadata, options.memory_pool) |
| 1327 | .Value(out); |
| 1328 | } |
| 1329 | |
| 1330 | Result<std::shared_ptr<Buffer>> WriteTensorMessage(const Tensor& tensor, |
| 1331 | int64_t buffer_start_offset, |