OmniSciDB  c1a53651b2
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Parser::LocalQueryConnector Class Reference

#include <ParserNode.h>

+ Inheritance diagram for Parser::LocalQueryConnector:
+ Collaboration diagram for Parser::LocalQueryConnector:

Public Member Functions

size_t getOuterFragmentCount (QueryStateProxy, std::string &sql_query_string) override
 
AggregatedResult query (QueryStateProxy, std::string &sql_query_string, std::vector< size_t > outer_frag_indices, bool validate_only, bool allow_interrupt)
 
std::vector< AggregatedResultquery (QueryStateProxy, std::string &sql_query_string, std::vector< size_t > outer_frag_indices, bool allow_interrupt) override
 
std::list< ColumnDescriptorgetColumnDescriptors (AggregatedResult &result, bool for_create)
 
size_t leafCount () override
 
void insertChunksToLeaf (const Catalog_Namespace::SessionInfo &parent_session_info, const size_t leaf_idx, const Fragmenter_Namespace::InsertChunks &insert_chunks) override
 
void insertDataToLeaf (const Catalog_Namespace::SessionInfo &parent_session_info, const size_t leaf_idx, Fragmenter_Namespace::InsertData &insert_data) override
 
void checkpoint (const Catalog_Namespace::SessionInfo &parent_session_info, int tableId) override
 
void rollback (const Catalog_Namespace::SessionInfo &parent_session_info, int tableId) override
 
- Public Member Functions inherited from Parser::QueryConnector
 ~QueryConnector ()=default
 
- Public Member Functions inherited from Fragmenter_Namespace::InsertDataLoader::InsertConnector
virtual ~InsertConnector ()=default
 

Additional Inherited Members

- Private Member Functions inherited from Fragmenter_Namespace::InsertDataLoader::InsertConnector
virtual ~InsertConnector ()=default
 

Detailed Description

Definition at line 1041 of file ParserNode.h.

Member Function Documentation

void Parser::LocalQueryConnector::checkpoint ( const Catalog_Namespace::SessionInfo parent_session_info,
int  tableId 
)
inlineoverridevirtual

Reimplemented from Fragmenter_Namespace::LocalInsertConnector.

Definition at line 1071 of file ParserNode.h.

1072  {
1073  return LocalInsertConnector::checkpoint(parent_session_info, tableId);
1074  }
std::list< ColumnDescriptor > Parser::LocalQueryConnector::getColumnDescriptors ( AggregatedResult result,
bool  for_create 
)

Definition at line 3562 of file ParserNode.cpp.

References ColumnDescriptor::columnName, ColumnDescriptor::columnType, SQLTypeInfo::get_compression(), SQLTypeInfo::get_size(), SQLTypeInfo::is_array(), SQLTypeInfo::is_date(), SQLTypeInfo::is_date_in_days(), kENCODING_DATE_IN_DAYS, kENCODING_DICT, SQLTypeInfo::set_comp_param(), SQLTypeInfo::set_compression(), AggregatedResult::targets_meta, and to_string().

Referenced by Parser::CreateTableAsSelectStmt::execute(), and Parser::InsertIntoTableAsSelectStmt::populateData().

3564  {
3565  std::list<ColumnDescriptor> column_descriptors;
3566  std::list<ColumnDescriptor> column_descriptors_for_create;
3567 
3568  int rowid_suffix = 0;
3569  for (const auto& target_metainfo : result.targets_meta) {
3570  ColumnDescriptor cd;
3571  cd.columnName = target_metainfo.get_resname();
3572  if (cd.columnName == "rowid") {
3573  cd.columnName += std::to_string(rowid_suffix++);
3574  }
3575  cd.columnType = target_metainfo.get_physical_type_info();
3576 
3577  ColumnDescriptor cd_for_create = cd;
3578 
3580  // we need to reset the comp param (as this points to the actual dictionary)
3581  if (cd.columnType.is_array()) {
3582  // for dict encoded arrays, it is always 4 bytes
3583  cd_for_create.columnType.set_comp_param(32);
3584  } else {
3585  cd_for_create.columnType.set_comp_param(cd.columnType.get_size() * 8);
3586  }
3587  }
3588 
3589  if (cd.columnType.is_date() && !cd.columnType.is_date_in_days()) {
3590  // default to kENCODING_DATE_IN_DAYS encoding
3592  cd_for_create.columnType.set_comp_param(0);
3593  }
3594 
3595  column_descriptors_for_create.push_back(cd_for_create);
3596  column_descriptors.push_back(cd);
3597  }
3598 
3599  if (for_create) {
3600  return column_descriptors_for_create;
3601  }
3602 
3603  return column_descriptors;
3604 }
void set_compression(EncodingType c)
Definition: sqltypes.h:504
HOST DEVICE int get_size() const
Definition: sqltypes.h:393
const std::vector< TargetMetaInfo > targets_meta
std::string to_string(char const *&&v)
specifies the content in-memory of a row in the column metadata table
void set_comp_param(int p)
Definition: sqltypes.h:505
HOST DEVICE EncodingType get_compression() const
Definition: sqltypes.h:389
bool is_date_in_days() const
Definition: sqltypes.h:988
SQLTypeInfo columnType
std::string columnName
bool is_date() const
Definition: sqltypes.h:998
bool is_array() const
Definition: sqltypes.h:588

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

