Question

我有一个使用模板的C ++方法findID，我希望能够根据输入类型在此方法中运行条件。模板参数U可以是int类型，也可以是string类型。我想根据ID的类型运行不同的条件。

我的代码如下：

template <typename S>
template <typename U>
S * findID(U ID){
    for (typename vector<S*>::collectionsIter element = collection.begin() ; element != collection.end(); ++element)
        if((*element)->getID() == ID) return *element;
    return NULL;
}

我希望我的代码执行以下操作：

template <typename S>
    template <typename U>
    S * findID(U ID){

      ***if ID is an int:
        for (typename vector<S*>::collectionsIter element = collection.begin() ; element != collection.end(); ++element)
            if((*element)->getID() == ID) return *element;


      ***if ID is a string:

         for (typename vector<S*>::collectionsIter element = collection.begin() ; element != collection.end(); ++element)
            if((*element)->getStringID() == ID) return *element;

       ***else 

        return NULL;


    }

我想这样做的原因是因为我希望能够将ID的字符串变量与getStringID（）的字符串方法进行比较，并将ID的int变量与getID（）的int方法进行比较。另外我不想将它们分解为单独的方法，因此我尝试使用模板和这些条件将其重构为1方法。

Answer 1

只需使用2次重载：

template <typename S>
S* findID(int ID){
    for (auto* element : collection)
        if (element->getID() == ID) return element;
    return nullptr;
}

template <typename S>
S* findID(const std::string& ID){
    for (auto* element : collection)
        if (element->getStringID() == ID) return element;
    return nullptr;
}

Answer 2

以下是使用C ++ 17 if constexpr：

进行此操作的一种方法

struct Foo {
    int id;
    std::string stringId;

    int getId() const { return id; }
    const std::string& getStringId() const { return stringId; }
};

template <typename Cont, typename T>
auto findId(const Cont& c, const T& id) {
    const auto pred = [&id](const auto& x) {
        if constexpr (std::is_convertible_v<T, std::string>)
            return x.getStringId() == id;
        else if constexpr (std::is_convertible_v<T, int>)
            return x.getId() == id;
        else
            static_assert(false, "Unsupported id type.");
        return false;
    };
    const auto findIt = std::find_if(begin(c), end(c), pred);
    return findIt == end(c) ? nullptr : &(*findIt);
}

int main() {
    using namespace std;
    vector<Foo> foos{{1, "1"}, {2, "2"}, {3, "3"}};
    auto foo2Int = findId(foos, 2);
    auto foo2String = findId(foos, "2"s);
    cout << foo2Int->id << ", " << foo2String->stringId << '\n';
}

C ++条件语句，用于测试变量是否为字符串

2 个答案: