【Cherno的C++视频】Track memory allocations the easy way in C++

原创于 2022-08-16 19:40:38 发布 · 153 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++

Cherno的《C++教程》专栏收录该内容

87 篇文章

订阅专栏

这篇博客展示了如何使用C++自定义内存分配器来跟踪内存使用情况。通过`new`和`delete`操作符的重载，实现了全局的内存分配和释放计数。在创建字符串和智能指针对象时，可以看到内存使用量的变化。这有助于理解和优化程序的内存消耗。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <iostream>
#include <string>
#include <memory>

struct AllocationMetrics
{
	uint32_t TotalAllocated = 0;
	uint32_t TotalFreed = 0;

	uint32_t CurrentUsage() { return TotalAllocated - TotalFreed; }
};

static AllocationMetrics s_AllocationMetrics;

static void PrintMemoryUsage()
{
	std::cout << "Memory Usage: " << s_AllocationMetrics.CurrentUsage() << " bytes\n";
}

void* operator new(size_t size)
{
	s_AllocationMetrics.TotalAllocated += size;
	return malloc(size);
}

void operator delete(void* memory, size_t size)
{
	s_AllocationMetrics.TotalFreed += size;
	free(memory);
}

struct Object
{
	int x, y, z;
};

int main(void)
{
	PrintMemoryUsage();
	std::string str = "Cherno";//8 bytes
	PrintMemoryUsage();
	{
		std::unique_ptr<Object> obj = std::make_unique<Object>();//12 bytes
		PrintMemoryUsage();// 20 bytes.
	}
	PrintMemoryUsage();
	
	std::cin.get();
}