三、定长内存池

三、定长内存池

我们知道申请内存使用的是malloc，malloc其实就是一个通用的大众货，什么场景下都可以使用，而什么场景下都可以用就意味着什么场景下都不会有很高的性能，下面我们就先来设计一个定长内存池作为一个开胃菜，当然这个定长内存池在后面的高并发内存池中也是有价值的，所以学习他目的有两层，第一是先熟悉一下简单内存池是如何控制的，第二是它会作为我们后面内存池的一个基础组件。

#pragma once

#include <iostream>
#include <assert.h>
using std::cout;
using std::endl;


#include "Common.h"

#ifdef _WIN32
#include <windows.h>
#endif // _WIN32

inline static void* SystemAlloc(size_t Kpage)
{
#ifdef _WIN32

	void* ptr= VirtualAlloc(0, Kpage * (1 << 13), MEM_COMMIT | MEM_RESERVE,PAGE_READWRITE);
#else
	//其他环境
#endif // _WIN32

	if (ptr == nullptr)
	{
		throw std::bad_alloc();
	}

	return ptr;
}

template <class T>
class ObjectPool
{
public:
	T* New()
	{
		T* obj = nullptr;

		//如果在自由链表中有还回来的内存块，则优先使用
		if (_freeList != nullptr)
		{
			//从_freeList链表头删一个内存块，顺便返回
			//_freeList指向的是自由链表的头上的小对象，这个小对象
			//的前4个或者8个字节(看机器环境)存放着下一个小对象的地址
			//*(void**)就拿到了下一个小对象的地址
			void* next = *(void**)_freeList;

			//取_freeList自由链表的头上T类型大小个字节赋值给obj即可
			obj = (T*)_freeList;

			//把_freeList更新为next相当于在_freeList中头删一个小对象
			_freeList = next;
		}
		else
		{
			//如果剩余的内存不够一个对象的大小，则需要重新向堆申请内存
			if (_RemainBytes < sizeof(T))
			{
				_RemainBytes = 128 * 1024;
				//_memory = (char*)malloc(_RemainBytes);

				//按页为单位向堆申请内存，一页8k，1k是1024字节，所以一页是2^13次方字节
				//所以字节数_Remaintypes右移13位就得到页数
				//直接调用系统调用接口SystemAlloc向堆申请内存，使它完全脱离开malloc函数
				_memory = (char*)SystemAlloc(_RemainBytes >> 13);
				if (_memory == nullptr)
				{
					//申请内存失败就抛异常
					throw std::bad_alloc();
				}
			}
			//在_memory中切一个T类型大小的对象给obj
			obj = (T*)_memory;
			//如果T的的大小太小的话就要多给一点内存，使它拥有能够放下一个指针变量大小的空间
			size_t objSize = sizeof(T) > sizeof(void*) ? sizeof(T) : sizeof(void*);
			//切走了一块空间就要使地址+=一个对象大小
			_memory += objSize;
			//剩余的字节数也要-=一个对象的大小
			_RemainBytes -= objSize;
		}

		//对已有的内存，利用定位new调用T的默认构造函数初始化
		new(obj)T;

		return obj;
	}

	void Delete(T* obj)
	{
		//显式调用T的析构函数清理还回来的内存块的资源
		obj->~T();

		//把还回来的内存块头插到自由链表中
		*(void**)obj = _freeList;
		_freeList = obj;
	}
private:
	char* _memory = nullptr;//向堆申请的大块内存
	void* _freeList = nullptr;//用完之后还回来的内存块连接而成的自由链表
	size_t _RemainBytes = 0;//申请的大块内存用过之后剩余的字节数
};