anonymous_namespace{MapDHandler.cpp} Namespace Reference

Classes
struct	ForceDisconnect
struct	ProjectionTokensForCompletion

Functions
SessionMap::iterator	get_session_from_map (const TSessionId &session, SessionMap &session_map)
ProjectionTokensForCompletion	extract_projection_tokens_for_completion (const std::string &sql)
std::string	dump_table_col_names (const std::map< std::string, std::vector< std::string >> &table_col_names)
void	fixup_geo_column_descriptor (TColumnType &col_type, const SQLTypes subtype, const int output_srid)
void	check_table_not_sharded (const Catalog &cat, const std::string &table_name)
RecordBatchVector	loadArrowStream (const std::string &stream)
bool	TTypeInfo_IsGeo (const TDatumType::type &t)

Function Documentation

check_table_not_sharded()

void anonymous_namespace{MapDHandler.cpp}::check_table_not_sharded	(	const Catalog &	cat,
		const std::string &	table_name
	)

Definition at line 2196 of file MapDHandler.cpp.

References Catalog_Namespace::Catalog::getMetadataForTable().

Referenced by MapDHandler::load_table(), MapDHandler::load_table_binary(), and MapDHandler::prepare_columnar_loader().

                                                                              {
  const auto td = cat.getMetadataForTable(table_name);
  if (td && td->nShards) {
    throw std::runtime_error("Cannot import a sharded table directly to a leaf");
  }
}

Here is the call graph for this function:

Here is the caller graph for this function:

dump_table_col_names()

std::string anonymous_namespace{MapDHandler.cpp}::dump_table_col_names

(

const std::map< std::string, std::vector< std::string >> &

table_col_names

)

Definition at line 1667 of file MapDHandler.cpp.

Referenced by MapDHandler::get_result_row_for_pixel().

                                                                    {
  std::ostringstream oss;
  for (const auto& [table_name, col_names] : table_col_names) {
    oss << ":" << table_name;
    for (const auto& col_name : col_names) {
      oss << "," << col_name;
    }
  }
  return oss.str();
}

Here is the caller graph for this function:

extract_projection_tokens_for_completion()

ProjectionTokensForCompletion anonymous_namespace{MapDHandler.cpp}::extract_projection_tokens_for_completion

(

const std::string &

sql

)

Definition at line 1099 of file MapDHandler.cpp.

References split(), and to_upper().

Referenced by MapDHandler::get_completion_hints().

                          {
  boost::regex id_regex{R"(([[:alnum:]]|_|\.)+)",
                        boost::regex::extended | boost::regex::icase};
  boost::sregex_token_iterator tok_it(sql.begin(), sql.end(), id_regex, 0);
  boost::sregex_token_iterator end;
  std::unordered_set<std::string> uc_column_names;
  std::unordered_set<std::string> uc_column_table_qualifiers;
  for (; tok_it != end; ++tok_it) {
    std::string column_name = *tok_it;
    std::vector<std::string> column_tokens;
    boost::split(column_tokens, column_name, boost::is_any_of("."));
    if (column_tokens.size() == 2) {
      // If the column name is qualified, take user's word.
      uc_column_table_qualifiers.insert(to_upper(column_tokens.front()));
    } else {
      uc_column_names.insert(to_upper(column_name));
    }
  }
  return {uc_column_names, uc_column_table_qualifiers};
}

Here is the call graph for this function:

Here is the caller graph for this function:

fixup_geo_column_descriptor()

void anonymous_namespace{MapDHandler.cpp}::fixup_geo_column_descriptor ( TColumnType &  col_type, const SQLTypes  subtype, const int  output_srid  )

inline

Definition at line 1725 of file MapDHandler.cpp.

Referenced by MapDHandler::convert_target_metainfo(), and MapDHandler::populateThriftColumnType().

                                                               {
  col_type.col_type.precision = static_cast<int>(subtype);
  col_type.col_type.scale = output_srid;
}

Here is the caller graph for this function:

get_session_from_map()

SessionMap::iterator anonymous_namespace{MapDHandler.cpp}::get_session_from_map	(	const TSessionId &	session,
		SessionMap &	session_map
	)

Definition at line 120 of file MapDHandler.cpp.

References THROW_MAPD_EXCEPTION.

Referenced by MapDHandler::get_session_it_unsafe().

                                                                   {
  auto session_it = session_map.find(session);
  if (session_it == session_map.end()) {
    THROW_MAPD_EXCEPTION("Session not valid.");
  }
  return session_it;
}

Here is the caller graph for this function:

loadArrowStream()

RecordBatchVector anonymous_namespace{MapDHandler.cpp}::loadArrowStream

(

const std::string &

stream

)

Definition at line 2402 of file MapDHandler.cpp.

References ARROW_THRIFT_THROW_NOT_OK, logger::ERROR, and LOG.

Referenced by MapDHandler::load_table_binary_arrow().

                                                           {
  RecordBatchVector batches;
  try {
    // TODO(wesm): Make this simpler in general, see ARROW-1600
    auto stream_buffer =
        std::make_shared<arrow::Buffer>(reinterpret_cast<const uint8_t*>(stream.c_str()),
                                        static_cast<int64_t>(stream.size()));

    arrow::io::BufferReader buf_reader(stream_buffer);
    std::shared_ptr<arrow::RecordBatchReader> batch_reader;
    ARROW_THRIFT_THROW_NOT_OK(
        arrow::ipc::RecordBatchStreamReader::Open(&buf_reader, &batch_reader));

    while (true) {
      std::shared_ptr<arrow::RecordBatch> batch;
      // Read batch (zero-copy) from the stream
      ARROW_THRIFT_THROW_NOT_OK(batch_reader->ReadNext(&batch));
      if (batch == nullptr) {
        break;
      }
      batches.emplace_back(std::move(batch));
    }
  } catch (const std::exception& e) {
    LOG(ERROR) << "Error parsing Arrow stream: " << e.what() << ". Import aborted";
  }
  return batches;
}

Here is the caller graph for this function:

TTypeInfo_IsGeo()

bool anonymous_namespace{MapDHandler.cpp}::TTypeInfo_IsGeo

(

const TDatumType::type &

t

)

Definition at line 3703 of file MapDHandler.cpp.

References kGEOGRAPHY, kGEOMETRY, run_benchmark_import::result, and run_benchmark_import::type.

Referenced by MapDHandler::import_geo_table().

                                              {
  return (t == TDatumType::POLYGON || t == TDatumType::MULTIPOLYGON ||
          t == TDatumType::LINESTRING || t == TDatumType::POINT);
}

Here is the caller graph for this function: