Function MakeListViewRecordBatchSized

cpp/src/arrow/ipc/test_common.cc:465–487

}

Status MakeListViewRecordBatchSized(const int length, std::shared_ptr<RecordBatch>* out) {
  // Make the schema
  auto f0 = field("f0", list_view(int32()));
  auto f1 = field("f1", list_view(list_view(int32())));
  auto f2 = field("f2", large_list_view(int32()));
  auto schema = ::arrow::schema({f0, f1, f2});

  // Example data

  MemoryPool* pool = default_memory_pool();
  std::shared_ptr<Array> leaf_values, list_array, list_list_array, large_list_array;
  const bool include_nulls = true;
  RETURN_NOT_OK(MakeRandomInt32Array(1000, include_nulls, pool, &leaf_values));
  RETURN_NOT_OK(
      MakeRandomListViewArray(leaf_values, length, include_nulls, pool, &list_array));
  RETURN_NOT_OK(
      MakeRandomListViewArray(list_array, length, include_nulls, pool, &list_list_array));
  RETURN_NOT_OK(MakeRandomLargeListViewArray(leaf_values, length, include_nulls, pool,
                                             &large_list_array));
  *out =
      RecordBatch::Make(schema, length, {list_array, list_list_array, large_list_array});
  return Status::OK();
}

Status MakeListViewRecordBatch(std::shared_ptr<RecordBatch>* out) {
  return MakeListViewRecordBatchSized(200, out);