C++实现简单内存池

代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;

class Player
{
 public:
    int score;
    int age;
    // 内存池
    static char* pool;
    Player(int _score, int _age): score(_score), age(_age) {}
    static bool initPool()
    {
        // 向系统申请18字节的内存
        // 我们一个对象需要9个字节，第一个字节表示当前内存是否被占用
        pool = (char*)malloc(18);
        // 如果申请失败就返回false
        if (pool == 0) return false;
        cout << "内存池申请成功！！！" << endl;
        return true;
    }
    static void freePool()
    {
        if (pool == 0) return ;
        free(pool);
        cout << "内存池释放成功！！！" << endl;
    }
    // 重载new使得能够向内存池申请内存
    // 注意重载的new以及delete都是静态成员函数，无法访问非静态成员变量
    // 因此将内存池pool声明成静态成员变量
    void* operator new(size_t size)
    {
        // 如果内存池内的第一块内存没有被占用，我们就用这块内存
        if (pool[0] == 0)
        {
            pool[0] = 1; // 将当前内存标记为被占用的状态
            cout << "内存池中第一块内存分配成功！！！" << endl;
            return pool + 1;  // 表示从下一个字节开始存放对象数据
        }
        // 同理内存池中的第二块内存没有被占用，我们就用这块内存
        if (pool[9] == 0)
        {
            pool[9] = 1;
            cout << "内存池中第二块内存分配成功！！！" << endl;
            return pool + 10;
        }
        // 如果内存池中没有剩余的内存，那么就向系统申请内存
        void* ptr = (void*)malloc(size);
        cout << "向系统申请一块内存！！！" << endl;
        return ptr;
    }
    // 重载delete
    void operator delete(void* ptr)
    {
        if (ptr == 0) return ;
        // 如果传进来的内存是内存池第一块内存
        if (ptr == pool + 1)
        {
            // 将第一块内存标志清除
            pool[0] = 0;
            cout << "内存池中第一块内存释放成功！！！" << endl;
            return ;
        }
        if (ptr == pool + 10)
        {
            pool[9] = 0;
            cout << "内存池中第二块内存释放成功！！！" << endl;
            return ;
        }
        // 释放系统内存
        free(ptr);
        cout << "系统内存释放成功！！！" << endl;
    }
};

// 静态成员变量需要在全局进行初始化
char* Player::pool = 0;


int main()
{
    // 申请一块内存池
    Player::initPool();
    Player* p1 = new Player(39, 15);
    Player* p2 = new Player(59, 67);
    Player* p3 = new Player(93, 45);
    delete p2;
    Player* p4 = new Player(47, 18);
    delete p1;
    delete p3;
    delete p4;
    Player::freePool();
    
    return 0;
}

控制台输出

内存池申请成功！！！
内存池中第一块内存分配成功！！！
内存池中第二块内存分配成功！！！
向系统申请一块内存！！！
内存池中第二块内存释放成功！！！
内存池中第二块内存分配成功！！！
内存池中第一块内存释放成功！！！
系统内存释放成功！！！
内存池中第二块内存释放成功！！！
内存池释放成功！！！