Swift 对于泛型的支持使用我们可以避免为类似的功能多次书写重复的代码,这是一种很好的简化。而对于泛型类型,我们也可以使用extension 为泛型类型添加新的方法。
与为普通的类型添加扩展不同的是,泛型类型在类型定义时候就引入了类型标志,我们可以直接使用。例如Swift 的Array 类型的定义是:
@frozen public struct Array<Element> {
}
在这个定义中,已经声明了Element 为可变类型
。在为类似这样的泛型类型写扩展的时候,我们不需要
在extension
关键字后的声明中重复
去写<Element>
这样的泛型类型名字(其实编译器也不允许我们这么做),在扩展中可以使用和原来定义一样的Element
来指代类型本体声明的泛型。比如我们想在扩展中实现一个random 方法
来随机的取出Array 中的元素:
extension Array {
var random: Element? {
return self.count != 0 ? self[Int(arc4random_uniform(UInt32(self.count)))] : nil
}
}
let languages = ["Swift", "Objc", "C++", "Java"]
languages.random!
// 随机输出这四个字符串中的某个
let ranks = [1, 2, 3, 4]
ranks.random!
// 随机输出这四个数字中的某个
在扩展中是也可以添加整个类型可用新泛型符号的,我们可以添加Element 以外的其他泛型符号。比如在刚才的扩展中加上:
func appendRandomDescription<U>(input: U) -> String {
if let element = self.random {
return "\(element) \(input)"
} else {
return "empty array"
}
}