STL源码剖析—Iterator与Traits

Iterator行为类似指针，实现较为简单。这里主要介绍一下Traits。

为什么要用Traits？

使用Traits可以方便我们获取迭代器的相关信息。

获取迭代器的类型

如果函数中想要获取迭代器的类型该怎么做呢？下面提供一种方法：

template <class I, class T>
void func_impl(I iter, T t) {
    T tmp;
    // ...
}
template <class I>
void func(I iter) {
    func_impl(iter, *iter);
}
int main(){
    int i;
    func(&i);
}

通过模板的参数推导，我们得到了迭代器的类型。但是如果函数的返回值用到了该类型怎么办呢？模板是无法推导函数的返回值类型的。

template <class T>
class MyIter {
    typedef T value_type;
    T* ptr;
    MyIter(T* p) : ptr(p) {}
    T& operator*() const { return *ptr; }
};
template <class I>
typename I::value_type func(I ite) {
    return *ite;
}
MyIter<int> ite(new int(1));
cout << func(ite);

我们通过将推导出的类型Ｔ定义为value_type来实现。然而还有一个问题，如果迭代器就是一个指针，不是一个class，无法对其进行typedef，该怎么办呢？

我们设计一个类用来萃取迭代器的类型。

template <class T>
class iterator_traits {
    typedef typename I::value_type value_type;
};
template <class T>
class iterator_traits<T*> {
    typedef T value_type;
};
template <class I>
typename iterator_traits<I>::value_type func(I ite) {
    return *ite;
}

多了iterator_traits这一层，我们可以对其进行偏特化，实现对指针类型的萃取，这样就完美解决了问题。