有没有一种简单的方法来重构这段代码？

Question

我有一个具有非常相似的重复代码的函数。我喜欢重构它，但不想要任何复杂的映射代码。

该代码基本上过滤掉表中的列。我通过让比较语句具有简单类型使这个示例变得简单，但真正的比较可能更复杂。

我希望可能有一些模板或 lambda 技术可以做到这一点。

vector<MyRecord*>& MyDb::Find(bool* field1, std::string * field2, int* field3)
{
    std::vector<MyRecord*>::iterator iter;
    filterList_.clear();
    std::copy(list_.begin(), list_.end(), back_inserter(filterList_));

    if (field1)
    {
        iter = filterList_.begin();
        while (iter != filterList_.end())
        {
            MyRecord* rec = *iter;
            if (rec->field1 != *field1)
            {
                filterList_.erase(iter);
                continue;
            }
            iter++;
        }
    }

    if (field2)
    {
        iter = filterList_.begin();
        while (iter != filterList_.end())
        {
            MyRecord* rec = *iter;

            if (rec->field2 != *field2)
            {
                filterList_.erase(iter);
                continue;
            }
            iter++;
        }
    }

    if (field3)
    {
        iter = filterList_.begin();
        while (iter != filterList_.end())
        {
            MyRecord* rec = *iter;

            if (rec->field3 != *field3)
            {
                filterList_.erase(iter);
                continue;
            }
            iter++;
        }
    }
    return filterList_;
}

---

更新: 以防万一有人好奇，这是我的最终代码。再次感谢大家。很容易理解和维护。

vector<MyRecord*>& MyDb::Find(bool* field1, std::string* field2, int* field3)
{
    auto compare = [&](MyRecord* rec) {
        bool add = true;
        if (field1 && rec->field1 != *field1) {
            add = false;
        }
        if (field2 && rec->field2 != *field2) {
            add = false;
        }
        if (field3 && rec->field3 != *field3) {
            add = false;
        }
        return add;
    };

    filterList_.clear();

    std::copy_if(list_.begin(), list_.end(), back_inserter(filterList_), compare);
    return filterList_;
}

Answer 1

您可以使用std::copy_if（因为您已经/将要复制一份）

vector<MyRecord*>& MyDb::Find(bool* field1, std::string* field2, int* field3){
  filterList_.clear();
  std::copy_if(list_.begin(), list_.end(), back_inserter(filterList_),[&](MyRecord* rec){
    // whatever condition you want.
    return field3 && rec->field3 != *field3;
  });
  return filterList_;
}

Answer 2

有没有一种简单的方法来重构这段代码？

据我了解您的算法/意图，使用 std::erase_if (c++20) 您可以替换整个 while 循环，如下所示 (Demo code)：

#include <vector> // std::erase_if

std::vector<MyRecord*> // return by copy as filterList_ is local to function scope
Find(bool* field1 = nullptr, std::string* field2 = nullptr, int* field3 = nullptr)
{
    std::vector<MyRecord*> filterList_{ list_ }; // copy of original
    const auto erased = std::erase_if(filterList_, [=](MyRecord* record) { 
        return record 
            && ((field1 && record->field1 != *field1)
            || (field2 && record->field2 != *field2)
            || (field3 && record->field3 != *field3));
        }
    );
    return filterList_;
}

---

如果不支持 C++20，或者您可以使用 erase–remove idiom，这实际上是在 std::erase_if 的底层发生的。