OmniSciDB  72c90bc290
 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 1042 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 1072 of file ParserNode.h.

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

Definition at line 4019 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::CreateModelStmt::build_model_query(), EvaluateModelCommand::execute(), Parser::CreateTableAsSelectStmt::execute(), and Parser::InsertIntoTableAsSelectStmt::populateData().

4021  {
4022  std::list<ColumnDescriptor> column_descriptors;
4023  std::list<ColumnDescriptor> column_descriptors_for_create;
4024 
4025  int rowid_suffix = 0;
4026  for (const auto& target_metainfo : result.targets_meta) {
4027  ColumnDescriptor cd;
4028  cd.columnName = target_metainfo.get_resname();
4029  if (cd.columnName == "rowid") {
4030  cd.columnName += std::to_string(rowid_suffix++);
4031  }
4032  cd.columnType = target_metainfo.get_physical_type_info();
4033 
4034  ColumnDescriptor cd_for_create = cd;
4035 
4037  // we need to reset the comp param (as this points to the actual dictionary)
4038  if (cd.columnType.is_array()) {
4039  // for dict encoded arrays, it is always 4 bytes
4040  cd_for_create.columnType.set_comp_param(32);
4041  } else {
4042  cd_for_create.columnType.set_comp_param(cd.columnType.get_size() * 8);
4043  }
4044  }
4045 
4046  if (cd.columnType.is_date() && !cd.columnType.is_date_in_days()) {
4047  // default to kENCODING_DATE_IN_DAYS encoding
4049  cd_for_create.columnType.set_comp_param(0);
4050  }
4051 
4052  column_descriptors_for_create.push_back(cd_for_create);
4053  column_descriptors.push_back(cd);
4054  }
4055 
4056  if (for_create) {
4057  return column_descriptors_for_create;
4058  }
4059 
4060  return column_descriptors;
4061 }
void set_compression(EncodingType c)
Definition: sqltypes.h:479
HOST DEVICE int get_size() const
Definition: sqltypes.h:403
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:480
HOST DEVICE EncodingType get_compression() const
Definition: sqltypes.h:399
bool is_date_in_days() const
Definition: sqltypes.h:1016
SQLTypeInfo columnType
std::string columnName
bool is_date() const
Definition: sqltypes.h:1026
bool is_array() const
Definition: sqltypes.h:583

+ 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 3929 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.

3930  {
3931  auto const session = query_state_proxy->getConstSessionInfo();
3932  auto& catalog = session->getCatalog();
3933 
3935 #ifdef HAVE_CUDA
3936  const auto device_type = session->get_executor_device_type();
3937 #else
3938  const auto device_type = ExecutorDeviceType::CPU;
3939 #endif // HAVE_CUDA
3940  auto calcite_mgr = catalog.getCalciteMgr();
3941 
3942  // TODO MAT this should actually get the global or the session parameter for
3943  // view optimization
3944  const auto calciteQueryParsingOption =
3945  calcite_mgr->getCalciteQueryParsingOption(true, false, true, false);
3946  const auto calciteOptimizationOption = calcite_mgr->getCalciteOptimizationOption(
3947  false,
3949  {},
3951  const auto query_ra = calcite_mgr
3952  ->process(query_state_proxy,
3953  pg_shim(sql_query_string),
3954  calciteQueryParsingOption,
3955  calciteOptimizationOption)
3956  .plan_result;
3957  RelAlgExecutor ra_executor(executor.get(), query_ra);
3958  CompilationOptions co = {device_type, true, ExecutorOptLevel::Default, false};
3959  // TODO(adb): Need a better method of dropping constants into this ExecutionOptions
3960  // struct
3961  ExecutionOptions eo = {false,
3962  false,
3963  true,
3964  false,
3965  true,
3966  false,
3967  false,
3968  false,
3969  false,
3970  10000,
3971  false,
3972  false,
3973  0.9,
3974  false,
3975  false};
3976  return ra_executor.getOuterFragmentCount(co, eo);
3977 }
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:509
static SysCatalog & instance()
Definition: SysCatalog.h:343
bool g_enable_watchdog
static constexpr ExecutorId UNITARY_EXECUTOR_ID
Definition: Execute.h:423
std::shared_ptr< Catalog_Namespace::SessionInfo const > getConstSessionInfo() const
Definition: QueryState.cpp:84
std::string pg_shim(const std::string &query)

+ 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 1059 of file ParserNode.h.

1062  {
1063  return LocalInsertConnector::insertChunksToLeaf(
1064  parent_session_info, leaf_idx, insert_chunks);
1065  }
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 1066 of file ParserNode.h.

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

Reimplemented from Fragmenter_Namespace::LocalInsertConnector.

Definition at line 1058 of file ParserNode.h.

1058 { 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 3979 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::CreateModelStmt::build_model_query(), EvaluateModelCommand::execute(), Parser::CreateTableAsSelectStmt::execute(), Parser::ExportQueryStmt::execute(), Parser::InsertIntoTableAsSelectStmt::populateData(), query(), and Parser::CreateModelStmt::train_model().

3983  {
3984  // TODO(PS): Should we be using the shimmed query in getResultSet?
3985  std::string pg_shimmed_select_query = pg_shim(sql_query_string);
3986 
3987  std::vector<TargetMetaInfo> target_metainfos;
3989  auto const session = query_state_proxy->getConstSessionInfo();
3990  auto query_session = session ? session->get_session_id() : "";
3991  auto query_submitted_time = query_state_proxy->getQuerySubmittedTime();
3992  if (allow_interrupt && !validate_only && !query_session.empty()) {
3993  executor->enrollQuerySession(query_session,
3994  sql_query_string,
3995  query_submitted_time,
3997  QuerySessionStatus::QueryStatus::PENDING_EXECUTOR);
3998  }
3999  auto result_rows = getResultSet(query_state_proxy,
4000  sql_query_string,
4001  target_metainfos,
4002  validate_only,
4003  outer_frag_indices,
4004  allow_interrupt);
4005  AggregatedResult res = {result_rows, target_metainfos};
4006  return res;
4007 }
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:509
static constexpr ExecutorId UNITARY_EXECUTOR_ID
Definition: Execute.h:423
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

+ 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 4009 of file ParserNode.cpp.

References query(), and run_benchmark_import::res.

4013  {
4014  auto res = query(
4015  query_state_proxy, sql_query_string, outer_frag_indices, false, allow_interrupt);
4016  return {res};
4017 }
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 1076 of file ParserNode.h.

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

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