【Swift 3.1】23 - 泛型 (Generics)

【Swift 3.1】23 - 泛型 (Generics)

自从苹果2014年发布Swift,到现在已经两年多了,而Swift也来到了3.1版本。去年利用工作之余,共花了两个多月的时间把官方的Swift编程指南看完。现在整理一下笔记,回顾一下以前的知识。有需要的同学可以去看官方文档>>。


泛型代码能让我们编写灵活、可重复使用的方法和类型。可以避免编写重复代码,并以清晰和抽象的方法表达其目的。

泛型是Swift非常强大的一个特性,并且很多Swift标准库都是用泛型编写的。实际上我们已经使用过泛型,例如Swift的ArrayDictionary都是泛型集合。

泛型能解决的问题 (The Problem That Generics Solve)

下面这个方法只能交换两个Int值,使用in-out参数来交换ab

func swapTwoInts(_ a: inout Int, _ b: inout Int) {
    let temporaryA = a
    a = b
    b = temporaryA
}

上面这个方法只能交换Int,如果我想交换两个String类型的值或者两个Double类型的值,我们又得写两个方法swapTwoStrings(_:_:)swapTwoDoubles(_:_:):

func swapTwoStrings(_ a: inout String, _ b: inout String) {
    let temporaryA = a
    a = b
    b = temporaryA
}
 
func swapTwoDoubles(_ a: inout Double, _ b: inout Double) {
    let temporaryA = a
    a = b
    b = temporaryA
}

使用泛型代码让我们用一个方法就能解决上面的问题。

泛型方法 (Generic Functions)

上面的三个方法可以用下面的额泛型方法代替:

func swapTwoValues(_ a: inout T, _ b: inout T) {
    let temporaryA = a
    a = b
    b = temporaryA
}

这个通用的方法使用T作为类型名称,而不是IntString或者DoubleT没有说明必须是什么类型,但是ab的类型必须是一样的。

方法名后面还有一个,告诉Swift只是一个占位类型,不是实际类型。

swapTwoValues(_:_:)方法可以像之前的swapTwoInts一样调用,只要两个用于交换的值类型是一样的。

用通用的方法交换两个String类型的值或者两个Double类型的值:

var someInt = 3
var anotherInt = 107
swapTwoValues(&someInt, &anotherInt)
// someInt is now 107, and anotherInt is now 3
 
var someString = "hello"
var anotherString = "world"
swapTwoValues(&someString, &anotherString)
// someString is now "world", and anotherString is now "hello"

注意:其实Swift的标准库中已经有一个功能与swapTwoValues(_:_:)一样的方法swap。如果我们需要交换两个值,直接使用swap即可,无需自己另外实现这个功能。

类型参数 (Type Parameters)

上面的泛型方法swapTwoValues(_:_:),占位类型T其实是一个类型参数。类型参数指定并命名占位类型,紧跟在方法名后面,用表示。

只要定义好了类型参数,我们就可以用来定义方法的参数类型(例如swapTwoValues(_:_:)的参数ab的类型),或者作为方法的返回类型。

我们可以定义多个类型参数,写法是:

类型参数命名 (Type Parameters)

在很多情况下,类型参数有描述性的名称,例如Dictionary中的KeyValueArray中的Element,这些名字都能告诉读者类型参数和泛型的关系。但是,在没有任何意义的情况下,我们一般把类型参数名命名为TUV

注意:需要使用骆驼命名法给类型参数命名,例如TMyTypeParameter,以表示他们是一个类型的占位。

泛型类型 (Generic Types)

除了泛型方法,Swift还可以定义泛型类型,例如ArrayDictionary

这一部分将演示如何写一个泛型集合Stack。一个栈是一个有序的值的集合,类似一个数组,但是比数组有更严格的运算。 数组可以在特定的位置移除或插入元素。而栈只允许在集合的最后添加元素,也只允许从最后面移除元素。

注意:栈的概念就被用于UINavigationController管理控制器。使用pushViewController(_:animated:)来添加新的控制器到栈中,用popViewControllerAnimated(_:)移除控制器。栈非常适合用来管理“后进先去”的集合。

下图演示了栈的push/pop:

【Swift 3.1】23 - 泛型 (Generics)_第1张图片
Stack
  1. 目前栈中有三个值
  2. 第四个值被push到栈顶
  3. 栈中有四个值
  4. 第四个值被移除(popped)
  5. 移除第四个值后,栈中剩下三个值

下面是一个不通用的IntStack:

struct IntStack {
    var items = [Int]()
    
    mutating fuc push(_ item: Int) {
        items.append(item)
    }
    
