std::pmr是C++17引入的多态内存资源机制,通过memory_resource、polymorphic_allocator和pmr容器三层协作,实现内存分配策略与容器逻辑解耦,支持自定义分配行为。
std::pmr(Polymorphic Memory Resource)是 C++17 引入的内存资源抽象机制,核心目标是解耦容器的内存分配行为与具体实现,让 std::vector、std::string 等容器能灵活切换底层内存策略(如栈分配、池分配、线程局部缓存等),而无需修改容器逻辑。
它基于三个关键类型:
allocate/deallocate 和 is_equal 接口,是所有自定义内存资源的统一入口;memory_resource*,将 std::allocator 风格接口(如 construct/destroy)转发给底层资源;std::pmr::vector,本质是 std::vector> 的类型别名。这是最常用的内置资源,提供单向增长的缓冲区(类似“内存池”),适合短生命周期批量分配,无释放开销:
#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,只支持整体重置或析构释放。
实现一个带日志的包装器,用于调试分配行为:
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;new_delete_resource 是线程安全的,但 monotonic_buffer_resource 默认不是,需配合 synchronized_pool_resource 或加锁。std::pmr 不是银弹,但它为高性能、嵌入式或内存敏感场景提供了标准化的扩展能力。掌握 resource + allocator + pmr 容器三层协作关系,就能真正控制内存的来龙去脉。