OmniSciDB  c1a53651b2
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
AbstractBufferMgr.h
Go to the documentation of this file.
1 /*
2  * Copyright 2022 HEAVY.AI, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
22 #pragma once
23 
24 #include <boost/preprocessor.hpp>
25 #include "../Shared/types.h"
26 #include "AbstractBuffer.h"
27 
28 #define X_DEFINE_ENUM_WITH_STRING_CONVERSIONS_TOSTRING_CASE(r, data, elem) \
29  case elem: \
30  return BOOST_PP_STRINGIZE(elem);
31 
32 #define DEFINE_ENUM_WITH_STRING_CONVERSIONS(name, enumerators) \
33  enum name { BOOST_PP_SEQ_ENUM(enumerators) }; \
34  \
35  inline const char* ToString(name v) { \
36  switch (v) { \
37  BOOST_PP_SEQ_FOR_EACH( \
38  X_DEFINE_ENUM_WITH_STRING_CONVERSIONS_TOSTRING_CASE, name, enumerators) \
39  default: \
40  return "[Unknown " BOOST_PP_STRINGIZE(name) "]"; \
41  } \
42  }
43 
45  MgrType,
46  (CACHING_FILE_MGR)(FILE_MGR)(CPU_MGR)(GPU_MGR)(GLOBAL_FILE_MGR)(
47  PERSISTENT_STORAGE_MGR)(FOREIGN_STORAGE_MGR)(TIERED_CPU_MGR))
48 
49 namespace Data_Namespace {
50 
63 class AbstractBufferMgr {
64  public:
65  virtual ~AbstractBufferMgr() {}
66  AbstractBufferMgr(const int deviceId) : device_id_(deviceId) {}
67 
68  // Chunk API
69  virtual AbstractBuffer* createBuffer(const ChunkKey& key,
70  const size_t pageSize = 0,
71  const size_t initialSize = 0) = 0;
72  virtual void deleteBuffer(
73  const ChunkKey& key,
74  const bool purge = true) = 0; // purge param only used in FileMgr
75  virtual void deleteBuffersWithPrefix(const ChunkKey& keyPrefix,
76  const bool purge = true) = 0;
77  virtual AbstractBuffer* getBuffer(const ChunkKey& key, const size_t numBytes = 0) = 0;
78  virtual void fetchBuffer(const ChunkKey& key,
79  AbstractBuffer* destBuffer,
80  const size_t numBytes = 0) = 0;
81  virtual AbstractBuffer* putBuffer(const ChunkKey& key,
82  AbstractBuffer* srcBuffer,
83  const size_t numBytes = 0) = 0;
84  virtual void getChunkMetadataVecForKeyPrefix(ChunkMetadataVector& chunkMetadataVec,
85  const ChunkKey& keyPrefix) = 0;
86 
87  virtual bool isBufferOnDevice(const ChunkKey& key) = 0;
88  virtual std::string printSlabs() = 0;
89  virtual size_t getMaxSize() = 0;
90  virtual size_t getInUseSize() = 0;
91  virtual size_t getAllocated() = 0;
92  virtual bool isAllocationCapped() = 0;
93 
94  virtual void checkpoint() = 0;
95  virtual void checkpoint(const int db_id, const int tb_id) = 0;
96  virtual void removeTableRelatedDS(const int db_id, const int table_id) = 0;
97 
98  // Buffer API
99  virtual AbstractBuffer* alloc(const size_t numBytes = 0) = 0;
100  virtual void free(AbstractBuffer* buffer) = 0;
101  virtual MgrType getMgrType() = 0;
102  virtual std::string getStringMgrType() = 0;
103  virtual size_t getNumChunks() = 0;
104  inline int getDeviceId() { return device_id_; }
105 
106  protected:
107  AbstractBufferMgr* parentMgr_;
108  int device_id_;
109 };
110 
111 } // namespace Data_Namespace
std::vector< int > ChunkKey
Definition: types.h:36
#define DEFINE_ENUM_WITH_STRING_CONVERSIONS(name, enumerators)
std::vector< std::pair< ChunkKey, std::shared_ptr< ChunkMetadata >>> ChunkMetadataVector