c++的std::pmr::polymorphic_allocator怎么用_c++运行时多态内存分配

std::pmr::polymorphic_allocator通过绑定memory_resource实现运行时多态内存分配,委托具体分配行为给memory_resource,支持切换策略而不改变容器类型。使用时需包含,创建如monotonic_buffer_resource等资源实例,再通过std::pmr::vector、std::pmr::string等容器传入资源指针,实现内存池或自定义分配。可提升性能、减少碎片、便于追踪,但需注意resource生命周期长于allocator、避免跨resource混用及线程安全问题。

std::pmr::polymorphic_allocator 是 C++17 引入的内存资源(Memory Resource)体系中的一部分,用于实现运行时多态的内存分配。它不直接管理内存,而是通过绑定一个 std::pmr::memory_resource,在运行时决定具体的分配行为。这种机制让你可以在不改变模板实例或容器类型的情况下,切换不同的内存分配策略。

基本原理

std::pmr::polymorphic_allocator 是一个模板类,位于 头文件中。它的关键特性是:

  • 本身不执行分配,而是委托给一个 memory_resource 对象。
  • 可以和其他使用相同 memory_resource 的容器共享内存池。
  • 常用于 STL 容器如 vector、string、map 等,替换默认的 new/delete 分配方式。

如何使用 polymorphic_allocator

实际使用中,通常不会直接构造 polymorphic_allocator 实例,而是通过 std::pmr 容器间接使用。以下是具体步骤和示例:

1. 包含头文件

立即学习“C++免费学习笔记(深入)”;

需要引入 memory resource 相关支持:

#include 
#include 
#include 

2. 创建 memory_resource

选择或定义一个 memory_resource,例如使用内置的 monotonic_buffer_resource(基于内存池的连续分配器):

char buffer[1024];
std::pmr::monotonic_buffer_resource pool{buffer, sizeof(buffer)};

这会在栈上预留一块缓冲区作为内存池,优先使用这块区域分配内存,不够时可回退到 std::pmr::get_default_resource()(通常是堆)。

3. 使用 pmr 容器

std::pmr 提供了常用容器的别名版本,内部使用 polymorphic_allocator:

std::pmr::vector vec(&pool);
vec.push_back(1);
vec.push_back(2);

std::pmr::string str("hello", &pool);

上面的 vec 和 str 都会通过 &pool 进行内存分配,而不是直接调用 new。

4. 自定义 memory_resource

你也可以继承 std::pmr::memory_resource 实现自己的分配逻辑:

struct MyResource : std::pmr::memory_resource {
    void* do_allocate(std::size_t bytes, std::size_t alignment) override {
        return ::operator new(bytes, std::align_val_t{alignment});
    }
void do_deallocate(void* p, std::size_t bytes, std::size_t alignment) override {
    ::operator delete(p, bytes, std::align_val_t{alignment});
}

bool do_is_equal(const memory_resource& other) const noexcept override {
    return this == &other;
}

};

然后将你的 resource 实例传给 pmr 容器即可。

常见用途与优势

  • 性能优化:配合 monotonic_buffer_resource 可避免频繁系统调用,提升小对象分配速度。
  • 内存追踪:自定义 resource 可记录分配/释放日志,便于调试内存问题。
  • 嵌入式场景:在无堆环境或固定内存预算下,使用预分配缓冲区更安全。
  • 减少碎片:内存池式分配能降低堆碎片。

注意事项

  • memory_resource 的生命周期必须长于使用它的 allocator。
  • 不同 resource 之间的内存不能混用(do_is_equal 控制相等性)。
  • 标准未规定线程安全性,多线程使用需自行同步 resource 实现。

基本上就这些。通过 std::pmr::polymorphic_allocator 和 memory_resource 体系,C++ 实现了灵活的运行时内存分配策略切换,既保持接口统一,又支持高性能定制。不复杂但容易忽略的是 resource 生命周期管理和对齐处理。