您好,登录后才能下订单哦!
密码登录
登录注册
点击 登录注册 即表示同意《亿速云用户服务条款》
代码参考leveldb实现内存池的方法,由于实际工作中暂时未用到过内存池,因此这里只是一个简单的内存池实现,后续有需要时,可以根据实际需求再对代码进行修改。
arena.h
#ifndef ARENA_H #define ARENA_H #include <stdio.h> #include <vector> #include <stdint.h> #include <assert.h> //内存池类 //析构函数中自动释放内存 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<char*> 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 <assert.h> //固定块内存大小 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; }
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。