    mutating fuc pop() -> Int{
        return items.removeLast()
    }
}

下面是一个通用版本的Stack:

struct Stack {
    var items = [Element]()
    
    mutating func push(_ item: Element) {
        items.append(item)
    }
    
    mutating func pop() -> Element {
        return items.removeLast()
    }
}

Element定义了一个占位名字,并且在三个地方被用到:

  • 创建一个空数组items,其元素类型为Element
  • 作为push(_:)方法的参数类型
  • 作为pop()方法的返回值

因为它是一个通用类型,所以Stack可以用来创建任意有效类型的栈:

var stackOfThings = Stack()
stackOfStrings.push("uno")
stackOfStrings.push("dos")
stackOfStrings.push("tres")
stackOfStrings.push("cuatro")
// the stack now contains 4 strings

下面是stackOfThings的push过程:

【Swift 3.1】23 - 泛型 (Generics)_第2张图片
push

移除栈顶的值"cuatro"

let fromTheTop = stackOfStrings.pop()
// fromTheTop is equal to "cuatro", and the stack now contains 3 strings

移除过程如下:

【Swift 3.1】23 - 泛型 (Generics)_第3张图片
pop

扩展泛型类型 (Extending a Generic Type)

在扩展泛型类型是,我们不必提供类型参数,可以直接使用已有泛型类型的类型参数。

下面是对Stack的扩展:

extension Stack {
    va topItem: Element? {
        return items.isEnpty ? nil : items[items.count - 1]
    }
}

访问topItem属性:

if let topItem = stackOfStrings.topItem {
    print("The top item on the stack is \(topItem).")
}
// Prints "The top item on the stack is tres."

泛型类型的扩展也可以指定一些限定条件,满足这些条件才可以访问扩展里面的成员,下面会讲到。

类型约束 (Type Constraints)

swapTwoValues(_:_:)Stack可以用于任何类型。然而,在某些类型上添加一些类型约束是非常有用的。类型约束指定一个类型参数必须继承于特定的类或者遵循特定的协议。

例如,Swift的Dictionary限定了键的类型。字典的键必须是hashable的,也就是说,必须提供一个方法保证它自己是唯一的,然后字典才能根据这个键去查找对应的值。Swift的基本类型默认都是hashable的,例如StringIntDoubleBool

类型约束语法 (Type Constraint Syntax)

语法如下:

func someFunction(someT: T, someU: U) {
    // function body goes here
}

上面的语法要求T必须是SomeClass的子类,U必须遵循SomeProtocol协议。

类型约束实践 (Type Constraint in Action)

下面是一个不通用的findIndex(ofString:in:),用于查找指定字符串在字符串数组的索引:

func findIndex(ofString valueToFind: String, in array: [String]) -> Int? {
    for (index, value) in array.enumerated() {
        if value == valueToFind {
            return index
        }
    }
    return nil
}

let strings = ["cat", "dog", "llama", "parakeet", "terrapin"]
if let foundIndex = findIndex(ofString: "llama", in: strings) {
    print("The index of llama is \(foundIndex)")
}
// Prints "The index of llama is 2"

我们可以把上面的方法改为通用形式:

func findIndex(of valueToFind: T in array: [T]) -> Int? {
    for (index, value) in array.enumerated() {
        if value == valueToFind {
            return value
        }
    }
    return nil
}

其实这个方法不能编译通过,因为value == valueToFind。不是所有Swift类型的都可使用==进行比较。

Swift定义了Equatable协议,要求所有遵循这个协议的类型必须实现==!=运算符。Swift的标准类型都遵循了Equatable协议。

所以上面的通用方法改写为:

func findIndex(of valueToFind: T, in array: [T]) -> Int? {
    for (index, value) in array.enumerated() {
        if value == valueToFind {
            return index
        }
    }
    return nil
}

T: Equatable类型参数意味着任意类型T必须遵循Equatalbe协议。

例如:

let doubleIndex = findIndex(of: 9.3, in: [3.14159, 0.1, 0.25])
// doubleIndex is an optional Int with no value, because 9.3 is not in the array
let stringIndex = findIndex(of: "Andrea", in: ["Mike", "Malcolm", "Andrea"])
// stringIndex is an optional Int containing a value of 2

关联类型 (Associated Types)

当定义协议时,定义一个或多个关联类型有时候是非常有用的。关联类型提供了一个类型名字,然后这个类型可以在协议中使用。使用associatedtype关键字来定义关联类型。

关联类型实践 (Associated Types in Action)

下面是一个例子:

