自定义内存分配器用于控制C++容器内存分配行为,通过实现标准接口如allocate、deallocate、construct等,可优化性能,常用于内存池、共享内存等场景,提升高频分配效率。
在C++中,自定义内存分配器(allocator)主要用于控制容器(如std::vector、std::list等)的内存分配行为。标准库中的容器都支持通
过模板参数传入自定义的allocator,从而实现更高效的内存管理,比如用于内存池、共享内存、栈上分配或性能优化场景。
C++标准要求一个合法的分配器必须满足一定的接口规范。最基本的成员包括:
value_type的指针value_type
从C++17起,construct和destroy不再是强制要求,而是推荐使用std::allocator_traits来统一调用。
下面是一个基于堆的简单分配器示例,功能类似std::allocator,但便于扩展:
templatestruct MyAllocator { using value_type = T; using pointer = T*; using const_pointer = const T*; using reference = T&; using const_reference = const T&; using size_type = std::size_t; using difference_type = std::ptrdiff_t; template zuojiankuohaophpcntypename Uyoujiankuohaophpcn struct rebind { using other = MyAllocatorzuojiankuohaophpcnUyoujiankuohaophpcn; }; MyAllocator() = default; template zuojiankuohaophpcntypename Uyoujiankuohaophpcn MyAllocator(const MyAllocatorzuojiankuohaophpcnUyoujiankuohaophpcn&) {} pointer allocate(size_type n) { if (n == 0) return nullptr; pointer p = static_castzuojiankuohaophpcnpointeryoujiankuohaophpcn(::operator new(n * sizeof(T))); // 可添加日志或统计 return p; } void deallocate(pointer p, size_type n) { ::operator delete(p); // 可记录释放信息 } template zuojiankuohaophpcntypename U, typename... Argsyoujiankuohaophpcn void construct(U* p, Args&&... args) { new(p) U(std::forwardzuojiankuohaophpcnArgsyoujiankuohaophpcn(args)...); } template zuojiankuohaophpcntypename Uyoujiankuohaophpcn void destroy(U* p) { p-youjiankuohaophpcn~U(); }};
这个分配器可用于STL容器:
std::vector> vec; vec.push_back(10); vec.push_back(20);
实际应用中,自定义分配器常用于减少频繁调用new/delete带来的开销。例如,实现一个简单的内存池:
templateclass PoolAllocator { private: struct Block { std::aligned_storage_t data; }; union Node { T data; Node* next; }; Node* free_list = nullptr; std::vectorzuojiankuohaophpcnBlockyoujiankuohaophpcn blocks; size_t current_offset = 0;public: using value_type = T; using pointer = T; using const_pointer = const T; using size_type = std::size_t;
template zuojiankuohaophpcntypename Uyoujiankuohaophpcn struct rebind { using other = PoolAllocatorzuojiankuohaophpcnU, BlockSizeyoujiankuohaophpcn; }; PoolAllocator() = default; ~PoolAllocator() = default; pointer allocate(size_type n) { if (n != 1) { throw std::bad_alloc{}; } if (!free_list) { refill_pool(); } Node* p = free_list; free_list = free_list-youjiankuohaophpcnnext; return reinterpret_castzuojiankuohaophpcnpointeryoujiankuohaophpcn(p); } void deallocate(pointer p, size_type n) { if (n != 1 || !p) return; Node* node = reinterpret_castzuojiankuohaophpcnNode*youjiankuohaophpcn(p); node-youjiankuohaophpcnnext = free_list; free_list = node; } template zuojiankuohaophpcntypename U, typename... Argsyoujiankuohaophpcn void construct(U* p, Args&&... args) { new(p) U(std::forwardzuojiankuohaophpcnArgsyoujiankuohaophpcn(args)...); } template zuojiankuohaophpcntypename Uyoujiankuohaophpcn void destroy(U* p) { p-youjiankuohaophpcn~U(); }private: void refill_pool() { Block& block = blocks.emplace_back(); char start = reinterpret_cast
>(&block.data); const size_t nodes_per_block = BlockSize / sizeof(Node); Node nodes = reinterpret_cast >(start); for (size_t i = 0; i zuojiankuohaophpcn nodes_per_block - 1; ++i) { nodes[i].next = &nodes[i + 1]; } nodes[nodes_per_block - 1].next = nullptr; free_list = &nodes[0]; }};
该分配器预先分配大块内存,并将对象以链表形式管理空闲节点,显著提升小对象频繁分配/释放的性能。
编写自定义分配器时需注意以下几点:
allocate返回的是未初始化的原始内存,不要调用构造函数deallocate不应调用析构,只负责释放内存allocate/deallocate中抛出异常,除非确实无法分配使用std::allocator_traits可以增强分配器的通用性,它提供了默认的construct和destroy实现,兼容旧式和新式分配器。
基本上就这些。自定义分配器是C++高性能编程的重要工具,合理使用可显著改善程序性能,尤其在高频分配场景下。