Question

我需要编写一个函数来检索和处理一些数据。这些数据可以通过多种方式分配（在数据段，堆上，共享内存段等）：

T *data;
if( global ) data = &d;
if( heap )   data = new T [ size ];
if( shm )    data = (T*) shmat( id, 0, 0 );
// processing data ...

由于data可以动态分配，我认为处理它的最佳方法是使用unique_ptr或其他类型的智能指针。然而，并不总是动态分配：我需要在运行时选择unique_ptr的删除器，但这是不可能的。

我应该如何定义和处理data？

Answer 1

您可以让自定义删除器获取运行时值！

struct MyCustomDeleter
{
   MemoryType type;
   template <typename T>
   void operator()(T* value) const
   {
      switch (type)
      {
         case MemoryType::heap:
             delete[] value;
             break;
         case MemoryType::shm:
             unmap_from_shm(value);
             break;
         // etc.
      }
   }
};

...

std::unique_ptr<T, MyCustomDeleter> ptr (new T[size], 
                                         MyCustomDeleter{MemoryType::heap});

Answer 2

我不确定std::unique_ptr，但您可以使用std::shared_ptr。它的自定义删除器不依赖于类模板参数。

Answer 3

除了KennyTm的好答案之外，另一种可能性是使用函数指针作为删除器，然后在运行时提供不同的函数：

typedef std::unique_ptr<T, void(*)(void*)> Ptr;
void delete_global(void*);
void delete_heap(void*);
// etc.

Ptr get(/* whatever */)
{
    if ( ... )
       return Ptr(data, delete_global);
    if (... )
       return Ptr(data, delete_heap);

    // etc.
}

Answer 4

将自己的智能指针与您选择的删除器一起使用：

enum DataPointerType
{
    Stack,
    Dynamic,
    SharedMem,
    ...
};

template <class T>
class DataPointer
{
public:
    DataPointer(T* pointer, DataPointerType type)
        : _ptr(pointer), _type(type)
    { }

    ~DataPointer()
    {
        switch (type) {
        case Stack: break;
        case Dynamic: delete _ptr; _ptr = nullptr; break;
        ...
        }
    }

    T& operator*() { return *_ptr; }
    const T& operator*() const { return *ptr; }

    T* operator->() { return _ptr; }
    const T* operator->() const { return ptr; }

private:
   T* _ptr;
    DataPointerType _type;

    // Prevent copying and destruction, dangerous
    DataPointer(const DataPointer& other);
    DataPointer& operator=(const DataPointer& other);
};

Answer 5

如果您使用共享shared_ptr<>，则可以在运行时选择删除器。只要您不复制/ ... shared_ptr它的行为应与unique_ptr相同。

C ++：带有未知删除器的unique_ptr

5 个答案: