ResultSetLogicalValuesBuilder Class Reference

#include <ResultSetBuilder.h>

Inheritance diagram for ResultSetLogicalValuesBuilder:

Collaboration diagram for ResultSetLogicalValuesBuilder:

Public Member Functions
	ResultSetLogicalValuesBuilder (const std::vector< TargetInfo > &targets, const QueryMemoryDescriptor &query_mem_desc, const std::shared_ptr< RowSetMemoryOwner > row_set_mem_owner)
	ResultSetLogicalValuesBuilder (const RelLogicalValues *logical_values, const std::vector< TargetInfo > &targets, const ExecutorDeviceType device_type, const QueryMemoryDescriptor &query_mem_desc, const std::shared_ptr< RowSetMemoryOwner > row_set_mem_owner, const Executor *executor)
ResultSet *	build () override

Static Public Member Functions
static ResultSet *	create (std::vector< TargetMetaInfo > &label_infos, std::vector< RelLogicalValues::RowValues > &logical_values)

Private Attributes
const RelLogicalValues *	logical_values
const std::vector< TargetInfo > &	targets
const ExecutorDeviceType	device_type
const QueryMemoryDescriptor &	query_mem_desc
const std::shared_ptr < RowSetMemoryOwner >	row_set_mem_owner
const Executor *	executor

Additional Inherited Members
Protected Member Functions inherited from ResultSetBuilder
ResultSet *	makeResultSet (const std::vector< TargetInfo > &targets, const ExecutorDeviceType device_type, const QueryMemoryDescriptor &query_mem_desc, const std::shared_ptr< RowSetMemoryOwner > row_set_mem_owner, const Executor *executor)
void	addVarlenBuffer (ResultSet *result_set, std::vector< std::string > &varlen_storage)

Detailed Description

Definition at line 64 of file ResultSetBuilder.h.

Constructor & Destructor Documentation

ResultSetLogicalValuesBuilder::ResultSetLogicalValuesBuilder	(	const std::vector< TargetInfo > &	targets,
		const QueryMemoryDescriptor &	query_mem_desc,
		const std::shared_ptr< RowSetMemoryOwner >	row_set_mem_owner
	)

Definition at line 66 of file ResultSetBuilder.cpp.

References CPU.

Referenced by create().

    : logical_values(nullptr)
    , targets(_targets)
    , device_type(ExecutorDeviceType::CPU)
    , query_mem_desc(_query_mem_desc)
    , row_set_mem_owner(_row_set_mem_owner)
    , executor(nullptr) {}

Here is the caller graph for this function:

ResultSetLogicalValuesBuilder::ResultSetLogicalValuesBuilder	(	const RelLogicalValues *	logical_values,
		const std::vector< TargetInfo > &	targets,
		const ExecutorDeviceType	device_type,
		const QueryMemoryDescriptor &	query_mem_desc,
		const std::shared_ptr< RowSetMemoryOwner >	row_set_mem_owner,
		const Executor *	executor
	)

Definition at line 77 of file ResultSetBuilder.cpp.

    : logical_values(_logical_values)
    , targets(_targets)
    , device_type(_device_type)
    , query_mem_desc(_query_mem_desc)
    , row_set_mem_owner(_row_set_mem_owner)
    , executor(_executor) {}

Member Function Documentation

ResultSet * ResultSetLogicalValuesBuilder::build ( )

overridevirtual

Implements ResultSetBuilder.

Definition at line 91 of file ResultSetBuilder.cpp.

References ResultSetBuilder::addVarlenBuffer(), CHECK, CHECK_EQ, CHECK_GE, device_type, executor, RelLogicalValues::getNumRows(), QueryMemoryDescriptor::getRowSize(), RelLogicalValues::getRowsSize(), RelLogicalValues::getValueAt(), RelLogicalValues::hasRows(), inline_int_null_val(), logical_values, ResultSetBuilder::makeResultSet(), query_mem_desc, row_set_mem_owner, RelLogicalValues::size(), targets, TRANSIENT_DICT_DB_ID, TRANSIENT_DICT_ID, and RelAlgTranslator::translateLiteral().

