查找 unique_ptr 集合的原始指针

我经常发现自己想要编写这样的代码：

class MyClass
{
public:
  void addObject(std::unique_ptr<Object>&& newObject);

  void removeObject(const Object* target);

private:
  std::set<std::unique_ptr<Object>> objects;
};

然而，大部分 std::set 接口对于 std::unique_ptrs 来说毫无用处，因为查找函数需要 std::unique_ptr 参数（我显然没有这些参数，因为它们由集合本身拥有）。

我可以想到两个主要的解决方案。

1. 创建一个临时的 unique_ptr 用于查找。例如，上面的removeObject()可以这样实现：

   void MyClass::removeObject(const Object* target)
   {
     std::unique_ptr<Object> targetSmartPtr(target);
     objects.erase(targetSmartPtr);
     targetSmartPtr.release();
   }
   

2. 用指向 unique_ptr 的原始指针映射替换该集合。

     // ...
     std::map<const Object*, std::unique_ptr<Object>> objects;
   };
   

然而，两者对我来说似乎都有点愚蠢。在解决方案 1 中，erase() 不是 noexcept，因此临时的 unique_ptr 可能会删除它并不真正拥有的对象，而解决方案 2 则不必要地需要双倍的容器存储空间。

我了解 Boost 的指针容器，但与现代 C++11 标准库容器相比，它们当前的功能有限。

我最近正在阅读有关 C++14 的内容，并发现“向关联容器添加异构比较查找”。但根据我的理解，查找类型必须与关键类型相当，但原始指针不能与 unique_ptrs 比较。

有人知道更优雅的解决方案或即将推出的 C++ 解决方案来解决这个问题吗？

In C++14 http://en.cppreference.com/w/cpp/container/set/find, std::set<Key>::find is a template函数如果Compare::is_transparent存在。您传入的类型不需要是Key，相当于您的比较器下的值。

所以写一个比较器：

template<class T>
struct pointer_comp {
  typedef std::true_type is_transparent;
  // helper does some magic in order to reduce the number of
  // pairs of types we need to know how to compare: it turns
  // everything into a pointer, and then uses `std::less<T*>`
  // to do the comparison:
  struct helper {
    T* ptr;
    helper():ptr(nullptr) {}
    helper(helper const&) = default;
    helper(T* p):ptr(p) {}
    template<class U, class...Ts>
    helper( std::shared_ptr<U,Ts...> const& sp ):ptr(sp.get()) {}
    template<class U, class...Ts>
    helper( std::unique_ptr<U, Ts...> const& up ):ptr(up.get()) {}
    // && optional: enforces rvalue use only
    bool operator<( helper o ) const {
      return std::less<T*>()( ptr, o.ptr );
    }
  };
  // without helper, we would need 2^n different overloads, where
  // n is the number of types we want to support (so, 8 with
  // raw pointers, unique pointers, and shared pointers).  That
  // seems silly:
  // && helps enforce rvalue use only
  bool operator()( helper const&& lhs, helper const&& rhs ) const {
    return lhs < rhs;
  }
};

然后使用它：

typedef std::set< std::unique_ptr<Foo>, pointer_comp<Foo> > owning_foo_set;

now, owning_foo_set::find会接受unique_ptr<Foo> or Foo* or shared_ptr<Foo>（或任何派生类Foo）并找到正确的元素。

在 C++14 之外，您被迫使用map to unique_ptr方法，或等效的东西，作为签名find限制过于严格。或者自己写set相等的。