代码参考leveldb实现内存池的方法,由于实际工作中暂时未用到过内存池,因此这里只是一个简单的内存池实现,后续有需要时,可以根据实际需求再对代码进行修改。
10年积累的成都做网站、网站设计、外贸营销网站建设经验,可以快速应对客户对网站的新想法和需求。提供各种问题对应的解决方案。让选择我们的客户得到更好、更有力的网络服务。我虽然不认识你,你也不认识我。但先网站设计后付款的网站建设流程,更有旅顺口免费网站建设让你可以放心的选择与我们合作。
arena.h
#ifndef ARENA_H #define ARENA_H #include#include #include #include //内存池类 //析构函数中自动释放内存 class Arena { public: Arena(); ~Arena(); //内存申请函数 //@bytes 需要申请的内存大小 //返回指向内存的指针 char* Allocate(size_t bytes); //内存使用情况函数 size_t MemoryUsage() const { return memory_usage_; } size_t MemoryRemain() { return alloc_bytes_remaining_; } private: //申请内存函数,当内存池中剩余内存少于申请内存时使用 //@bytes 需要申请的内存大小 //返回指向内存的指针 char* AllocateFallback(size_t bytes); //申请块内存函数,直接申请一块新的内存块 //@block_bytes 需要申请的块内存大小 //返回指向内存的指针 char* AllocateNewBlock(size_t block_bytes); char* alloc_ptr_; //指向内存的指针 size_t alloc_bytes_remaining_; //剩余可用内存大小 std::vector blocks_; //内存池 size_t memory_usage_; //总共申请的内存大小 Arena(const Arena&); void operator=(const Arena&); }; 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); } #endif // ARENA_H
arena.cpp
#include "arena.h" #include//固定块内存大小 static const int kBlockSize = 4096; Arena::Arena() : memory_usage_(0) { alloc_ptr_ = NULL; alloc_bytes_remaining_ = 0; } Arena::~Arena() { for(size_t i = 0; i < blocks_.size(); i++) delete[] blocks_[i]; } char* Arena::AllocateFallback(size_t bytes) { if(bytes > kBlockSize/4) { char* result = AllocateNewBlock(bytes); return result; } alloc_ptr_ = AllocateNewBlock(kBlockSize); alloc_bytes_remaining_ = kBlockSize; char* result = alloc_ptr_; alloc_ptr_ += bytes; alloc_bytes_remaining_ -= bytes; return result; } char* Arena::AllocateNewBlock(size_t block_bytes) { char* result = new char[block_bytes]; blocks_.push_back(result); memory_usage_ = MemoryUsage() + block_bytes + sizeof(char*); return result; }