Function SetOutputColumns

cpp/src/arrow/acero/tpch_node.cc:563–588

}

Result<std::shared_ptr<Schema>> SetOutputColumns(
    const std::vector<std::string>& columns,
    const std::vector<std::shared_ptr<DataType>>& types,
    const std::unordered_map<std::string, int>& name_map, std::vector<int>& gen_list) {
  gen_list.clear();
  std::vector<std::shared_ptr<Field>> fields;
  if (columns.empty()) {
    fields.resize(name_map.size());
    gen_list.resize(name_map.size());
    for (auto pair : name_map) {
      int col_idx = pair.second;
      fields[col_idx] = field(pair.first, types[col_idx]);
      gen_list[col_idx] = col_idx;
    }
    return schema(std::move(fields));
  } else {
    for (const std::string& col : columns) {
      auto entry = name_map.find(col);
      if (entry == name_map.end()) return Status::Invalid("Not a valid column name");
      int col_idx = static_cast<int>(entry->second);
      fields.push_back(field(col, types[col_idx]));
      gen_list.push_back(col_idx);
    }
    return schema(std::move(fields));
  }
}

Result<Datum> RandomVString(random::pcg32_fast& rng, int64_t num_rows, int32_t min_length,
                            int32_t max_length) {