静态断言检查地图、set、unordered_map、unordered_set

Question

我正在编写一个函数，将map、set、unordered_map、unordered_set的键复制到vector，现在我想添加一个编译时断言，以获得明确的错误，如果有人试图在该函数中传递vector，list。

template <typename container>
auto CopyKeyToVector(conatiner c)
{
  //static assert to check c is map, unordered map only?


}

你知道我们该怎么做吗?就像map一样，unordered_map本身就是经过临时处理的容器。

Answer 1

您可以实现并使用is_specialization_of作为通用解决方案：

template <typename, template <typename...> class>
struct is_specialization_of : std::false_type
{
};

template <template <typename...> class TTemplate, typename... Ts>
struct is_specialization_of<TTemplate<Ts...>, TTemplate> : std::true_type
{
};

template <typename container>
auto CopyKeyToVector(container c)
{
    static_assert(
        is_specialization_of<container, std::map>{} || 
        is_specialization_of<container, std::unordered_map>{}, "");
}

Full example on wandbox。

注意，它也可以与std::enable_if一起使用。

template <typename T>
constexpr auto is_map_or_umap = 
    is_specialization_of<T, std::map>{} || 
    is_specialization_of<T, std::unordered_map>{};

template <typename container>
auto CopyKeyToVector(container) -> std::enable_if_t<is_map_or_umap<container>>
{
    // (1)
}

template <typename container>
auto CopyKeyToVector(container) -> std::enable_if_t<!is_map_or_umap<container>>
{
    // (2)
}

int main()
{
    CopyKeyToVector(std::map<int,float>{});           // Calls (1) 
    CopyKeyToVector(std::unordered_map<int,float>{}); // Calls (1)
    CopyKeyToVector(std::vector<int>{});              // Calls (2)  
}

Full example on wandbox。

Answer 2

不要使用静态断言。相反，添加一个间接级别：

template <typename C>
void CopyKeyToVectorImpl(const C& c) { /* ... */ }

template <typename K, typename T, typename Pred, typename Alloc>
void CopyKeyToVector(const std::map<K, T, Pred, Alloc>& m) {
  CopyKeyToVectorImpl(m);
}

template <typename K, typename T, typename Hash, typename Eq, typename Alloc>
void CopyKeyToVector(const std::unordered_map<K, T, Hash, Eq, Alloc>& m) {
  CopyKeyToVectorImpl(m);
}

(您可以进一步修改Impl函数，通过参数化值到键的投影，使集合在相同的洗涤中。)