size_t Parser::LocalQueryConnector::getOuterFragmentCount ( QueryStateProxy  query_state_proxy,
std::string &  sql_query_string 
)
overridevirtual

Implements Parser::QueryConnector.

Definition at line 3472 of file ParserNode.cpp.

References CPU, Default, g_enable_watchdog, query_state::QueryState::getConstSessionInfo(), Executor::getExecutor(), Catalog_Namespace::SysCatalog::instance(), Catalog_Namespace::SysCatalog::isAggregator(), pg_shim(), and Executor::UNITARY_EXECUTOR_ID.

3473  {
3474  auto const session = query_state_proxy->getConstSessionInfo();
3475  auto& catalog = session->getCatalog();
3476 
3478 #ifdef HAVE_CUDA
3479  const auto device_type = session->get_executor_device_type();
3480 #else
3481  const auto device_type = ExecutorDeviceType::CPU;
3482 #endif // HAVE_CUDA
3483  auto calcite_mgr = catalog.getCalciteMgr();
3484 
3485  // TODO MAT this should actually get the global or the session parameter for
3486  // view optimization
3487  const auto calciteQueryParsingOption =
3488  calcite_mgr->getCalciteQueryParsingOption(true, false, true);
3489  const auto calciteOptimizationOption = calcite_mgr->getCalciteOptimizationOption(
3490  false,
3492  {},
3494  const auto query_ra = calcite_mgr
3495  ->process(query_state_proxy,
3496  pg_shim(sql_query_string),
3497  calciteQueryParsingOption,
3498  calciteOptimizationOption)
3499  .plan_result;
3500  RelAlgExecutor ra_executor(executor.get(), query_ra);
3501  CompilationOptions co = {device_type, true, ExecutorOptLevel::Default, false};
3502  // TODO(adb): Need a better method of dropping constants into this ExecutionOptions
3503  // struct
3504  ExecutionOptions eo = {false,
3505  false,
3506  true,
3507  false,
3508  true,
3509  false,
3510  false,
3511  false,
3512  false,
3513  10000,
3514  false,
3515  false,
3516  0.9,
3517  false,
3518  false};
3519  return ra_executor.getOuterFragmentCount(co, eo);
3520 }
static std::shared_ptr< Executor > getExecutor(const ExecutorId id, const std::string &debug_dir="", const std::string &debug_file="", const SystemParameters &system_parameters=SystemParameters())
Definition: Execute.cpp:475
static SysCatalog & instance()
Definition: SysCatalog.h:343
bool g_enable_watchdog
std::shared_ptr< Catalog_Namespace::SessionInfo const > getConstSessionInfo() const
Definition: QueryState.cpp:84
std::string pg_shim(const std::string &query)
static const ExecutorId UNITARY_EXECUTOR_ID
Definition: Execute.h:373

+ Here is the call graph for this function:

void Parser::LocalQueryConnector::insertChunksToLeaf ( const Catalog_Namespace::SessionInfo parent_session_info,
const size_t  leaf_idx,
const Fragmenter_Namespace::InsertChunks insert_chunks 
)
inlineoverridevirtual

Reimplemented from Fragmenter_Namespace::LocalInsertConnector.

Definition at line 1058 of file ParserNode.h.

1061  {
1062  return LocalInsertConnector::insertChunksToLeaf(
1063  parent_session_info, leaf_idx, insert_chunks);
1064  }
void Parser::LocalQueryConnector::insertDataToLeaf ( const Catalog_Namespace::SessionInfo parent_session_info,
const size_t  leaf_idx,
Fragmenter_Namespace::InsertData insert_data 
)
inlineoverridevirtual

Reimplemented from Fragmenter_Namespace::LocalInsertConnector.

Definition at line 1065 of file ParserNode.h.

1067  {
1068  return LocalInsertConnector::insertDataToLeaf(
1069  parent_session_info, leaf_idx, insert_data);
1070  }
size_t Parser::LocalQueryConnector::leafCount ( )
inlineoverridevirtual

Reimplemented from Fragmenter_Namespace::LocalInsertConnector.

Definition at line 1057 of file ParserNode.h.

1057 { return LocalInsertConnector::leafCount(); }
AggregatedResult Parser::LocalQueryConnector::query ( QueryStateProxy  query_state_proxy,
std::string &  sql_query_string,
std::vector< size_t >  outer_frag_indices,
bool  validate_only,
bool  allow_interrupt 
)

Definition at line 3522 of file ParserNode.cpp.

References query_state::QueryState::getConstSessionInfo(), Executor::getExecutor(), query_state::QueryState::getQuerySubmittedTime(), Parser::getResultSet(), pg_shim(), run_benchmark_import::res, and Executor::UNITARY_EXECUTOR_ID.

Referenced by Parser::CreateTableAsSelectStmt::execute(), Parser::ExportQueryStmt::execute(), Parser::InsertIntoTableAsSelectStmt::populateData(), and query().

3526  {
3527  // TODO(PS): Should we be using the shimmed query in getResultSet?
3528  std::string pg_shimmed_select_query = pg_shim(sql_query_string);
3529 
3530  std::vector<TargetMetaInfo> target_metainfos;
3532  auto const session = query_state_proxy->getConstSessionInfo();
3533  auto query_session = session ? session->get_session_id() : "";
3534  auto query_submitted_time = query_state_proxy->getQuerySubmittedTime();
3535  if (allow_interrupt && !validate_only && !query_session.empty()) {
3536  executor->enrollQuerySession(query_session,
3537  sql_query_string,
3538  query_submitted_time,
3540  QuerySessionStatus::QueryStatus::PENDING_EXECUTOR);
3541  }
3542  auto result_rows = getResultSet(query_state_proxy,
3543  sql_query_string,
3544  target_metainfos,
3545  validate_only,
3546  outer_frag_indices,
3547  allow_interrupt);
3548  AggregatedResult res = {result_rows, target_metainfos};
3549  return res;
3550 }
std::shared_ptr< ResultSet > getResultSet(QueryStateProxy query_state_proxy, const std::string select_stmt, std::vector< TargetMetaInfo > &targets, bool validate_only=false, std::vector< size_t > outer_fragment_indices={}, bool allow_interrupt=false)
static std::shared_ptr< Executor > getExecutor(const ExecutorId id, const std::string &debug_dir="", const std::string &debug_file="", const SystemParameters &system_parameters=SystemParameters())
Definition: Execute.cpp:475
std::shared_ptr< Catalog_Namespace::SessionInfo const > getConstSessionInfo() const
Definition: QueryState.cpp:84
std::string pg_shim(const std::string &query)
const std::string getQuerySubmittedTime() const
Definition: QueryState.cpp:101
static const ExecutorId UNITARY_EXECUTOR_ID
Definition: Execute.h:373

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

std::vector< AggregatedResult > Parser::LocalQueryConnector::query ( QueryStateProxy  query_state_proxy,
std::string &  sql_query_string,
std::vector< size_t >  outer_frag_indices,
bool  allow_interrupt 
)
overridevirtual

Implements Parser::QueryConnector.

Definition at line 3552 of file ParserNode.cpp.

References query(), and run_benchmark_import::res.

3556  {
3557  auto res = query(
3558  query_state_proxy, sql_query_string, outer_frag_indices, false, allow_interrupt);
3559  return {res};
3560 }
AggregatedResult query(QueryStateProxy, std::string &sql_query_string, std::vector< size_t > outer_frag_indices, bool validate_only, bool allow_interrupt)

+ Here is the call graph for this function:

void Parser::LocalQueryConnector::rollback ( const Catalog_Namespace::SessionInfo parent_session_info,
int  tableId 
)
inlineoverridevirtual

Reimplemented from Fragmenter_Namespace::LocalInsertConnector.

Definition at line 1075 of file ParserNode.h.

1076  {
1077  return LocalInsertConnector::rollback(parent_session_info, tableId);
1078  }

The documentation for this class was generated from the following files: