#include <InternalMemoryStatsDataWrapper.h>

Inheritance diagram for foreign_storage::InternalMemoryStatsDataWrapper:

Collaboration diagram for foreign_storage::InternalMemoryStatsDataWrapper:

Public Member Functions
	InternalMemoryStatsDataWrapper ()

	InternalMemoryStatsDataWrapper (const int db_id, const ForeignTable *foreign_table)

Public Member Functions inherited from foreign_storage::InternalSystemDataWrapper
	InternalSystemDataWrapper ()

	InternalSystemDataWrapper (const int db_id, const ForeignTable *foreign_table)

void	populateChunkMetadata (ChunkMetadataVector &chunk_metadata_vector) override

void	populateChunkBuffers (const ChunkToBufferMap &required_buffers, const ChunkToBufferMap &optional_buffers, AbstractBuffer *delete_buffer) override

void	validateServerOptions (const ForeignServer *foreign_server) const override

void	validateTableOptions (const ForeignTable *foreign_table) const override

const std::set < std::string_view > &	getSupportedTableOptions () const override

void	validateUserMappingOptions (const UserMapping user_mapping, const ForeignServer foreign_server) const override

const std::set < std::string_view > &	getSupportedUserMappingOptions () const override

std::string	getSerializedDataWrapper () const override

void	restoreDataWrapperInternals (const std::string &file_path, const ChunkMetadataVector &chunk_metadata) override

bool	isRestored () const override

Public Member Functions inherited from foreign_storage::ForeignDataWrapper
	ForeignDataWrapper ()=default

virtual	~ForeignDataWrapper ()=default

virtual const std::set < std::string >	getAlterableTableOptions () const

virtual void	validateSchema (const std::list< ColumnDescriptor > &columns) const

virtual ParallelismLevel	getCachedParallelismLevel () const

virtual ParallelismLevel	getNonCachedParallelismLevel () const

virtual bool	isLazyFragmentFetchingEnabled () const

Private Member Functions
void	initializeObjectsForTable (const std::string &table_name) override

void	populateChunkBuffersForTable (const std::string &table_name, std::map< std::string, import_export::TypedImportBuffer * > &import_buffers) override

Private Attributes
std::map< std::string, std::vector< MemoryInfo > >	memory_info_by_device_type_

Additional Inherited Members
Public Types inherited from foreign_storage::ForeignDataWrapper
enum	ParallelismLevel { NONE, INTRA_FRAGMENT, INTER_FRAGMENT }

Protected Attributes inherited from foreign_storage::InternalSystemDataWrapper
const int	db_id_

const ForeignTable *	foreign_table_

size_t	row_count_ {0}

Detailed Description

Definition at line 29 of file InternalMemoryStatsDataWrapper.h.

Constructor & Destructor Documentation

foreign_storage::InternalMemoryStatsDataWrapper::InternalMemoryStatsDataWrapper ( )

Definition at line 23 of file InternalMemoryStatsDataWrapper.cpp.

24 : InternalSystemDataWrapper() {}

foreign_storage::InternalSystemDataWrapper::InternalSystemDataWrapper

InternalSystemDataWrapper()

Definition: InternalSystemDataWrapper.cpp:67

foreign_storage::InternalMemoryStatsDataWrapper::InternalMemoryStatsDataWrapper	(	const int	db_id,
		const ForeignTable *	foreign_table
	)

Definition at line 26 of file InternalMemoryStatsDataWrapper.cpp.

29 : InternalSystemDataWrapper(db_id, foreign_table) {}

foreign_storage::InternalSystemDataWrapper::InternalSystemDataWrapper

InternalSystemDataWrapper()

Definition: InternalSystemDataWrapper.cpp:67

Member Function Documentation

void foreign_storage::InternalMemoryStatsDataWrapper::initializeObjectsForTable ( const std::string & table_name )

overrideprivatevirtual

Implements foreign_storage::InternalSystemDataWrapper.

Definition at line 197 of file InternalMemoryStatsDataWrapper.cpp.

