如何传递 unique_ptr<T> 代替原始 *output* 指针参数？

Question

我在 external 库中有一个 预先存在的 函数，它看起来像这样；

bool CreateTheThing(MyThing *& pOut);

简而言之；我给它一个原始指针（通过引用），该函数分配内存并将我的指针分配给新分配的对象。当函数返回时，我的有责任在我完成后释放内存。

显然，我想将此结果存储到unique_ptr<MyThing>，并避免手动使用delete。

我可以创建一个临时原始指针以用于 API 调用，并将其传递给 unique_ptr 的构造函数；

MyThing* tempPtr;
CreateTheThing(tempPtr);
unique_ptr<MyThing> realPtr = unique_ptr<MyThing>(tempPtr);

还有比这更直接的方法吗？一个不需要临时原始指针的？理想情况下，会有unique_ptr 的方法以一种可以直接与CreateTheThing 方法一起使用的方式公开其内部指针？

unique_ptr<T>::get() 不允许这样做，据我所知。它返回的指针不是对内部使用的指针的引用。

Answer 1

您可以通过一次编写多行代码来节省一行代码（可能多次）：

class Wrapper
{
  std::unique_ptr<MyThing> &u;
  MyThing *p;

public:
  Wrapper(std::unique_ptr<MyThing> &u) : u(u), p() {}

  operator MyThing* & ()
  { return p; }

  ~Wrapper()
  { u.reset(p); }
};

用法：

std::unique_ptr<MyThing> u;
CreateTheThing(Wrapper(u));

Answer 2

还有比这更直接的方法吗？一个不需要临时原始指针的？

不，没有。

理想情况下，会有一个 unique_ptr 的方法，它以一种可以直接与 CreateTheThing 方法一起使用的方式公开其内部指针？据我所知，unique_ptr::get() 不允许这样做。

你的知识是正确的。这将破坏std::unique_ptr 的全部目的，因此unique_ptr::get() 是一个const 函数，并且指针按值返回。

尽管与您的构造函数示例类似，您始终可以使用std::unique_ptr::reset() 传递外部分配的指针。

另请注意：如果第 3 方 API 要求您使用 free() 释放内存，您可能需要为 std::unique_ptr 提供特殊的删除函数。

Answer 3

如果你经常使用这个函数，你可以把转换放在一个函数中。

最好是更改 API，但这也可以。

inline std::unique_ptr<MyThing> CreateTheThing()
{
  MyThing* p;
  if (CreateTheThing(p))
  {
    return std::unique_ptr<MyThing>(p);
  }
  return std::unique_ptr<MyThing>();
}

你也可以使这个重载更容易重构：

inline bool CreateTheThing(std::unique_ptr<MyThing>& ptr)
{
    try 
    {
        MyThing* p;
        if (CreateTheThing(p))
            ptr = std::unique_ptr<MyThing>(p);
        else
            return false;
    }
    catch (std::bad_alloc&)
    {
        return false;
    }
    return true;
}