c++ std::pmr怎么用 c++多态内存资源【详解】

std::pmr是C++17引入的多态内存资源机制,通过memory_resource、polymorphic_allocator和pmr容器三层协作,实现内存分配策略与容器逻辑解耦,支持自定义分配行为。

std::pmr(Polymorphic Memory Resource)是 C++17 引入的内存资源抽象机制,核心目标是解耦容器的内存分配行为与具体实现,让 std::vectorstd::string 等容器能灵活切换底层内存策略(如栈分配、池分配、线程局部缓存等),而无需修改容器逻辑。

一、理解 std::pmr 的核心组件

它基于三个关键类型:

  • std::pmr::memory_resource:纯虚基类,定义 allocate/deallocateis_equal 接口,是所有自定义内存资源的统一入口;
  • std::pmr::polymorphic_allocator:模板分配器,持有 memory_resource*,将 std::allocator 风格接口(如 construct/destroy)转发给底层资源;
  • std::pmr::vector/string/map 等别名:标准容器的 PMR 版本,如 std::pmr::vector,本质是 std::vector> 的类型别名。

二、快速上手:用 std::pmr::monotonic_buffer_resource

这是最常用的内置资源,提供单向增长的缓冲区(类似“内存池”),适合短生命周期批量分配,无释放开销:

#include 
#include 

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

std::pmr::vector v{alloc}; // 所有元素和内部容量都从 pool 分配 v.push_back(1); v.push_back(2); // pool 析构时自动回收全部内存(不调用 individual deallocate)

注意:monotonic_buffer_resource 不支持单独 deallocate,只支持整体重置或析构释放。

三、自定义 memory_resource 示例

实现一个带日志的包装器,用于调试分配行为:

struct logging_resource : public std::pmr::memory_resource {
private:
    std::pmr::memory_resource* upstream_;

protected: void* do_allocate(std::size_t bytes, std::sizet align) override { std::cout << "Allocating " << bytes << " bytes, align=" << align << '\n'; return upstream->allocate(bytes, align); }

void do_deallocate(void* p, std::size_t bytes, std::size_t align) override {
    std::cout << "Deallocating " << bytes << " bytes\n";
    upstream_->deallocate(p, bytes, align);
}

bool do_is_equal(const memory_resource& other) const noexcept override {
    return this == &other || upstream_->is_equal(other);
}

public: explicit logging_resource(std::pmr::memory_resource* up = std::pmr::new_deleteresource()) : upstream(up) {} };

使用时只需传入该资源指针给 polymorphic_allocator,后续所有分配行为都会被拦截打印。

四、注意事项与常见陷阱

  • 资源生命周期必须长于使用它的分配器和容器:若 memory_resource 提前析构,再访问会导致未定义行为;
  • 不同资源之间默认不可互换:两个 monotonic_buffer_resource 实例通常 is_equal 返回 false,因此不能跨资源 deallocate
  • std::pmr::vector 的 capacity 变化可能触发新 allocation,但只要 allocator 持有同一 resource,仍走定制路径;
  • 不要混用普通 new/delete 与 pmr 容器:例如把 pmr::vector.data() 传给 free() 是错误的;
  • 多线程安全需自行保证:标准资源如 new_delete_resource 是线程安全的,但 monotonic_buffer_resource 默认不是,需配合 synchronized_pool_resource 或加锁。

std::pmr 不是银弹,但它为高性能、嵌入式或内存敏感场景提供了标准化的扩展能力。掌握 resource + allocator + pmr 容器三层协作关系,就能真正控制内存的来龙去脉。