/Users/deen/code/yugabyte-db/src/yb/util/memory/mc_types.h

Source
//--------------------------------------------------------------------------------------------------
// Copyright (c) YugaByte, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
// in compliance with the License.  You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software distributed under the License
// is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
// or implied.  See the License for the specific language governing permissions and limitations
// under the License.
//
//
// This module contains several datatypes that are to be used together with the MemoryContext.
// With current implementation, we focus on creating STL collection of pointers such as
//   MCVector<MCString *>
//   MCList<MCString *>
// Our SQL processes work on pointers exclusively. For example, the variable nodes that share the
// same name would all be point to the same MCString and also the same entry in our symbol table.
// Eventually, our symbol table should use unique pointer and treenode would contain the raw
// pointers of the unique pointers.
//
// Examples:
// - String type.
//     MCString mc_string(memctx.get(), "abc");
//     mc_string += "xyz";
//
// - STL types.
//     MCVector<int> mc_vec(memctx.get());
//     vec.reserve(77);
//
// - SQL user-defined object.
//     class MyClass : public MCBase {
//     };
//     MyClass *mc_obj = new(memctx.get()) MyClass();
//
// - All of the above instances - mc_string, mc_vec, mc_obj - contain memory context, which
//   can be use to allocate new classes.
//     MCList<int> mc_list(mc_string.memory_context());
//--------------------------------------------------------------------------------------------------
#ifndef YB_UTIL_MEMORY_MC_TYPES_H
#define YB_UTIL_MEMORY_MC_TYPES_H

#include <atomic>
#include <deque>
#include <list>
#include <map>
#include <memory>
#include <new>
#include <set>
#include <unordered_map>
#include <unordered_set>
#include <vector>

#include <boost/tti/has_type.hpp>

#include "yb/util/memory/arena_fwd.h"

namespace yb {

//--------------------------------------------------------------------------------------------------
// Buffer (char*) support.
char *MCStrdup(Arena *arena, const char *str);

template<class MCObject> using MCDeque = std::deque<MCObject, ArenaAllocator<MCObject>>;

// Class MCList.
template<class MCObject> using MCList = std::list<MCObject, ArenaAllocator<MCObject>>;

// Class MCVector.
template<class MCObject> using MCVector = std::vector<MCObject, ArenaAllocator<MCObject>>;

// Class MCSet.
template<class MCObject, class Compare = std::less<MCObject>>
using MCSet = std::set<MCObject, Compare, ArenaAllocator<MCObject>>;

// Class MCMap.
template<class MCKey, class MCObject, class Compare = std::less<MCKey>>
using MCMap = std::map<MCKey, MCObject, Compare, ArenaAllocator<std::pair<const MCKey, MCObject>>>;

template<class Key, class Tp, class Hash = std::hash<Key>, class Pred = std::equal_to<Key>>
using MCUnorderedMap = std::unordered_map<Key,
                                          Tp,
                                          Hash,
                                          Pred,
                                          ArenaAllocator<std::pair<const Key, Tp>>>;

template<class Key, class Hash = std::hash<Key>, class Pred = std::equal_to<Key>>
using MCUnorderedSet = std::unordered_set<Key,
                                          Hash,
                                          Pred,
                                          ArenaAllocator<Key>>;

//--------------------------------------------------------------------------------------------------
// String support.
// To use MCAllocator, strings should be declared as one of the following.
//   MCString s(memctx);
//   MCSharedPtr<MCString> s = MCMakeSharedString(memctx);

typedef std::basic_string<char, std::char_traits<char>, ArenaAllocator<char>> MCString;

typedef Arena MemoryContext;

//--------------------------------------------------------------------------------------------------
// Context-control shared_ptr and unique_ptr
template<class MCObject> using MCUniPtr = std::unique_ptr<MCObject, ArenaObjectDeleter>;
template<class MCObject> using MCSharedPtr = std::shared_ptr<MCObject>;

//--------------------------------------------------------------------------------------------------
// User-defined object support.
// All objects that use MCAllocator should be derived from MCBase. For example:
// class MCMyObject : public MCBase {
// };

BOOST_TTI_HAS_TYPE(allocator_type);

template<class MCObject, typename... TypeArgs>
typename std::enable_if<!has_type_allocator_type<MCObject>::value, MCSharedPtr<MCObject>>::type
MCAllocateSharedHelper(MCObject*, ArenaAllocator<MCObject> allocator, TypeArgs&&... args) {
  return std::allocate_shared<MCObject>(allocator,
                                        allocator.arena(),
                                        std::forward<TypeArgs>(args)...);
}

template<class MCObject, typename... TypeArgs>
typename std::enable_if<has_type_allocator_type<MCObject>::value, MCSharedPtr<MCObject>>::type
MCAllocateSharedHelper(MCObject*, ArenaAllocator<MCObject> allocator, TypeArgs&&... args) {
  return std::allocate_shared<MCObject>(allocator, std::forward<TypeArgs>(args)..., allocator);
}

template<class MCObject, typename... TypeArgs>
MCSharedPtr<MCObject> MCAllocateShared(ArenaAllocator<MCObject> allocator, TypeArgs&&... args) {
  return MCAllocateSharedHelper(static_cast<MCObject*>(nullptr),
                                allocator,
                                std::forward<TypeArgs>(args)...);
}

// Construct a shared_ptr to any MC object.
template<class MCObject, typename... TypeArgs>
MCSharedPtr<MCObject> MCMakeShared(Arena *arena, TypeArgs&&... args) {
  ArenaAllocator<MCObject> allocator(arena);
  return MCAllocateShared<MCObject>(allocator, std::forward<TypeArgs>(args)...);
}

// MC base class.
class MCBase {
 public:
  //------------------------------------------------------------------------------------------------
  // Public types.
  typedef MCSharedPtr<MCBase> SharedPtr;
  typedef MCSharedPtr<const MCBase> SharedPtrConst;

  // Constructors.
  explicit MCBase(Arena *arena = nullptr);
  virtual ~MCBase();

  template<typename... TypeArgs>
  inline static MCBase::SharedPtr MakeShared(Arena *arena, TypeArgs&&... args) {
    return MCMakeShared<MCBase>(arena, std::forward<TypeArgs>(args)...);
  }

  void operator delete(void* ptr) noexcept;
  void *operator new(size_t bytes, Arena* arena) noexcept;
  void *operator new(size_t bytes, void* address) noexcept;

 private:
  //------------------------------------------------------------------------------------------------
  // The following functions are deprecated and not supported for MC objects.

  // Default new/delete operators are disabled.
  void operator delete[](void* ptr) noexcept = delete;
  void *operator new(size_t bytes) noexcept = delete;
  void *operator new[](size_t bytes) noexcept = delete;
};

}  // namespace yb

#endif // YB_UTIL_MEMORY_MC_TYPES_H

YugabyteDB (2.13.1.0-b60, 21121d69985fbf76aa6958d8f04a9bfa936293b5)

Coverage Report

Created: 2022-03-22 16:43