用于扩展任意标准符合分配器的C ++设计模式

时间:2011-06-13 18:26:43

标签: c++ design-patterns stl allocator

我目前正在寻找扩展任意标准符合分配器类型的最佳方法。要明确:我不想编写自定义分配器。我只想“添加”特定的扩展或行为到现有的扩展或行为。我已经创建了一个样本的样子。请注意,以下代码仅用于说明目的。

#ifndef HPP_SMART_ALLOCATOR_INCLUDED
#define HPP_SMART_ALLOCATOR_INCLUDED


#include <memory>
#include <map>


template<typename T>
struct allocator_traits;

template<typename T, class allocator_type = std::allocator<T>>
class smart_allocator;


template<>
struct allocator_traits<void>
{
    typedef std::allocator<void>::const_pointer const_pointer;
    typedef std::allocator<void>::pointer       pointer;
    typedef std::allocator<void>::value_type    value_type;
};

template<typename T>
struct allocator_traits
{
    typedef typename std::allocator<T>::const_pointer   const_pointer;
    typedef typename std::allocator<T>::const_reference const_reference;
    typedef typename std::allocator<T>::difference_type difference_type;
    typedef typename std::allocator<T>::pointer         pointer;
    typedef typename std::allocator<T>::reference       reference;
    typedef typename std::allocator<T>::size_type       size_type;
    typedef typename std::allocator<T>::value_type      value_type;
};


template<class allocator_type>
class smart_allocator<void, allocator_type>
    : public allocator_traits<void>
{
public:
    template<typename U> struct rebind { typedef smart_allocator<U, typename allocator_type::rebind<U>::other> other; };
};

template<typename T, class allocator_type>
class smart_allocator
    : public  allocator_traits<T>,
      private allocator_type
{
public:
    using typename allocator_traits<T>::const_pointer;
    using typename allocator_traits<T>::const_reference;
    using typename allocator_traits<T>::difference_type;
    using typename allocator_traits<T>::pointer;
    using typename allocator_traits<T>::reference;
    using typename allocator_traits<T>::size_type;
    using typename allocator_traits<T>::value_type;
    template<typename U> struct rebind { typedef smart_allocator<U, typename allocator_type::rebind<U>::other> other; };

    smart_allocator() throw() /*noexcept*/;
    smart_allocator(allocator_type const&) throw() /*noexcept*/;
    virtual ~smart_allocator() throw();

    virtual ~smart_allocator()
    {
        std::map<pointer, size_type>::iterator i = this->m_map.begin();
        while (i != this->m_map.end())
        {
            this->allocator_type::deallocate(i->first, i->second);
            ++i;
        }
    }

    pointer allocate(size_type n, allocator_traits<void>::const_pointer hint = 0)
    {
        pointer p = this->allocator_type::allocate(n, hint);
        this->m_map.insert(std::pair<pointer, size_type>(p, n));
        return p;
    }

    void deallocate(pointer p, size_type n) /*noexcept*/
    {
        std::map<pointer, size_type>::iterator iter = this->m_map.find(p);
        if (iter != this->m_map.end())
            this->allocator_type::deallocate(iter->first, iter->second);
    }

    using allocator_type::address;
    using allocator_type::construct;
    using allocator_type::destroy;
    using allocator_type::max_size;

private:
    smart_allocator(smart_allocator const&) throw();
    smart_allocator& operator=(smart_allocator const&);

    std::map<pointer, size_type> m_map;
};


#endif /* HPP_SMART_ALLOCATOR_INCLUDED */

请考虑以下注意事项:

  • 模板参数allocator_type可以是任何标准符合类型。它不限于std :: allocator。这与所有STL实现使用的技术相同。
  • 我们需要在从allocator_type派生时使用私有继承,因为std :: allocator成员函数不是虚拟的。但是,std :: allocator&amp; alloc = smart_allocator()不会做你所期望的。

你认为这是适用的吗?

2 个答案:

答案 0 :(得分:1)

你当然需要实现一个复制构造函数和复制赋值运算符,否则当容器按值传递分配器时,你的映射可能会被破坏(特别是你最终可能会被双重删除)。我可能还有其他一些注意事项。

答案 1 :(得分:1)

立即想到的是Decorator;作为参考说明,“装饰器可以在不重写原始对象代码的情况下使对象适应新情况。”如果我理解你的问题,那听起来就像你追求的那样。