时间戳

首页 > 解决方案 > 使用模板函数的默认参数

c++ - 使用模板函数的默认参数

问题描述

我有两个看起来与我非常相似的功能：

template <typename Predicate>
vector<int> FindTopItems(const string& query,
                         const Predicate& predicate) const;
                         
vector<int> FindTopItems(const string& query,
                         ItemStatus status = ItemStatus::Initial) const;

是否可以仅使用此功能的模板版本？如何将它的谓词设置为默认值，由枚举类设置？

所以我希望以下代码是可编译的：

int main() {

   FindTopItems(""s, [](){}); /* predicate version */
   FindTopItems(""s); /* default parameter as enum class should be used */
   
   return 0;
}

标签： c++

解决方案

您可以将 predicate 的默认值设置为ItemStatus，然后创建一个返回默认值的辅助类。

#include <iostream>
#include <vector>
#include <string>

enum class ItemStatus {
    Initial
};

template <typename T>
auto defaultValue() {
    return T{};
}

template <>
auto defaultValue<ItemStatus>() {
    return ItemStatus::Initial;
}

template <typename Predicate = ItemStatus>
std::vector<int> FindTopItems(const std::string& query, const Predicate& predicate = defaultValue<Predicate>()) {
    if constexpr (std::is_same_v<Predicate, ItemStatus>) {
        std::cout << "ItemStatus " << static_cast<int>(predicate) << '\n';
    } else {
        std::cout << "Predicate " << predicate() << '\n';
    }
    return {};
}

int main() {
   FindTopItems(std::string{}, [](){ return 1; }); /* predicate version */
   FindTopItems(std::string{}); /* default parameter as enum class should be used */
   
   return 0;
}

推荐阅读