c++ - 有效地从 unordered_set 中删除 unique_ptr

Question

我使用 unique_ptr 将某些对象的所有权存储在 unordered_set 中。但当时机成熟时，我不知道如何从集合中删除其中一个。

代码看起来像这样:

typedef unique_ptr<MyType> MyPtr;

unordered_set<MyPtr> owner;

MyPtr p = make_unique<MyType>("foo")
MyType *pRaw = p.get();
owner.insert(std::move(p));

// Later ...

// I want to do something like this (cannot be written as-is, of course):
// owner.erase(pRaw);

有办法做到这一点吗？当然，我可以使用 begin() 和 end() 迭代整个集合，但将它们放入集合中的全部目的是使这些查找高效。

我已经想到的一些事情:

• 使用shared_ptr。对于我的案例来说，这是错误的抽象。所有权是独一无二的。
• 使用原始指针，忘记 unique_ptr。这放弃了 unique_ptr 提供的所有优点。
• 使用 unordered_set::begin(key) 查找存储桶。据我所知，我无法创建与我要删除的 unique_ptr 相匹配的 key 。但我很高兴被证明是错误的(:

(事实上，我使用 eaSTL::unordered_set 及其 find_as 自定义键函数解决了这个问题)

Answer 1

这是一个棘手的案例。 erase有一个重载，需要 const key_type&参数，所以我们可以尝试创建一个“陈旧”unique_ptr获取要删除的元素的哈希值:

template <typename T>
auto erase(std::unordered_set<std::unique_ptr<T>>& set, T* ptr)
{
    std::unique_ptr<T> stale_ptr{ptr};
    auto ret = set.erase(stale_ptr);
    stale_ptr.release();
    return ret;
}

( live demo )

<小时/>

但是，此版本通常不是异常安全的，因为 release如果 set.erase 则不会被调用抛出异常。在这种情况下这不是问题，因为 std::equal_to<std::unique_ptr<T>>::operator()从不抛出异常。一般情况下，我们可以滥用unique_ptr (!) 通过确保 release 来强制异常安全无论函数是正常退出还是异常退出都会被调用:

template <typename T>
auto erase(std::unordered_set<std::unique_ptr<T>>& set, T* ptr)
{
    std::unique_ptr<T> stale_ptr{ptr};

    auto release = [](std::unique_ptr<T>* p) { p->release(); };
    std::unique_ptr<std::unique_ptr<T>, decltype(release)> release_helper{&stale_ptr, release};

    return set.erase(stale_ptr);
}

( live demo )