Referenced by create().

                                                {
  ResultSet* rs = ResultSetBuilder::makeResultSet(
      targets, device_type, query_mem_desc, row_set_mem_owner, executor);

  if (logical_values && logical_values->hasRows()) {
    CHECK_EQ(logical_values->getRowsSize(), logical_values->size());

    std::vector<std::string> separate_varlen_storage;

    auto storage = rs->allocateStorage();
    auto buff = storage->getUnderlyingBuffer();

    for (size_t i = 0; i < logical_values->getNumRows(); i++) {
      std::vector<std::shared_ptr<Analyzer::Expr>> row_literals;
      int8_t* ptr = buff + i * query_mem_desc.getRowSize();

      for (size_t j = 0; j < logical_values->getRowsSize(); j++) {
        auto rex_literal =
            dynamic_cast<const RexLiteral*>(logical_values->getValueAt(i, j));
        CHECK(rex_literal);
        const auto expr = RelAlgTranslator::translateLiteral(rex_literal);
        const auto constant = std::dynamic_pointer_cast<Analyzer::Constant>(expr);
        CHECK(constant);

        if (constant->get_is_null()) {
          CHECK(!targets[j].sql_type.is_varlen());
          *reinterpret_cast<int64_t*>(ptr) = inline_int_null_val(targets[j].sql_type);
        } else {
          const auto ti = constant->get_type_info();
          const auto datum = constant->get_constval();

          if (ti.is_string()) {
            CHECK(ti.is_none_encoded_string());
            if (targets[j].sql_type.is_dict_encoded_string()) {
              // Translate none-encoded string literal to entry in transient string
              // dictionary. All proper DML queries go down this path
              CHECK_EQ(targets[j].sql_type.get_comp_param(), TRANSIENT_DICT_ID);
              CHECK_EQ(targets[j].sql_type.get_size(), 4);
              const auto sdp = executor->getStringDictionaryProxy(
                  {TRANSIENT_DICT_DB_ID, TRANSIENT_DICT_ID},
                  executor->getRowSetMemoryOwner(),
                  true);
              const auto string_id = sdp->getOrAddTransient(*datum.stringval);
              // Initialize the entire 8-byte slot
              std::memset(ptr, 0, 8);
              std::memcpy(ptr, &string_id, 4);
            } else {
              // Project none-encoded string - used for various psuedo-query
              // outputs like SHOW TABLE, SHOW TABLE DETAILS, etc
              CHECK(targets[j].sql_type.is_none_encoded_string());
              // get string from datum and push to vector
              separate_varlen_storage.push_back(*(datum.stringval));
              // store the index/offset in ResultSet's storage
              //   (the # of the string in the varlen_storage, not strLen)
              *reinterpret_cast<int64_t*>(ptr) =
                  static_cast<int64_t>(separate_varlen_storage.size() - 1);
            }
          } else {
            // Initialize the entire 8-byte slot
            std::memset(ptr, 0, 8);
            const auto sz = ti.get_size();
            CHECK_GE(sz, int(0));
            std::memcpy(ptr, &datum, sz);
          }
        }
        ptr += 8;
      }
    }

    // store the varlen data (ex. strings) into the ResultSet
    if (separate_varlen_storage.size()) {
      ResultSetBuilder::addVarlenBuffer(rs, separate_varlen_storage);
      rs->setSeparateVarlenStorageValid(true);
    }
  }

  return rs;
}

Here is the call graph for this function:

Here is the caller graph for this function:

ResultSet * ResultSetLogicalValuesBuilder::create ( std::vector< TargetMetaInfo > &  label_infos, std::vector< RelLogicalValues::RowValues > &  logical_values  )

static

Definition at line 171 of file ResultSetBuilder.cpp.

