Function MakeBatch

cpp/src/arrow/csv/generate_fuzz_corpus.cc:58–76

}

Result<std::shared_ptr<RecordBatch>> MakeBatch(
    std::function<Result<std::shared_ptr<Array>>(int64_t length, double null_probability)>
        array_factory,
    int64_t length) {
  ArrayVector columns;
  FieldVector fields;

  struct ColumnSpec {
    std::string name;
    double null_probability;
  };
  for (auto spec : {ColumnSpec{"with_nulls", 0.2}, ColumnSpec{"without_nulls", 0.0}}) {
    ARROW_ASSIGN_OR_RAISE(std::shared_ptr<Array> column,
                          array_factory(length, spec.null_probability));
    columns.push_back(column);
    fields.push_back(field(spec.name, column->type()));
  }
  return RecordBatch::Make(schema(std::move(fields)), length, std::move(columns));
}

Result<RecordBatchVector> Batches() {
  ::arrow::random::RandomArrayGenerator gen(/*seed=*/42);