C++memory_resource抽象内存资源基类

2026-04-01 22:40:25 969阅读 0评论

C++ memory_resource 抽象内存资源基类详解

在现代C++编程中，内存管理是一个至关重要的问题。为了提高代码的灵活性和性能，C++标准库引入了std::pmr（Pointer Memory Resources）模块，其中最核心的概念就是memory_resource抽象基类。本文将详细介绍memory_resource及其子类，帮助你更好地理解和应用这一强大工具。

什么是 `memory_resource`？

memory_resource是C++17引入的一个抽象基类，位于头文件 <memory_resource> 中。它定义了一个内存分配接口，允许程序员自定义内存分配策略。通过继承memory_resource并实现其纯虚函数，你可以创建自己的内存池、缓存或其他高级内存管理方案。

主要功能

内存分配：allocate(size_t bytes, size_t alignment = alignof(std::max_align_t))
- 分配指定大小和对齐方式的内存块。
内存释放：deallocate(void* p, size_t bytes, size_t alignment)
- 释放之前分配的内存块。
获取内存资源名称：do_name() const noexcept
- 返回当前内存资源的名称。
获取父资源：do_is_equal(const memory_resource& other) const noexcept
- 判断当前资源是否等于另一个资源。

如何使用 `memory_resource`

使用memory_resource非常简单，只需继承并实现其纯虚函数即可。下面是一个简单的示例，展示如何创建一个自定义的内存资源：

#include <iostream>
#include <memory_resource>

class MyMemoryResource : public std::pmr::memory_resource {
public:
    void* do_allocate(std::size_t bytes, std::size_t alignment) override {
        // 自定义内存分配逻辑
        return std::aligned_alloc(alignment, bytes);
    }

    void do_deallocate(void* p, std::size_t bytes, std::size_t alignment) override {
        // 自定义内存释放逻辑
        std::free(p);
    }

    bool do_is_equal(const std::pmr::memory_resource& other) const noexcept override {
        // 判断当前资源是否等于另一个资源
        return this == &other;
    }
};

int main() {
    MyMemoryResource myRes;
    std::pmr::vector<int> vec(&myRes);

    for (int i = 0; i < 10; ++i) {
        vec.push_back(i);
    }

    for (const auto& elem : vec) {
        std::cout << elem << " ";
    }
    std::cout << std::endl;

    return 0;
}

在这个示例中，我们创建了一个名为MyMemoryResource的类，继承自std::pmr::memory_resource，并实现了do_allocate、do_deallocate和do_is_equal三个纯虚函数。然后，我们在main函数中使用这个自定义内存资源来创建一个std::pmr::vector。

常见的 `memory_resource` 子类

除了自定义内存资源外，C++标准库还提供了几个常用的memory_resource子类，它们可以帮助你更高效地管理内存。

`std::pmr::monotonic_buffer_resource`

monotonic_buffer_resource是一种单次分配内存资源，适用于需要连续内存分配的场景。它内部维护一个缓冲区，当缓冲区满时会重新分配一个新的缓冲区。

#include <iostream>
#include <memory_resource>
#include <string>

int main() {
    std::pmr::monotonic_buffer_resource bufRes(1024);
    std::pmr::string str("Hello, World!", &bufRes);

    std::cout << str << std::endl;

    return 0;
}

`std::pmr::pool_options`

pool_options用于配置内存池的参数，如每个桶的大小、桶的数量等。你可以将其传递给std::pmr::unsynchronized_pool_resource或std::pmr::synchronized_pool_resource，以优化内存分配效率。

#include <iostream>
#include <memory_resource>
#include <string>

int main() {
    std::pmr::pool_options options;
    options.max_blocks_per_chunk = 16;
    options.largest_required_pool_block = 1024;

    std::pmr::unsynchronized_pool_resource poolRes(options);
    std::pmr::string str("Hello, World!", &poolRes);

    std::cout << str << std::endl;

    return 0;
}

`std::pmr::new_delete_resource`

new_delete_resource是一个基于全局new和delete操作符的内存资源，适用于不需要特殊内存管理策略的场景。

#include <iostream>
#include <memory_resource>
#include <string>

int main() {
    std::pmr::new_delete_resource newDelRes;
    std::pmr::string str("Hello, World!", &newDelRes);

    std::cout << str << std::endl;

    return 0;
}