ShowModelsCommand Class Reference

#include <DdlCommandExecutor.h>

Inheritance diagram for ShowModelsCommand:

Collaboration diagram for ShowModelsCommand:

Public Member Functions
	ShowModelsCommand (const DdlCommandData &ddl_data, std::shared_ptr< Catalog_Namespace::SessionInfo const > session_ptr)
ExecutionResult	execute (bool read_only_mode) override
Public Member Functions inherited from DdlCommand
	DdlCommand (const DdlCommandData &ddl_data, std::shared_ptr< Catalog_Namespace::SessionInfo const > session_ptr)

Additional Inherited Members
Protected Attributes inherited from DdlCommand
const DdlCommandData &	ddl_data_
std::shared_ptr < Catalog_Namespace::SessionInfo const >	session_ptr_

Detailed Description

Definition at line 272 of file DdlCommandExecutor.h.

Constructor & Destructor Documentation

ShowModelsCommand::ShowModelsCommand	(	const DdlCommandData &	ddl_data,
		std::shared_ptr< Catalog_Namespace::SessionInfo const >	session_ptr
	)

Definition at line 2050 of file DdlCommandExecutor.cpp.

References g_enable_ml_functions, and g_restrict_ml_model_metadata_to_superusers.

    : DdlCommand(ddl_data, session_ptr) {
  if (!g_enable_ml_functions) {
    throw std::runtime_error("Cannot show models. ML functions are disabled.");
  }
  if (g_restrict_ml_model_metadata_to_superusers) {
    // Check if user is super user
    const auto& current_user = session_ptr->get_currentUser();
    if (!current_user.isSuper) {
      throw std::runtime_error(
          "Cannot show models. Showing model information to non-superusers is disabled.");
    }
  }
}

Member Function Documentation

ExecutionResult ShowModelsCommand::execute ( bool  read_only_mode )

overridevirtual

Executes the DDL command corresponding to provided JSON payload.

Parameters

_return

result of DDL command execution (if applicable)

Implements DdlCommand.

Definition at line 2067 of file DdlCommandExecutor.cpp.

References ResultSetLogicalValuesBuilder::create(), g_ml_models, genLiteralStr(), legacylockmgr::getExecuteReadLock(), MLModelMap::getModelNames(), kTEXT, and run_benchmark_import::label.

Referenced by heavydb.cursor.Cursor::executemany().

                                                              {
  auto execute_read_lock = legacylockmgr::getExecuteReadLock();

  // label_infos -> column labels
  std::vector<std::string> labels{"model_name"};
  std::vector<TargetMetaInfo> label_infos;
  for (const auto& label : labels) {
    label_infos.emplace_back(label, SQLTypeInfo(kTEXT, true));
  }

  // Get all model names
  const auto model_names = g_ml_models.getModelNames();

  // logical_values -> table data
  std::vector<RelLogicalValues::RowValues> logical_values;
  for (auto model_name : model_names) {
    logical_values.emplace_back(RelLogicalValues::RowValues{});
    logical_values.back().emplace_back(genLiteralStr(model_name));
  }

  // Create ResultSet
  std::shared_ptr<ResultSet> rSet = std::shared_ptr<ResultSet>(
      ResultSetLogicalValuesBuilder::create(label_infos, logical_values));

  return ExecutionResult(rSet, label_infos);
}

Here is the call graph for this function:

Here is the caller graph for this function:

---

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

• /home/jenkins-slave/workspace/core-os-doxygen/Catalog/DdlCommandExecutor.h
• /home/jenkins-slave/workspace/core-os-doxygen/Catalog/DdlCommandExecutor.cpp