Function MakeRowTableFromExecBatch

cpp/src/arrow/compute/row/compare_test.cc:176–196

namespace {

Result<RowTableImpl> MakeRowTableFromExecBatch(const ExecBatch& batch) {
  RowTableImpl row_table;

  std::vector<KeyColumnMetadata> column_metadatas;
  RETURN_NOT_OK(ColumnMetadatasFromExecBatch(batch, &column_metadatas));
  RowTableMetadata table_metadata;
  table_metadata.FromColumnMetadataVector(column_metadatas, sizeof(uint64_t),
                                          sizeof(uint64_t));
  RETURN_NOT_OK(row_table.Init(default_memory_pool(), table_metadata));
  std::vector<uint16_t> row_ids(batch.length);
  std::iota(row_ids.begin(), row_ids.end(), 0);
  RowTableEncoder row_encoder;
  row_encoder.Init(column_metadatas, sizeof(uint64_t), sizeof(uint64_t));
  std::vector<KeyColumnArray> column_arrays;
  RETURN_NOT_OK(ColumnArraysFromExecBatch(batch, &column_arrays));
  row_encoder.PrepareEncodeSelected(0, batch.length, column_arrays);
  RETURN_NOT_OK(row_encoder.EncodeSelected(
      &row_table, static_cast<uint32_t>(batch.length), row_ids.data()));

  return row_table;
}

Result<RowTableImpl> RepeatRowTableUntil(const RowTableImpl& seed, int64_t num_rows) {
  RowTableImpl row_table;