References CHECK_GE, Data_Namespace::CPU_LEVEL, foreign_storage::InternalSystemDataWrapper::foreign_table_, Buffer_Namespace::FREE, Catalog_Namespace::SysCatalog::getDataMgr(), Data_Namespace::GPU_LEVEL, Catalog_Namespace::SysCatalog::instance(), foreign_storage::anonymous_namespace{InternalMemoryStatsDataWrapper.cpp}::is_table_chunk(), Catalog_Namespace::MEMORY_DETAILS_SYS_TABLE_NAME, memory_info_by_device_type_, Catalog_Namespace::MEMORY_SUMMARY_SYS_TABLE_NAME, foreign_storage::InternalSystemDataWrapper::row_count_, TableDescriptor::tableName, and UNREACHABLE.

                                  {
   memory_info_by_device_type_.clear();
   row_count_ = 0;
   const auto& data_mgr = Catalog_Namespace::SysCatalog::instance().getDataMgr();
   // DataMgr::getMemoryInfoUnlocked() is used here because a lock on buffer_access_mutex_
   // is already acquired in DataMgr::getChunkMetadataVecForKeyPrefix()
   memory_info_by_device_type_["CPU"] =
       data_mgr.getMemoryInfoUnlocked(MemoryLevel::CPU_LEVEL);
   memory_info_by_device_type_["GPU"] =
       data_mgr.getMemoryInfoUnlocked(MemoryLevel::GPU_LEVEL);
   if (foreign_table_->tableName == Catalog_Namespace::MEMORY_SUMMARY_SYS_TABLE_NAME) {
     for (const auto& [device_type, memory_info_vector] : memory_info_by_device_type_) {
       row_count_ += memory_info_vector.size();
     }
   } else if (foreign_table_->tableName ==
              Catalog_Namespace::MEMORY_DETAILS_SYS_TABLE_NAME) {
     for (auto& [device_type, memory_info_vector] : memory_info_by_device_type_) {
       for (auto& memory_info : memory_info_vector) {
         for (auto& memory_data : memory_info.nodeMemoryData) {
           if (memory_data.memStatus == Buffer_Namespace::MemStatus::FREE) {
             memory_data.chunk_key.clear();
           } else {
             if (is_table_chunk(memory_data.chunk_key)) {
               CHECK_GE(memory_data.chunk_key.size(), static_cast<size_t>(4));
             }
           }
         }
         row_count_ += memory_info.nodeMemoryData.size();
       }
     }
   } else {
     UNREACHABLE() << "Unexpected table name: " << table_name;
   }
 }

Here is the call graph for this function:

void foreign_storage::InternalMemoryStatsDataWrapper::populateChunkBuffersForTable	(	const std::string &	table_name,
		std::map< std::string, import_export::TypedImportBuffer * > &	import_buffers
	)

overrideprivatevirtual

Implements foreign_storage::InternalSystemDataWrapper.

Definition at line 233 of file InternalMemoryStatsDataWrapper.cpp.

References foreign_storage::InternalSystemDataWrapper::foreign_table_, Catalog_Namespace::MEMORY_DETAILS_SYS_TABLE_NAME, memory_info_by_device_type_, Catalog_Namespace::MEMORY_SUMMARY_SYS_TABLE_NAME, foreign_storage::anonymous_namespace{InternalMemoryStatsDataWrapper.cpp}::populate_import_buffers_for_memory_details(), foreign_storage::anonymous_namespace{InternalMemoryStatsDataWrapper.cpp}::populate_import_buffers_for_memory_summary(), TableDescriptor::tableName, and UNREACHABLE.

                                                                         {
   if (foreign_table_->tableName == Catalog_Namespace::MEMORY_SUMMARY_SYS_TABLE_NAME) {
     populate_import_buffers_for_memory_summary(memory_info_by_device_type_,
                                                import_buffers);
   } else if (foreign_table_->tableName ==
              Catalog_Namespace::MEMORY_DETAILS_SYS_TABLE_NAME) {
     populate_import_buffers_for_memory_details(memory_info_by_device_type_,
                                                import_buffers);
   } else {
     UNREACHABLE() << "Unexpected table name: " << foreign_table_->tableName;
   }
 }

Here is the call graph for this function:

Member Data Documentation

std::map<std::string, std::vector<MemoryInfo> > foreign_storage::InternalMemoryStatsDataWrapper::memory_info_by_device_type_

private

Definition at line 42 of file InternalMemoryStatsDataWrapper.h.

Referenced by initializeObjectsForTable(), and populateChunkBuffersForTable().

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

/home/jenkins-slave/workspace/core-os-doxygen/DataMgr/ForeignStorage/InternalMemoryStatsDataWrapper.h
/home/jenkins-slave/workspace/core-os-doxygen/DataMgr/ForeignStorage/InternalMemoryStatsDataWrapper.cpp

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation