Method parse_cudf_string

engine/src/io/data_parser/sql/PostgreSQLParser.cpp:329–364

}

std::uint8_t postgresql_parser::parse_cudf_string(
  void * src, std::size_t col, std::size_t row, cudf_string_col * v) {
  PGresult * pgResult = static_cast<PGresult *>(src);

  Oid oid = PQftype(pgResult, col);
  if (oid == InvalidOid) { throw std::runtime_error("Bad postgresql type"); }

  if (PQgetisnull(pgResult, row, col)) {
    v->offsets.push_back(v->offsets.back());
    return 0;
  }
  const char * result = PQgetvalue(pgResult, row, col);
  std::string data;

  // TODO(cristhian): convert oid to data type using postgresql pgtype table
  // https://www.postgresql.org/docs/13/catalog-pg-type.html
  switch (oid) {
  case 1082: {  // date
    const std::int32_t value =
      ntohl(*reinterpret_cast<const std::int32_t *>(result));
    data = date_to_string(value);
    break;
  }
  case 1114: {  // timestamp
    const std::int64_t value =
      __builtin_bswap64(*reinterpret_cast<const std::int64_t *>(result));
    data = timestamp_to_string(value);
    break;
  }
  default: data = result;
  }

  v->chars.insert(v->chars.end(), data.cbegin(), data.cend());
  v->offsets.push_back(v->offsets.back() + data.length());
  return 1;
}


}  // namespace io