C++ memory_resource怎么自定义STL内存分配器

夜敏大大_7400

夜敏大大_7400

2026-10-20

668人浏览

原创

memory_resource 是 c++17 引入的类型无关内存分配抽象接口,与 std::allocator 分层协作而非替代;必须通过 std::pmr::polymorphic_allocator 和 std::pmr 容器(如 pmr::vector)使用,不可直接用于传统模板 allocator。

c++ memory_resource怎么自定义stl内存分配器

memory_resource 是什么,和 allocator 有什么区别

memory_resource 是 C++17 引入的抽象接口(纯虚类),定义了 allocate / deallocate / is_equal 三个核心函数,不绑定具体类型,也不参与模板实例化。它和传统 std::allocator 不是替代关系,而是分层协作:STL 容器可通过 std::pmr::polymorphic_allocator 包装一个 memory_resource*,把类型无关的内存请求转发过去。

关键点在于:你不能直接把自定义 memory_resource 塞给 std::vector<int myallocator></int> 这种模板 allocator;必须走 std::pmr 路线,比如 std::pmr::vector<int></int>。

写一个最简可用的派生类(比如 arena_resource)

继承 std::memory_resource,实现三个虚函数即可。常见错误是忽略 is_equal —— 如果两个 resource 实例逻辑上等价(比如指向同一块预分配 buffer),必须返回 true,否则 std::pmr::polymorphic_allocator 可能绕过缓存或误判资源归属。

class arena_resource : public std::memory_resource {
    std::vector<:byte> storage_;
    size_t offset_ = 0;
<p>protected:
void<em> do_allocate(size_t bytes, size_t alignment) override {
const auto aligned_offset = align<em>offset(offset</em>, alignment);
if (aligned<em>offset + bytes > storage</em>.size()) throw std::bad_alloc{};
void</em> ptr = storage_.data() + aligned<em>offset;
offset</em> = aligned_offset + bytes;
return ptr;
}</p>
<pre class="brush:php;toolbar:false;">void do_deallocate(void* p, size_t, size_t) override {
    // arena 不支持 individual deallocation
}

bool do_is_equal(const std::memory_resource& other) const noexcept override {
    return this == &other; // 或按语义判断(如 shared_ptr 指向同一 arena)
}

public: explicit arena_resource(sizet cap) : storage(cap) {} };

注意:do_deallocate 空实现是合法的(arena 语义),但若容器在析构时调用 deallocate,而你的实现抛异常或崩溃,就会 UB。务必匹配资源语义。

怎么让 vector/string 等容器真正用上它

必须显式使用 std::pmr 版本,并传入对应 allocator:

C++14
C++14

C++14 对 C++11 的修正与增强版本,适合旧系统维护和较老工具链兼容。

下载
  • std::pmr::vector<int></int> 默认用 std::pmr::get_default_resource()
  • 要换资源,得构造时传 std::pmr::polymorphic_allocator:
    arena_resource ar(1024 * 1024);
    std::pmr::vector<int> v{std::pmr::polymorphic_allocator<int>{&ar}};
    v.push_back(42); // 内存来自 ar
    </int></int>

常见坑:

  • 忘记 std::pmr:: 前缀,写了 std::vector 就还是走全局 new
  • 把 polymorphic_allocator 当成普通 allocator 传给非 pmr 容器,编译失败
  • 在函数里临时创建 arena_resource,但容器生命周期更长,导致 use-after-free

和 set_default_resource 的配合与风险

std::pmr::set_default_resource(&my_res) 会影响所有未显式指定 allocator 的 std::pmr::* 对象。这看似方便,实则危险:

  • 全局状态,多线程下需加锁(set_default_resource 本身线程安全,但切换时机难控)
  • 第三方库若也用 pmr,可能意外被你的 resource 影响
  • 单元测试中容易污染后续用例

更稳妥的做法是局部显式传参,或者用 RAII 封装:

struct scoped_default {
    std::memory_resource* old_;
    scoped_default(std::memory_resource* r) : old_{std::pmr::get_default_resource()} {
        std::pmr::set_default_resource(r);
    }
    ~scoped_default() { std::pmr::set_default_resource(old_); }
};

memory_resource 的核心价值不在“替换 malloc”,而在控制内存生命周期和布局——比如 arena、pool、mmap 区域。一旦离开 pmr 生态链(比如混用 raw new 或 legacy allocator),整个设计就断掉了。

C++免费学习笔记(深入):立即使用
在学习笔记中,你将探索 C++ 的入门与实战技巧!

相关专题

更多
c++和c语言的区别有哪些
c++和c语言的区别有哪些

c++和c语言的区别:1、面向对象编程(OOP)支持不同;2、新增特性不同;3、标准库不同;4、编译方式不同;5、命名空间不同等等。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2024.03.14

2248

9

c++和python学习顺序推荐
c++和python学习顺序推荐

一般建议先学习C++,再学习Python,因为这样可以逐步从较为底层的编程语言向更高级的语言过渡。想了解更多python的相关内容,可以阅读本专题下面的文章。

2024.03.14

1019

6

python和c++学习性价比分析
python和c++学习性价比分析

Python易于学习,广泛应用于Web开发、数据科学和人工智能等领域,但性能较低。C语言性能高,适用于对性能要求较高的场景,如游戏开发和系统编程,但学习曲线陡峭,错误处理复杂。想了解更多python的相关内容,可以阅读本专题下面的文章。

2024.03.14

447

5

c语言和c++一样吗
c语言和c++一样吗

c语言和c++是两种不同的编程语言,虽然有相似之处,但存在显著差异。c语言专注于过程式编程和系统级开发,以简洁、高效著称。c++作为c语言的超集,引入了面向对象编程,增强了代码组织和管理能力,但学习曲线也更陡峭。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

327

5

c语言和c++先学哪个好
c语言和c++先学哪个好

初学者选择学习c语言还是c++语言,需要根据个人学习目标、背景以及编程兴趣和预期应用方向来决定。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

406

5

c语言和c++的区别和联系
c语言和c++的区别和联系

c语言和c++是计算机科学领域应用广泛的编程语言。虽然它们有着相似的基础,但它们在语言类型、语法功能和内存管理方面存在着显著差异。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

580

5

c++软件中文更改教程
c++软件中文更改教程

对于 ide,可通过打开设置,找到语言设置,选择中文,并保存更改。对于非 ide 应用程序,可查找设置或选项,选择语言设置,更改为中文,并保存更改。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2024.03.21

1409

9

python和java和c++学习性价比分析
python和java和c++学习性价比分析

Python以其易学性、丰富的库和活跃的社区而著称,适合数据科学、人工智能和Web开发。Java以其跨平台性、企业级应用开发和Android应用开发而闻名。C++以其底层控制能力、高效性能和游戏开发而著称。选择哪种语言取决于个人兴趣、职业方向和特定需求。想了解更多python和java和c++的相关内容,可以阅读本专题下面的文章。

2024.03.22

1217

7

c++和c语言学习顺序推荐
c++和c语言学习顺序推荐

对于初学者,建议先学习C语言,掌握编程基础后再转入C++,便于理解面向对象编程概念。有编程经验者可直接学习C++,快速接触高级编程技术。想了解更多c++和c语言的相关内容,可以阅读本专题下面的文章。

2024.03.25

1325

9

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
C++ 手册教程
C++ 手册教程

共41课时 | 78.4万人学习

C++ Core Guidelines
C++ Core Guidelines

共0课时 | 0人学习

C++ Reference
C++ Reference

共0课时 | 0人学习