C++实现内存池

发布时间:2020-07-18 22:40:24 作者:chengarm
来源:网络 阅读:25162

        代码参考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;
}




推荐阅读:
  1. C++如何实现线程池
  2. 内存池的实现方式

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

c++ 内存池

上一篇:c# 读取excel数据 改进版

下一篇:web人力资源信息系统开发日志②

相关阅读

您好,登录后才能下订单哦!

密码登录
登录注册
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》