
std::pmr::memory_resource 是什么,为什么不能直接 new/delete
它是一个抽象基类,定义了 do_allocate、do_deallocate、do_is_equal 三个纯虚函数。你不能实例化它,也不能靠重载 operator new 来“替换”它——std::pmr 的所有容器(如 std::pmr::vector)只认 std::pmr::memory_resource* 指针,且全程通过虚函数调用,不依赖全局或类作用域的 new。
写一个最简可用的自定义 memory_resource
核心是继承 std::pmr::memory_resource,实现三个虚函数。注意:do_is_equal 必须判断是否为同一对象(不是“等价”,而是地址相等),否则 std::pmr::polymorphic_allocator 可能跳过缓存或误判资源归属。
class SimplePoolResource : public std::pmr::memory_resource {
private:
std::vector pool_;
size_t offset_ = 0;
protected:
void do_allocate(size_t bytes, sizet alignment) override {
if (bytes == 0) return nullptr;
auto ptr = std::align(alignment, bytes, pool.data() + offset, pool.size() - offset_);
if (!ptr) throw std::badalloc{};
offset = static_cast>(ptr) - pool_.data() + bytes;
return ptr;
}
void do_deallocate(void* p, size_t bytes, size_t alignment) override {
// 简单池不回收,什么也不做
}
bool do_is_equal(const std::pmr::memory_resource& other) const noexcept override {
return this == &other; // 必须是同一对象
}public:
explicit SimplePoolResource(sizet capacity) : pool(capacity) {}
};
这个实现只支持一次性分配(类似 arena),不支持释放单块内存,但完全合法——std::pmr 不要求 deallocate 真正归还空间。
如何让 std::pmr::vector 用上你的 resource
关键不是传 resource 给 vector 构造函数,而是通过 std::pmr::polymorphic_allocator 包装后传入。vector 自身不持有 resource 指针,allocator 才持有。
立即学习“C++免费学习笔记(深入)”;
-
std::pmr::vector 是类型别名,等价于 std::vector>
- 构造时必须显式传入 allocator,否则默认使用
std::pmr::get_default_resource()
- resource 生命周期必须长于所有使用它的容器
SimplePoolResource pool{1024 * 1024};
std::pmr::polymorphic_allocator alloc{&pool};
std::pmr::vector v{alloc}; // 正确:v 的所有内存来自 pool
v.reserve(1000); // 分配成功
// pool 析构前,v 不能继续 allocate 常见陷阱和兼容性注意点
C++17 标准库对 std::pmr 实现程度不一;libstdc++(GCC)直到 GCC 10 才完整支持,MSVC 2019 Update 3 起稳定,Clang 需搭配 libc++ 7.0+。更隐蔽的问题是:
- 某些 STL 容器内部会额外调用
resource()->is_equal(other->resource()) 做优化,若 do_is_equal 返回错误结果,可能导致未定义行为
-
std::pmr::synchronized_pool_resource 和 std::pmr::unsynchronized_pool_resource 是标准提供的可复用实现,但它们内部仍依赖 upstream_resource(),若你传入的 upstream 是临时对象,极易悬垂
- 调试时无法直接 print resource 地址来比对,因为
std::pmr::polymorphic_allocator 的 resource() 返回的是 const 指针,且 operator== 默认比较的是 resource 对象本身而非指针值
真正难的不是写 allocate/deallocate,而是保证 resource 对象的生存期、线程安全(如果多线程用)和 is_equal 的语义正确性。这三个点出错,往往表现为偶发崩溃或静默内存泄漏,而不是编译报错。










