#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< AggregatedResult >	query (QueryStateProxy, std::string &sql_query_string, std::vector< size_t > outer_frag_indices, bool allow_interrupt) override

std::list< ColumnDescriptor >	getColumnDescriptors (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.

                                         {
     return LocalInsertConnector::checkpoint(parent_session_info, tableId);
   }

std::list< ColumnDescriptor > Parser::LocalQueryConnector::getColumnDescriptors	(	AggregatedResult &	result,
		bool	for_create
	)

Definition at line 4036 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().

                      {
   std::list<ColumnDescriptor> column_descriptors;
   std::list<ColumnDescriptor> column_descriptors_for_create;
 
   int rowid_suffix = 0;
   for (const auto& target_metainfo : result.targets_meta) {
     ColumnDescriptor cd;
     cd.columnName = target_metainfo.get_resname();
     if (cd.columnName == "rowid") {
       cd.columnName += std::to_string(rowid_suffix++);
     }
     cd.columnType = target_metainfo.get_physical_type_info();
 
     ColumnDescriptor cd_for_create = cd;
 
     if (cd.columnType.get_compression() == kENCODING_DICT) {
       // we need to reset the comp param (as this points to the actual dictionary)
       if (cd.columnType.is_array()) {
         // for dict encoded arrays, it is always 4 bytes
         cd_for_create.columnType.set_comp_param(32);
       } else {
         cd_for_create.columnType.set_comp_param(cd.columnType.get_size() * 8);
       }
     }
 
     if (cd.columnType.is_date() && !cd.columnType.is_date_in_days()) {
       // default to kENCODING_DATE_IN_DAYS encoding
       cd_for_create.columnType.set_compression(kENCODING_DATE_IN_DAYS);
       cd_for_create.columnType.set_comp_param(0);
     }
 
     column_descriptors_for_create.push_back(cd_for_create);
     column_descriptors.push_back(cd);
   }
 
   if (for_create) {
     return column_descriptors_for_create;
   }
 
   return column_descriptors;
 }

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 3946 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.

                                                                                {
   auto const session = query_state_proxy->getConstSessionInfo();
   auto& catalog = session->getCatalog();
 
   auto executor = Executor::getExecutor(Executor::UNITARY_EXECUTOR_ID);
 #ifdef HAVE_CUDA
   const auto device_type = session->get_executor_device_type();
 #else
   const auto device_type = ExecutorDeviceType::CPU;
 #endif  // HAVE_CUDA
   auto calcite_mgr = catalog.getCalciteMgr();
 
   // TODO MAT this should actually get the global or the session parameter for
   // view optimization
   const auto calciteQueryParsingOption =
       calcite_mgr->getCalciteQueryParsingOption(true, false, true, false);
   const auto calciteOptimizationOption = calcite_mgr->getCalciteOptimizationOption(
       false,
       g_enable_watchdog,
       {},
       Catalog_Namespace::SysCatalog::instance().isAggregator());
   const auto query_ra = calcite_mgr
                             ->process(query_state_proxy,
                                       pg_shim(sql_query_string),
                                       calciteQueryParsingOption,
                                       calciteOptimizationOption)
                             .plan_result;
   RelAlgExecutor ra_executor(executor.get(), query_ra);
   CompilationOptions co = {device_type, true, ExecutorOptLevel::Default, false};
   // TODO(adb): Need a better method of dropping constants into this ExecutionOptions
   // struct
   ExecutionOptions eo = {false,
                          false,
                          true,
                          false,
                          true,
                          false,
                          false,
                          false,
                          false,
                          10000,
                          false,
                          false,
                          0.9,
                          false,
                          false};
   return ra_executor.getOuterFragmentCount(co, eo);
 }

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.

                                                                       {
     return LocalInsertConnector::insertChunksToLeaf(
         parent_session_info, leaf_idx, insert_chunks);
   }

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.

                                                                               {
     return LocalInsertConnector::insertDataToLeaf(
         parent_session_info, leaf_idx, insert_data);
   }

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 3996 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().

                                                                   {
   // TODO(PS): Should we be using the shimmed query in getResultSet?
   std::string pg_shimmed_select_query = pg_shim(sql_query_string);
 
   std::vector<TargetMetaInfo> target_metainfos;
   auto executor = Executor::getExecutor(Executor::UNITARY_EXECUTOR_ID);
   auto const session = query_state_proxy->getConstSessionInfo();
   auto query_session = session ? session->get_session_id() : "";
   auto query_submitted_time = query_state_proxy->getQuerySubmittedTime();
   if (allow_interrupt && !validate_only && !query_session.empty()) {
     executor->enrollQuerySession(query_session,
                                  sql_query_string,
                                  query_submitted_time,
                                  Executor::UNITARY_EXECUTOR_ID,
                                  QuerySessionStatus::QueryStatus::PENDING_EXECUTOR);
   }
   auto result_rows = getResultSet(query_state_proxy,
                                   sql_query_string,
                                   target_metainfos,
                                   validate_only,
                                   outer_frag_indices,
                                   allow_interrupt);
   AggregatedResult res = {result_rows, target_metainfos};
   return res;
 }

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

References query(), and run_benchmark_import::res.

                           {
   auto res = query(
       query_state_proxy, sql_query_string, outer_frag_indices, false, allow_interrupt);
   return {res};
 }

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.

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

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

/home/jenkins-slave/workspace/core-os-doxygen/Parser/ParserNode.h
/home/jenkins-slave/workspace/core-os-doxygen/Parser/ParserNode.cpp

Public Member Functions

Additional Inherited Members

Detailed Description

Member Function Documentation