protocol Container {
    associatedtype Item
    mutating func append(_ item: Item)
    var count: Int { get }
    subscript(i: Int) -> Item { get }
}

这个容器协议中没有具体说明能存储那个类型的值。遵循这个协议的类型必须写清楚能存储的值类型。

之前那个IntStack类型遵循Container协议:

struct Instack: Container {
    // original Instack implementation
    var items = [Int]()
    mutating func push(_ item: Int) {
        items.append(item)
    }
    mutating func pop() -> Int {
        return items.removeLast()
    }
    
    // conformance to the Container protocol
    typealias Item = Int
    mutating func append(_ item: Int) {
        self.push(item)
    }
    var count: Int {
        return items.count
    }
    subscript(i: Int) -> Int {
        return items[i]
    }
}

IntStack指定了Item是一个Int类型。typealias Item = Int把抽象的Item转变为具体的Int

因为Swift有类型推断功能,所以我们可以不用写typealias Item = Int,因为IntStack实现了Container协议的所有要求,Swift能推断出需要用的Item的具体类型。

之前那个Stack类型遵循Container协议:

struct Stack: Container {
    // original Stack implementation
    var items = [Element]()
    mutating func push(_ item: Element) {
        items.append(item)
    }
    mutating func pop() -> Element {
        return items.removeLast()
    }
    // conformance to the Container protocol
    mutating func append(_ item: Element) {
        self.push(item)
    }
    var count: Int {
        return items.count
    }
    subscript(i: Int) -> Element {
        return items[i]
    }
}
通过扩展类型来指定关联类型 (Extending an Existing Type to Specifying an Associated Type)

我们可以使用扩展来遵循某一个协议,当然这个协议也可以是由关联类型的协议。

Swift的Array已经提供了append(_:)方法、count属性和下标,那么就满足了Container协议的要求。所以我们可以是下面这种形式来声明Array遵循Container协议:

extension Array: Container {}

我们就可以把Array当做是一个Container

泛型的Where语句 (Generic Where Clauses)

泛型的where语句可以让我们要求关联类型遵循一个特定的协议,或者关联类型和类型参数必须相同。

例如下面这个例子:

func allItemsMatch (_ someContainer: C1, _ anotherContainer: C2) -> Bool where C1.Item == C2.Item, C1.Item: Equatalbe {
    // Check that both containers contain the same number of items.
    if someContainer.count != anotherContainer.count { return false }
    
    // Check each pair of items to see if they are equivalent.
    for i in 0..

allItemsMatch(_:_:)方法的实际使用:

var stackOfStrings = Stack()
stackOfStrings.push("uno")
stackOfStrings.push("dos")
stackOfStrings.push("tres")
 
var arrayOfStrings = ["uno", "dos", "tres"]
 
if allItemsMatch(stackOfStrings, arrayOfStrings) {
    print("All items match.")
} else {
    print("Not all items match.")
}
// Prints "All items match."

虽然ArrayStack是不同的类型,但是他们都遵循Container协议,并且都包含相同类型的值。

有泛型Where语句的扩展 (Extensions with a Generic Where Clause)

例如下面这个例子:

extension Stack where Element: Equatable {
    func isTop(_ item: Element) -> Bool {
        guard let topItem = items.last else {
            return false
        }
        return topItem == item
    }
}

如果没有泛型where语句,将会出现一个问题:isTop(_:)方法使用了==运算符,但是Stack没有要求它的元素是可以比较的,所以使用==将会编译错误。

下面是isTop(_:)方法的使用:

if stackOfStrings.isTop("tres") {
    print("Top element is tres.")
} else {
    print("Top element is something else.")
}
// Prints "Top element is tres."

下面是使用where扩展Container:

extension Container where Item: Equatalbe {
    func startsWith(_ item: Item) -> Bool {
        return count >= 1 && self[0] == item
    }
}

startsWith(_:)适用于遵循了Container协议的类型:

if [9, 9, 9].startsWith(42) {
    print("Starts with 42.")
} else {
    print("Starts with something else.")
}
// Prints "Starts with something else."

上面对Container的扩展要求Item遵循Equatable协议,我们还可以要求Item是一个具体的类型:

extension Container where Item == Double {
    func average() -> Double {
        var sum = 0.0
        for index in 0..

我们还可以在where语句中可以添加多个约束条件,条件之间用逗号隔开即可。


第二十三部分完。下个部分:【Swift 3.1】24 -访问权限


如果有错误的地方,欢迎指正!谢谢!

你可能感兴趣的:(【Swift 3.1】23 - 泛型 (Generics))