【C++】内存池

C++实现内存池

#include<iostream>
#include<list>
#include<cassert>

class MemoryPool {
private:
	struct Block {
		Block* next;   //内存池的基本单元 ，每一个block包含一个指向下一个block的指针，使得所有内存块通过单链表连接起来
	};
	
	Block* freeList;//用于管理内存池中空闲内存块的链表的头指针
	//内存池大小
	size_t blockSize;
	//总块数
	size_t blockCount;

public:
	MemoryPool(size_t size, size_t count) :
		blockSize(size), blockCount(count), freeList(nullptr)
	{
		//初始化内存池
		size_t poolSize = blockSize * blockCount;
		freeList = (Block*)(std::malloc(poolSize));
		//freeList = reinterpret_cast<Block*>(std::malloc(poolSize));//reinterpret_cast是类型转换符，用于将新分配的类型转化为Block*指针类型
		//初始化空闲链表
		Block* current = freeList;
		for (size_t i = 0; i < blockCount - 1; ++i) {
			current->next = reinterpret_cast<Block*>(reinterpret_cast<char*>(current));
			current = current->next;
		}
		current->next = nullptr;
	}

	~MemoryPool() {
		std::free(freeList);
	}

	void* allocate() {
		if (!freeList) {
			return nullptr;
		}
		Block* head = freeList;
		freeList = head->next;
		return head;
	}
	//释放之前存储的内存块
	void deallocate(void* ptr) {
		Block* block = static_cast<Block*>(ptr);
		block->next = freeList;
		freeList = block;
	}
};

class smallInt {
public :
	explicit operator int() const {
		return val;
	}

	smallInt(int i=0):val(i) {
		if (i < 0 || i>255) {
			std::cout << "error"<<std::endl;
		}
	}

private:
	size_t val;

};


int main() {
	MemoryPool pool(128, 10);
	void* p = pool.allocate();


	std::cout << p << std::endl;
	pool.deallocate(p);
	std::cout << p << std::endl;

	return 0;
}