通过属性值从向量中移除对象的unique_ptr

时间:2015-10-18 20:34:32

标签: c++ c++11 vector smart-pointers unique-ptr

我有一个具有这三个功能的Holder对象

Id

我写过CPPunit测试如下:

unique_ptr<Object> Holder::remove(string objName){
    std::vector<unique_ptr<Object>>::iterator object = 
            find_if(objects.begin(), objects.end(),
            [&](unique_ptr<Object> & obj){ return obj->name() == objName;}
     );
    objects.erase(std::remove(objects.begin(), objects.end(), *object));
    return std::move(*object);
}

vector<unique_ptr<Object>> const& Holder::getContent() const {
    return this->objects;
}

void Holder::add(unique_ptr<Object> objPtr) {
    this->objects.push_back(move(objPtr));
}

这个测试没有问题,但对我来说非常奇怪的是,如果我添加以下行:

void HolderTest::removeObject() {
    Holder holder("bag");
    unique_ptr<Object> ringPtr(new Object("a"));
    holder.add(move(ringPtr));

    unique_ptr<Object> swordPtr(new Object("b"));
    holder.add(move(swordPtr));

    holder.remove("a");
    vector<unique_ptr<Object>> const& objects = holder.getContent();
    CPPUNIT_ASSERT(objects.size() == 1);
}

然后测试崩溃而没有任何消息。我已经在另一个测试中编写了这一行而没有调用remove,它没有任何问题。 如果我将vector的大小值更改为2或0 CPPUNIT_ASSERT(objects.size()== 2); 然后测试失败。所以似乎remove函数保留了unique_ptr之一,但它将它变为nullptr? 任何iea有什么问题?

1 个答案:

答案 0 :(得分:3)

    std::vector<unique_ptr<Object>>::iterator object = 
        find_if(objects.begin(), objects.end(),
                [&](unique_ptr<Object> & obj){ return obj->name() == objName;}
               );
    objects.erase(std::remove(objects.begin(), objects.end(), *object));
    return std::move(*object);

在失效后,你会对迭代器object 进行反驳。见Iterator invalidation rules

在擦除之前移动指针,然后就可以了。

其他说明:

  • removing与值一起使用很有趣(而不是只删除你得到的迭代器)。你期望向量包含重复吗?实际上,请注意:这会导致erase错误,因为它总是会删除一个元素
  • 在解除引用之前,您也不会检查object可能是end()迭代器。 Undefined Behaviour
  • 的另一个来源
  • 考虑将name const&提高效率

<强> Live On Coliru

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

using namespace std;

struct Object {
    Object(std::string name) : _name(std::move(name)) { }

    std::string const& name() const { return _name; }
  private:
    std::string _name;
};

struct Holder {
    using Ptr = unique_ptr<Object>;

    Ptr remove(string const& objName) {

        auto it = find_if(objects.begin(), objects.end(), [&](Ptr& obj){ return obj->name() == objName; });

        if (it != objects.end()) {
            auto retval = std::move(*it);
            objects.erase(it);
            return std::move(retval);
        }

        return {}; // or handle as error?
    }

    vector<Ptr> const& getContent() const {
        return this->objects;
    }

    void add(Ptr objPtr) {
        this->objects.push_back(move(objPtr));
    }

  private:
    vector<Ptr> objects;
};

int main() {

    Holder h;
    for(auto n: { "aap", "noot", "mies", "broer", "zus", "jet" })
        h.add(std::make_unique<Object>(n));

    h.remove("broer");
    h.remove("zus");

    for (auto& o : h.getContent())
        std::cout << o->name() << "\n";
}

打印

aap
noot
mies
jet