References QueryMemoryDescriptor::addColSlotInfo(), build(), CHECK_EQ, CPU, SQLTypeInfo::get_size(), Executor::getArenaBlockSize(), kENCODING_NONE, kSAMPLE, kTEXT, logical_values, heavyai::Projection, query_mem_desc, ResultSetLogicalValuesBuilder(), and row_set_mem_owner.

Referenced by ShowForeignServersCommand::execute(), ShowCreateServerCommand::execute(), ShowTablesCommand::execute(), ShowTableDetailsCommand::execute(), ShowDatabasesCommand::execute(), ShowFunctionsCommand::execute(), ShowRuntimeFunctionsCommand::execute(), ShowTableFunctionsCommand::execute(), ShowRuntimeTableFunctionsCommand::execute(), ShowModelsCommand::execute(), ShowModelDetailsCommand::execute(), ShowModelFeatureDetailsCommand::execute(), EvaluateModelCommand::execute(), ShowDiskCacheUsageCommand::execute(), ShowUserDetailsCommand::execute(), ShowRolesCommand::execute(), DBHandler::getQueries(), and DBHandler::getUserSessions().

                                                          {
  // check to see if number of columns matches (at least the first row)
  size_t numCols =
      logical_values.size() ? logical_values.front().size() : label_infos.size();
  CHECK_EQ(label_infos.size(), numCols);

  size_t numRows = logical_values.size();

  QueryMemoryDescriptor query_mem_desc(/*executor=*/nullptr,
                                       /*entry_count=*/numRows,
                                       QueryDescriptionType::Projection);

  // target_infos -> defines the table columns
  std::vector<TargetInfo> target_infos;
  SQLTypeInfo str_ti(kTEXT, true, kENCODING_NONE);
  for (size_t col = 0; col < numCols; col++) {
    SQLTypeInfo colType = label_infos[col].get_type_info();
    target_infos.push_back(TargetInfo{
        false, kSAMPLE, colType, colType, true, false, /*is_varlen_projection=*/false});
    query_mem_desc.addColSlotInfo({std::make_tuple(colType.get_size(), 8)});
  }

  std::shared_ptr<RelLogicalValues> rel_logical_values =
      std::make_shared<RelLogicalValues>(label_infos, logical_values);

  const auto row_set_mem_owner =
      std::make_shared<RowSetMemoryOwner>(Executor::getArenaBlockSize(), 0);

  // Construct ResultSet
  ResultSet* rsp(ResultSetLogicalValuesBuilder(rel_logical_values.get(),
                                               target_infos,
                                               ExecutorDeviceType::CPU,
                                               query_mem_desc,
                                               row_set_mem_owner,
                                               nullptr)
                     .build());

  return rsp;
}

Here is the call graph for this function:

Here is the caller graph for this function:

Member Data Documentation

const ExecutorDeviceType ResultSetLogicalValuesBuilder::device_type

private

Definition at line 68 of file ResultSetBuilder.h.

Referenced by build().

const Executor* ResultSetLogicalValuesBuilder::executor

private

Definition at line 71 of file ResultSetBuilder.h.

Referenced by build().

const RelLogicalValues* ResultSetLogicalValuesBuilder::logical_values

private

Definition at line 66 of file ResultSetBuilder.h.

Referenced by build(), and create().

const QueryMemoryDescriptor& ResultSetLogicalValuesBuilder::query_mem_desc

private

Definition at line 69 of file ResultSetBuilder.h.

Referenced by build(), and create().

const std::shared_ptr<RowSetMemoryOwner> ResultSetLogicalValuesBuilder::row_set_mem_owner

private

Definition at line 70 of file ResultSetBuilder.h.

Referenced by build(), and create().

const std::vector<TargetInfo>& ResultSetLogicalValuesBuilder::targets

private

Definition at line 67 of file ResultSetBuilder.h.

Referenced by build().

---

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

• /home/jenkins-slave/workspace/core-os-doxygen/QueryEngine/ResultSetBuilder.h
• /home/jenkins-slave/workspace/core-os-doxygen/QueryEngine/ResultSetBuilder.cpp