标准库Allocator的简易实现（二）

自己实现Allocator并不难，其实只需要改变allocate和deallocate，来实现自己的内存分配策略。

 

下面是一个std::allocator的模拟实现

#ifndef ALLOCATOR_HPP
#define ALLOCATOR_HPP

#include <stddef.h>
#include <limits>

template <typename T>
class Allocator
{
public:
    typedef size_t size_type;
    typedef ptrdiff_t difference_type;
    typedef T*  pointer;
    typedef const T* const_pointer;
    typedef T& reference;
    typedef const T& const_reference;
    typedef T value_type;

    //Allocator::rebind<T2>::other
    template <typename V>
    struct rebind
    {
        typedef Allocator<V> other;
    };

    pointer address(reference value) const { return &value; }
    const_pointer address(const_reference value) const { return &value; }

    Allocator() throw() {   }
    Allocator(const Allocator &) throw() {  }
    //不同类型的allcator可以相互复制
    template <typename V> Allocator(const Allocator<V> &other) { } 
    ~Allocator() throw() {  }

    //最多可以分配的数目
    size_type max_size() const throw()
    { return std::numeric_limits<size_type>::max() / sizeof(T); }

    //分配内存，返回该类型的指针
    pointer allocate(size_type num)
    { return (pointer)(::operator new(num * sizeof(T))); }

    //执行构造函数，构建一个对象
    void construct(pointer p, const T &value)
    { new ((void*)p) T(value); }

    //销毁对象
    void destroy(pointer p)
    { p->~T(); }

    //释放内存
    void deallocate(pointer p, size_type num)
    { ::operator delete((void *)p); }
};

//这两个运算符不需要friend声明
template <typename T, typename V>
bool operator==(const Allocator<T> &, const Allocator<V> &) throw()
{ return true; }

template <typename T, typename V>
bool operator!=(const Allocator<T> &, const Allocator<V> &) throw()
{ return false; }


#endif

这里注意rebind的实现，如果需要使用Test的分配器分配其他类型，就可以这样：

Allocator<Test>::rebind<Test2>::other alloc;

测试代码如下：

#include "Allocator.hpp"
#include <string>
#include <vector>
using namespace std;


int main(int argc, char const *argv[])
{
    vector<string, Allocator<string> > vec(10, "haha");

    vec.push_back("foo");
    vec.push_back("bar");

    //Allocator<Test>::rebind<Test2>::other alloc;

    return 0;
}