自定义Allocator用于控制STL容器内存分配行为，以提升性能、降低碎片、调试内存问题或满足特殊需求。通过实现allocate/deallocate等接口，可编写如内存池或计数型Allocator，但需注意对象拷贝、线程安全及不依赖n值做边界检查等问题。
在C++中，STL容器（如vector、list、map等）默认使用全局的new和delete进行内存管理。但在某些特定场景下，比如性能敏感的应用、嵌入式系统或需要跟踪内存使用情况时，我们希望控制容器的内存分配行为。这时，自定义Allocator就派上用场了。
为什么需要自定义Allocator？
标准库中的容器模板都接受一个可选的Allocator模板参数，例如：
std::vector<int, MyAllocator<int>> vec;通过提供自己的Allocator，你可以：
- 提升性能：使用对象池、内存池减少频繁调用系统malloc/free
- 降低碎片：集中管理固定大小内存块
- 调试内存问题：记录分配/释放日志，检测泄漏或越界
- 满足特殊需求：如共享内存、非一致性内存访问（NUMA）等
如何编写一个简单的自定义Allocator
一个合法的Allocator必须满足一定的接口要求。从C++11开始，标准对Allocator的要求较为宽松（称为Minimal Allocator），但仍需实现一些关键成员。
立即学习“C++免费学习笔记（深入）”；
下面是一个基于内存池的简单示例：
template <typename T>
class SimplePoolAllocator {
public:
using value_type = T;
// 构造函数（必须支持无参构造）
SimplePoolAllocator() = default;
template <typename U>
SimplePoolAllocator(const SimplePoolAllocator<U>&) {}
T allocate(std::size_t n) {
void ptr = ::operator new(n sizeof(T));
return static_cast<T>(ptr);
}
void deallocate(T* ptr, std::size_t n) {
::operator delete(ptr);
}
};
// 必须提供这个特化判断，否则可能编译失败
template <typename T, typename U>
bool operator==(const SimplePoolAllocator<T>&, const SimplePoolAllocator<U>&) {
return true;
}
template <typename T, typename U>
bool operator!=(const SimplePoolAllocator<T>&, const SimplePoolAllocator<U>&) {
return false;
}
说明：
- value_type：必须定义，表示所分配类型的别名
- allocate/deallocate：核心函数，负责实际内存获取与归还
- 提供其他类型U的构造函数：允许不同模板实例之间转换
- 重载==和!=：用于比较两个Allocator是否可以互换
使用自定义Allocator的注意事项
虽然接口简单，但实际使用中要注意以下几点：
- Allocator对象通常会被拷贝，应避免持有独占资源（除非设计为不可拷贝）
- deallocate传入的n值不一定和allocate一致，不能依赖它做边界检查
- 多线程环境下需自行保证线程安全
- 不要在allocate中构造对象，在deallocate中析构——那是容器的工作
实战：带计数功能的调试Allocator
下面是一个用于统计分配次数和总字节数的调试工具：
template <typename T>
class DebugAllocator {
public:
using value_type = T;
DebugAllocator() = default;
template <typename U>
DebugAllocator(const DebugAllocator<U>&) {}
T allocate(std::size_t n) {
size_t bytes = n sizeof(T);
total_allocated += bytes;
alloc_count++;
T ptr = static_cast<T>(::operator new(bytes));
return ptr;
}
void deallocate(T* ptr, std::size_t) {
dealloc_count++;
::operator delete(ptr);
}
static size_t get_total() { return total_allocated; }
static size_t get_alloc_count() { return alloc_count; }
static size_t get_dealloc_count() { return dealloc_count; }
private:
static inline size_t total_allocated = 0;
static inline size_t alloc_count = 0;
static inline size_t dealloc_count = 0;
};
// 静态成员显式声明（C++17起inline已足够）
使用方式：
using VecInt = std::vector<int, DebugAllocator<int>>;VecInt v;
v.resize(100);
v.resize(200);
std::cout << "Total: " << DebugAllocator<int>::get_total() << " bytes\n";
std::cout << "Alloc count: " << DebugAllocator<int>::get_alloc_count() << "\n";
基本上就这些。自定义Allocator机制灵活但容易误用。建议先从简单封装开始，逐步深入到内存池、对象池等高级形式。关键是理解STL容器与Allocator之间的契约关系：容器负责对象构造/析构，Allocator只管原始内存。
