Ошибка владения дублированными элементами unique_ptr в векторе - PullRequest
Новая
       59

Ошибка владения дублированными элементами unique_ptr в векторе

0 голосов
/ 22 апреля 2019

Учитывая следующий код, как я могу решить проблему владения, с которой я сталкиваюсь?

Мне нужно разобраться в этой ситуации, не делая копию Item, потому что я не могу.

Item и ItemContainer являются частью API, который пользователь не должен беспокоиться о дублировании, это означает, что мне нужно дважды добавить указатель, если пользователь дважды вызывает его для одного и того же указателя.

Я думал, что мой удалитель разрешит это, но это не так.

Например, я не могу использовать unordered_set, я должен справиться с этой ситуацией, как сейчас.

ItemContainer.h: 

typedef std::unique_ptr<Item, std::function<void(Item *)>> ItemDeleter;
std::vector<ItemDeleter> items_;

ItemContainer.cpp: 

void ItemContainer::addItem(Item *item)
{
  ItemDeleter uniqPtrItem(item, [](Item *p) {
    if (p != nullptr) { 
      delete p; //it's crashing here, obviously
      p = nullptr;
    }
  });
  items_.push_back(std::move(uniqPtrTask));
}

main.cpp 

int main() {
  Item *item = new Item();
  ItemContainer itemContainer;
  itemContainer.addItem(item);
  itemContainer.addItem(item);
}

Ответы [ 2 ]

1 голос
/ 23 апреля 2019

Если вы не можете использовать std::shared_ptr (что является очевидным и наиболее естественным решением), тогда вы можете сосчитать число ваших Item с. Вот простой демонстрационный код, показывающий, как это сделать: 

#include <vector>
#include <memory>
#include <iostream>

class RefCount
{
public:
    void Retain ()
    {
        ++refcount;
    }

    void Release ()
    {
        if (--refcount == 0)
            delete this;
    }

protected:
    virtual ~RefCount () {}

private:    
    int refcount = 1;
};

class Item : public RefCount
{
public:
    Item () { std::cout << "Item constructor\n"; }
private:
    ~Item () { std::cout << "Item destructor\n"; }
};

typedef std::unique_ptr <Item, void (*) (Item *)> ItemDeleter;

void addItem (std::vector <ItemDeleter> &items, Item *item)
{
    item->Retain ();
    ItemDeleter uniqPtrItem (item, [] (Item *p) { p->Release (); });
    items.push_back (std::move (uniqPtrItem));
}

int main()
{
    std::vector <ItemDeleter> items;
    Item *item = new Item;
    addItem (items, item);
    addItem (items, item);
    item->Release ();
}

Выход: 

Item constructor
Item destructor

Живая демоверсия

0 голосов
/ 23 апреля 2019

Учитывая ограничения, налагаемые вашей ситуацией, я бы так решил.

Если они передают необработанные указатели в контейнер, использование shared_ptr не будет работать, потому что вам все равно придется проверить, содержался ли указатель где-либо еще в контейнере.То же самое относится к unique_ptr - вы должны проверить, не появляется ли оно где-нибудь еще в контейнере.

Обойти это невозможно: нужно проверить, есть ли в контейнере предмет.При этом мы можем сделать это эффективно, используя unordered_map.Карта будет отслеживать, сколько раз каждый элемент появляется в контейнере, и когда элемент больше не появляется в контейнере, его можно безопасно удалить.

template<class Item>
class ItemContainer {
    std::unordered_map<Item*, int> item_counts;
    std::vector<Item*> items; 
   public:
    ItemContainer() = default;
    ItemContainer(ItemContainer&&) = default;
    ItemContainer(ItemContainer const&) = default; 

    void addItem(Item* item) {
        item_counts[item] += 1;
        items.push_back(item); 
    }
    void removeTopItem() {
        // Get the top item and remove it from the vector
        auto item = items.back(); 
        items.pop_back(); 
        // Find the number of times the item appears in the vector
        auto iter = item_counts.find(item);
        auto& count = iter->second;

        if(count == 1) {
            // If it appeared only once, erase it from the map and delete the item
            item_counts.erase(iter); 
            delete item; 
        } else {
            // Otherwise, just update the count
            count -= 1;
        }
    }
    ~ItemContainer() {
        for(auto& count : item_counts) {
            auto item = count.first;
            delete item; 
        }
    }
};
...