Method executeVector

bolt/exec/fuzzer/PrestoQueryRunner.cpp:522–584

}

std::vector<bolt::RowVectorPtr> PrestoQueryRunner::executeVector(
    const std::string& sql,
    const std::vector<bolt::RowVectorPtr>& input,
    const bolt::RowTypePtr& resultType) {
  auto inputType = asRowType(input[0]->type());
  if (inputType->size() == 0) {
    // The query doesn't need to read any columns, but it needs to see a
    // specific number of rows. Make new 'input' as single all-null BIGINT
    // column with as many rows as original input. This way we'll be able to
    // create a 'tmp' table will the necessary number of rows.
    vector_size_t numInput = 0;
    for (const auto& v : input) {
      numInput += v->size();
    }

    auto column = BaseVector::createNullConstant(BIGINT(), numInput, pool());
    auto rowVector = std::make_shared<RowVector>(
        pool(),
        ROW({"x"}, {BIGINT()}),
        nullptr,
        numInput,
        std::vector<VectorPtr>{column});
    return executeVector(sql, {rowVector}, resultType);
  }

  // Create tmp table in Presto using DWRF file format and add a single
  // all-null row to it.

  std::stringstream nullValues;
  for (auto i = 0; i < inputType->size(); ++i) {
    appendComma(i, nullValues);
    nullValues << fmt::format(
        "cast(null as {})", toTypeSql(inputType->childAt(i)));
  }

  execute("DROP TABLE IF EXISTS tmp");

  execute(fmt::format(
      "CREATE TABLE tmp({}) WITH (format = 'DWRF') AS SELECT {}",
      folly::join(", ", inputType->names()),
      nullValues.str()));

  // Query Presto to find out table's location on disk.
  auto results = execute("SELECT \"$path\" FROM tmp");

  auto filePath = extractSingleValue<StringView>(results);
  auto tableDirectoryPath = fs::path(filePath).parent_path();

  // Delete the all-null row.
  execute("DELETE FROM tmp");

  // Create a new file in table's directory with fuzzer-generated data.
  auto newFilePath = fs::path(tableDirectoryPath)
                         .append("fuzzer.dwrf")
                         .string()
                         .substr(strlen("file:"));

  auto writerPool = rootPool()->addAggregateChild("writer");