欢迎来到雄杰鑫电商资讯网
技术教程

C++怎么自定义STL容器的内存分配器_C++ Allocator与内存池技术

穿越時空 次阅读
自定义STL容器内存分配需实现符合标准的Allocator，通过重写allocate/deallocate控制内存行为，结合内存池可减少系统调用开销；示例中MyAllocator添加日志功能，PoolAllocator预分配大块内存管理小对象，提升性能；适用于list、map等节点型容器，注意rebind、线程安全及C++17后construct/destroy废弃问题。

在C++中，自定义STL容器的内存分配方式可以通过实现一个符合标准的 Allocator 来完成。这不仅让你能控制内存的分配与释放行为，还能结合内存池技术提升性能，减少频繁调用 ::operator new 和 ::operator delete 带来的开销。

Allocator的基本概念

STL中的容器（如 std::vector、std::list 等）都接受一个可选的模板参数——allocator类型。默认使用 std::allocator，它封装了 new 和 delete。你可以提供自己的分配器来改变内存管理策略。

一个合法的Allocator必须满足一定的接口要求：

  • value_type：被分配对象的类型
  • pointer、const_pointer、reference、const_reference、size_type、difference_type
  • rebind：允许为不同类型重新绑定分配器
  • allocate(n)：分配未初始化的内存，用于构造n个对象
  • deallocate(p, n)：释放由allocate返回的内存块
  • 支持两个实例间相等性比较（通常所有实例都视为相等）

从C++11开始，还建议实现 construct 和 destroy 方法，但C++17后这些已被弃用，改由容器内部直接调用 placement new 和析构函数。

写一个简单的自定义分配器

下面是一个基于堆的简单分配器示例，功能类似 std::allocator，但可以加入日志或检查逻辑：

template 
struct MyAllocator {
    using value_type = T;
    using pointer = T*;
    using const_pointer = const T*;
    using reference = T&;
    using const_reference = const T&;
    using size_type = std::size_t;
    using difference_type = std::ptrdiff_t;

template zuojiankuohaophpcntypename Uyoujiankuohaophpcn
struct rebind {
    using other = MyAllocatorzuojiankuohaophpcnUyoujiankuohaophpcn;
};

MyAllocator() = default;
template zuojiankuohaophpcntypename Uyoujiankuohaophpcn
MyAllocator(const MyAllocatorzuojiankuohaophpcnUyoujiankuohaophpcn&) {}

pointer allocate(size_type n) {
    std::cout zuojiankuohaophpcnzuojiankuohaophpcn "Allocating " zuojiankuohaophpcnzuojiankuohaophpcn n zuojiankuohaophpcnzuojiankuohaophpcn " elements of size " zuojiankuohaophpcnzuojiankuohaophpcn sizeof(T) zuojiankuohaophpcnzuojiankuohaophpcn std::endl;
    return static_castzuojiankuohaophpcnpointeryoujiankuohaophpcn(::operator new(n * sizeof(T)));
}

void deallocate(pointer p, size_type n) {
    std::cout zuojiankuohaophpcnzuojiankuohaophpcn "Deallocating " zuojiankuohaophpcnzuojiankuohaophpcn n zuojiankuohaophpcnzuojiankuohaophpcn " elements" zuojiankuohaophpcnzuojiankuohaophpcn std::endl;
    ::operator delete(p);
}
};
使用这个分配器创建 vector：


std::vector> vec;
vec.push_back(42); // 触发 allocate

结合内存池优化性能


频繁小对象分配会带来性能问题和内存碎片。通过自定义分配器集成内存池，可以显著提升效率。


基本思路是：预先申请一大块内存，按固定大小切分成槽位，每次分配从池中取出空闲块，避免系统调用。


这里是一个简化版的内存池分配器框架：


template 
class PoolAllocator {
private:
    struct Node {
        Node* next;
    };

Node* free_list = nullptr;
char* pool = nullptr;
size_t pool_size = 0;

void refill_pool() {
    size_t num_objects = BlockSize / sizeof(T);
    pool = new char[BlockSize];
    pool_size = BlockSize;

    // 将内存块链成自由链表
    free_list = reinterpret_castzuojiankuohaophpcnNode*youjiankuohaophpcn(pool);
    for (size_t i = 0; i zuojiankuohaophpcn num_objects - 1; ++i) {
        free_list[i].next = &free_list[i + 1];
    }
    free_list[num_objects - 1].next = nullptr;
}
public:
using value_type = T;
using pointer = T;
using const_pointer = const T;
using size_type = std::size_t;
template zuojiankuohaophpcntypename Uyoujiankuohaophpcn
struct rebind {
    using other = PoolAllocatorzuojiankuohaophpcnU, BlockSizeyoujiankuohaophpcn;
};

PoolAllocator() { refill_pool(); }
~PoolAllocator() { delete[] pool; }

template zuojiankuohaophpcntypename Uyoujiankuohaophpcn
PoolAllocator(const PoolAllocatorzuojiankuohaophpcnU, BlockSizeyoujiankuohaophpcn&) {}

pointer allocate(size_type n) {
    if (n != 1 || sizeof(T) youjiankuohaophpcn BlockSize) {
        throw std::bad_alloc();
    }
    if (!free_list) {
        refill_pool(); // 可扩展为多块管理
    }
    Node* node = free_list;
    free_list = free_list-youjiankuohaophpcnnext;
    return reinterpret_castzuojiankuohaophpcnpointeryoujiankuohaophpcn(node);
}

void deallocate(pointer p, size_type) {
    Node* node = reinterpret_castzuojiankuohaophpcnNode*youjiankuohaophpcn(p);
    node-youjiankuohaophpcnnext = free_list;
    free_list = node;
}
};
该分配器适合固定大小的小对象，比如节点类容器（list、map）。注意它只支持单对象分配（n=1），不能用于数组类批量分配场景。


实际使用与注意事项


将上述内存池分配器用于 std::list 或 std::unordered_map 能有效减少碎片：


std::list> my_list;
my_list.push_back(1);
my_list.push_back(2);

关键点提醒：


    

  • 确保 rebind 正确实现，因为STL容器内部可能需要为指针或节点类型创建对应分配器
    • 

  • 多个容器实例共用同一分配器类型时，其内存不应交叉释放（除非设计为全局池）
    • 

  • 线程安全需自行保证，标准不强制分配器线程安全
    • 

  • C++17起，construct 和 destroy 不再被调用，直接使用 placement new
    • 

  • 调试时可在 allocate/deallocate 中加入计数、地址记录等辅助信息
    • 



基本上就这些。自定义分配器加内存池是一种成熟的优化手段，尤其适用于高频小对象分配的场景，如游戏引擎、网络服务等对性能敏感的系统。






            
 还能 你可以 是一个 自己的 多个 是一种 适用于 接口 自定义 切分 对象 线程 red 堆 封装 map node public c++ const 指针 using delete operator pointer 析构函数 内存管理

相关文章

按ESC键退出。