C ++中基于类型的模板函数

时间:2015-08-14 10:13:48

标签: c++ templates specialization differentiation

我想编写一个故障安全访问std::map的函数。

在我的代码中的许多地方,我想按密钥访问std::map,但是如果密钥不存在,我希望有一种默认值而不是异常(这是很多代码为"没有")。

我写了这个基于模板的功能

template <typename T1, typename T2>
T2 mapGetByKey(std::map<T1, T2>& map, T1 key, T2 defaultValue={})
{
    auto it = map.find(key);

    if (it != map.end())
    {
        return it->second;
    }

    return defaultValue;
};

效果很好。但是对于std::map<int, const char*>我想要有不同的行为。所以我可以添加这个专业化:

template <typename T1>
const char* mapGetByKey(std::map<T1, const char*>& map, T1 key, const char* defaultValue="")
{
    auto it = map.find(key);

    if (it != map.end())
    {
        return it->second;
    }

    return defaultValue;
};

也有效。但我认为这只是一个案例的代码。

是否有人知道如何保存行而不将""调用的defaultValue设置为std::map<int, const char*>

有没有办法在编译时区分类型,可能有一些ifdef或类似的东西?

3 个答案:

答案 0 :(得分:4)

选项#1

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

template <>
const char* defaultValue<const char*>()
{
    return "default string";
}

template <typename T1, typename T2>
T2 mapGetByKey(std::map<T1, T2>& map, const T1& key)
{
    auto it = map.find(key);

    if (it != map.end())
    {
        return it->second;
    }

    return defaultValue<T2>();
}

DEMO 1

选项#2

template <typename T> struct identity { using type = T; };

template <typename T1, typename T2>
T2 mapGetByKey(std::map<T1, T2>& map, T1 key, const typename identity<T2>::type& defaultValue = {})
{
    auto it = map.find(key);

    if (it != map.end())
    {
        return it->second;
    }

    return defaultValue;
}

template <typename T1>
const char* mapGetByKey(std::map<T1, const char*>& map, const T1& key)
{
    return mapGetByKey(map, key, "default string");
}

DEMO 2

答案 1 :(得分:1)

非常感谢你!我没有看到这种简单但非常有效的可能性。

结合这两种方法,同时添加基于情境的不同defaultValue的功能,我最终做到了这一点:

template <typename T>
constexpr T mapGetByKeyDefaultValue()
{
    return {};
};

template <>
constexpr const char* mapGetByKeyDefaultValue<const char*>()
{
    return "";
};

template <typename T1, typename T2>
T2 mapGetByKey(std::map<T1, T2>& map, T1 key, T2 defaultValue=mapGetByKeyDefaultValue<T2>())
{
    auto it = map.find(key);

    if (it != map.end())
    {
        return it->second;
    };

    return defaultValue;
};

答案 2 :(得分:0)

您可以为默认值添加模板函数,并为特定类型添加其专用版本:

template <typename T>
T default_value() { return {}; }
template <>
const char* default_value<const char*>() { return ""; }

然后

template <typename T1, typename T2>
T2 mapGetByKey(std::map<T1, T2>& map, T1 key, T2 defaultValue=default_value<T2>())

LIVE