leveldb源码分析：内存池 Arena

Arena

模块信息

util/arena.h

util/arena.cc

模块概要

用于实现一个简单的内存池，有两种内存分配方式，一种是普通分配，一种是对齐分配

主要接口

方法	说明
char* Arena::Allocate(size_t bytes)	普通方式分配内存
char* Arena::AllocateAligned(size_t bytes)	对齐方式分配内存
char* Arena::AllocateFallback(size_t bytes)	内存追加
char* Arena::AllocateNewBlock(size_t block_bytes)	创建新的内存块

源码分析

Arena::Allocate

inline char* Arena::Allocate(size_t bytes) {
  assert(bytes > 0);
  //如果申请的内存大小小于剩余内存，可以看到是直接返回剩余内存的起始地址，同时起始地址往后加
  //相应的大小，剩余内存大小减去相应的大小
  if (bytes <= alloc_bytes_remaining_) {
    char* result = alloc_ptr_;
    alloc_ptr_ += bytes;
    alloc_bytes_remaining_ -= bytes;
    return result;
  }
  //否则要新申请空间
  return AllocateFallback(bytes);
}

Arena::AllocateFallback

char* Arena::AllocateFallback(size_t bytes) {
  if (bytes > kBlockSize / 4) {
   	//如果需求空间比剩余空间大，且需求空间大于1k，则直接申请需求空间大小的内存块
   	//这是为了避免浪费过多空间
    char* result = AllocateNewBlock(bytes);
    return result;
  }

  //如果需求空间比剩余空间大，且需求空间小于1k，这也说明剩余空间比1k小，就可以申请块大小的空间
  //虽然也会浪费部分空间，但是浪费的空间小于1k
  alloc_ptr_ = AllocateNewBlock(kBlockSize);
  alloc_bytes_remaining_ = kBlockSize;

  char* result = alloc_ptr_;
  alloc_ptr_ += bytes;
  alloc_bytes_remaining_ -= bytes;
  return result;
}

Arena::AllocateNewBlock

char* Arena::AllocateNewBlock(size_t block_bytes) {
  char* result = new char[block_bytes];
  //用vector来存内存块
  blocks_.push_back(result);
  memory_usage_.fetch_add(block_bytes + sizeof(char*),
                          std::memory_order_relaxed);
  return result;
}

Arena::AllocateAligned

char* Arena::AllocateAligned(size_t bytes) {
  const int align = (sizeof(void*) > 8) ? sizeof(void*) : 8;
  //判断是否是2的幂次的方法，比如8&7=0,4&3=0
  static_assert((align & (align - 1)) == 0,
                "Pointer size should be a power of 2");
  //得到的是地址%align的部分
  size_t current_mod = reinterpret_cast<uintptr_t>(alloc_ptr_) & (align - 1);
  //要增加align-current_mod才能凑整齐
  size_t slop = (current_mod == 0 ? 0 : align - current_mod);
  //实际要新增加的为bytes+slop
  size_t needed = bytes + slop;
  char* result;
  //增加的方法和Allocate一样
  if (needed <= alloc_bytes_remaining_) {
    result = alloc_ptr_ + slop;
    alloc_ptr_ += needed;
    alloc_bytes_remaining_ -= needed;
  } else {
    // AllocateFallback always returned aligned memory
    result = AllocateFallback(bytes);
  }
  assert((reinterpret_cast<uintptr_t>(result) & (align - 1)) == 0);
  return result;
}

TO_DO

• 成员中有一个std::atomic<size_t> memory_usage_，和线程安全有关