메모리 풀 구현 (C++)

지난번에 공부한 내용을 바탕으로 간단한 메모리 풀을 만들어 보았습니다.

#pragma once

#include <iostream>
#include <vector>
#include <cassert>

class CMemoryPool
{
private:
	size_t m_blockSize;
	size_t m_blockCount;

	std::vector<char>  m_pool;        // 메모리 풀
	std::vector<void*> m_freeBlocks;	// 할당 가능한 메모리 주소

public:
	CMemoryPool(size_t _blockSize, size_t _blockCount)
		: m_blockSize(_blockSize)
		, m_blockCount(_blockCount)
	{
		m_pool.resize(m_blockCount * m_blockSize);

		for (size_t i = 0; i < _blockCount; ++i)
		{
			m_freeBlocks.push_back(&m_pool[i * m_blockSize]);
		}
	}

	void* allocate()
	{
		// 더 이상 할당 불가하면 nullptr 반환
		if (m_freeBlocks.empty())
			return nullptr;
    
		void* block = m_freeBlocks.back();
		m_freeBlocks.pop_back();

		return block;
	}

	void deallocate(void* _block)
	{
		// _block의 주소가 m_pool의 범위 밖이면 경고
		assert(_block >= m_pool.data() && _block < m_pool.data() + m_pool.size());
		// m_pool의 주소 ~ _block의 주소 크기가 m_blockSize로 나눠떨어지지 않으면 경고
		assert(0 == ((char*)_block - m_pool.data()) % m_blockSize);
    
		// 반환
		m_freeBlocks.push_back(_block);
	}
};

 

메모리 풀을 사용했을 때와 메모리 풀을 사용하지 않고 malloc을 이용하여 할당 했을 때 속도 차이를 비교해보았습니다.

메모리 풀을 사용했을 때 할당 시간이 대략 2배 정도 빠른 것을 확인할 수 있었습니다.

https://github.com/beankong-github/MemoryPool

GitHub - beankong-github/MemoryPool: 간단한 C++ 메모리 풀 제작해보기

테스트 코드는 깃허브에서 확인할 수 있습니다.