C++中的std::pmr是什么_C++17中基于多态内存资源的可定制内存分配

std::pmr是C++17引入的多态内存资源工具,通过memory_resource接口实现可定制的内存分配策略。它支持池分配、区域分配等高效机制,核心组件包括polymorphic_allocator、synchronized_pool_resource、monotonic_buffer_resource等,允许STL容器使用不同底层资源而不改变类型。通过set_default_resource可全局切换分配策略,适用于高性能场景如游戏引擎和服务器。

std::pmr(Polymorphic Memory Resource,多态内存资源)是 C++17 引入的一个内存管理工具集,位于 头文件中。它的核心目标是提供一种灵活、高效且可定制的内存分配机制,特别适用于性能敏感或需要精细控制内存行为的场景,比如游戏引擎、高性能服务器或嵌入式系统。

什么是 std::pmr?

std::pmr 基于“内存资源”(memory resource)的概念,将内存分配和释放操作抽象为一个统一接口 std::pmr::memory_resource。通过继承这个基类,可以实现不同的内存分配策略,比如池分配、栈分配、区域分配等。

最关键的一点是:它实现了“多态”分配——代码可以在不改变类型的情况下,使用不同的底层分配器。这种多态性不是通过虚函数表实现的运行时多态,而是通过对象指针传递实现的行为多态。

核心组件包括:

  • std::pmr::memory_resource:抽象基类,定义了 allocatedeallocate 接口。
  • std::pmr::pool_options:用于配置池分配器的参数(如最大块大小、池数量等)。
  • std::pmr::get_default_resource():获取当前默认的内存资源(通常指向全局 new/delete)。
  • std::pmr::set_default_resource():设置新的默认内存资源。
  • 预定义资源类型,如:std::pmr::synchronized_pool_resourcestd::pmr::unsynchronized_pool_resourcestd::pmr::monotonic_buffer_resource 等。

常见的 pmr 内存资源类型

标准库提供了几种开箱即用的内存资源实现,适合不同用途:

  • synchronized_pool_resource:线程安全的池分配器,适合多线程环境。小对象分配非常快,减少内存碎片。
  • unsynchronized_pool_resource:非线程安全版本,单线程下性能更高。
  • monotonic_buffer_resource:也叫“arena allocator”(区域分配器)。从一个大缓冲区中顺序分配内存,仅在销毁时一次性释放所有内存。非常适合短生命周期的大量临时对象。

例如,使用 monotonic_buffer_resource 可以极大提升解析器或渲染器中临时数据结构的分配速度。

如何使用 std::pmr?

实际使用中,std::pmr 通常与 STL 容器配合,通过 std::pmr::polymorphic_allocator 来实现。

这个分配器包装了一个 memory_resource*,并将分配请求转发给该资源。所有支持 AllocatorAware 的容器(如 vector、string、map)都可以使用它。

示例:使用池分配器创建 vector

#include 
#include 
#include 

int main() { // 创建一个线程安全的池资源 std::pmr::synchronized_pool_resource pool;

// 使用 pool 的分配器创建 vector
std::pmr::vector vec(&pool);
for (int i = 0; i < 100; ++i) {
    vec.push_back(i);
}

std::cout << "Size: " << vec.size() << "\n";
// 析构时自动释放内存回 pool

}

在这个例子中,vector 的所有内存都由 pool 管理。当 vector 被销毁,内存返回给池,而不是直接调用 delete。

自定义内存资源

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

简单示例:一个只分配一次的大缓冲区资源

struct simple_arena : std::pmr::memory_resource {
    char* buffer;
    size_t capacity;
    size_t used = 0;
simple_arena(char* buf, size_t size) : buffer(buf), capacity(size) {}

protected: void do_allocate(size_t bytes, size_t) override { if (used + bytes > capacity) throw std::bad_alloc(); void ptr = buffer + used; used += bytes; return ptr; }

void do_deallocate(void* p, size_t bytes, size_t) override {
    // 不真正释放,monotonic 特性
    if (p == buffer + used - bytes)
        used -= bytes;
}

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

};

这种资源适合一次性批量处理任务,避免频繁系统调用。

基本上就这些。std::pmr 提供了一套现代、灵活的内存管理方式,让开发者能根据应用场景选择最优分配策略,同时保持接口一致性。它不复杂但容易忽略细节,掌握后能显著提升程序性能和可控性。