盘点 swift 中 where 关键字的所有用法，你知道多少？

杂雾无尘

博观而约取，厚积而薄发

关注我，每天分享一个关于 iOS 的新知识

前言

where 是 Swift 中一个强大的关键字，可以轻松过滤掉一些值。它可以用于许多不同的表达式中，今天就来盘点一下。

在 switch 中使用

在 switch 语句中的 case 中，可以用 where 关键字做二次筛选，举个例子：

let point = (x: 3, y: 4)

switch point {
case let (x, y) where x == y:
    print("(\(x), \(y)) is on the line x == y")
case let (x, y) where x == -y:
    print("(\(x), \(y)) is on the line x == -y")
case let (x, y):
    print("(\(x), \(y)) is just some arbitrary point")
}

在这个示例中，我们有一个点的坐标点 (x, y)。在 switch 的 case 中，我们使用了 where 子句来判断点是否在 x == y 这条直线上，或是在 x == -y 这条直线上。

如果不满足 where 子句的条件，就会执行默认的 case 分支。

最后的输出为：(3, 4) is just some arbitrary point

在 for 循环中的使用

在 for 循环中，也可以使用 where 子句来添加条件判断：

let numbers = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

for number in numbers where number % 2 == 0 {
    print(number)
}

// 输出:
// 2
// 4
// 6
// 8
// 10

在这个例子中，我们遍历数字数组，但只打印出能被 2 整除的偶数。

在协议扩展中的用法

我们都知道，在 swift 中可以通过为协议添加扩展 (protocol extension) 来实现协议的默认实现。此时可以在扩展的 where 子句中添加约束：

protocol Shape {
    func draw()
}

extension Shape where Self: Equilateral {
    func draw() {
        // 绘制等边形
    }
}

extension Shape where Self: Rectangle {
    func draw() {
        // 绘制矩形
    }
}

struct Square: Shape, Equilateral { }

let square = Square()
square.draw() // 绘制等边形

我们为 Shape 协议添加了两个扩展，其中为 Equilateral 和 Rectangle 添加了 draw 方法的默认实现。

在扩展的 where 子句中指定了 Self 的约束类型。然后 Square 结构体实现了 Shape 和 Equilateral 协议，此时会调用 Equilateral 的 draw 实现。

通过 where 子句，可以根据协议扩展的指定类型进行有条件的默认实现，使协议扩展更加灵活。

各种高阶函数中的用法

1、查找满足条件的第一个元素

let numbers = [1, 2, 3, 4, 5]
let number = numbers.first(where: { $0 > 2 })
print(number!) // 输出: 3

2、查找满足条件的最后一个元素

let numbers = [1, 2, 3, 4, 5]
let number = numbers.last(where: { $0 > 2 })
print(number!) // 输出: 5

3、判断是否包含满足条件的元素

let numbers = [1, 2, 3, 4, 5]
let isContains = numbers.contains(where: { $0 > 2 })
print(isContains) // 输出: true

4、查找满足条件的第一个元素的下标

let numbers = [1, 2, 3, 4, 5]
let index = numbers.firstIndex(where: { $0 > 2 })
print(index!) // 输出: 2

5、查找满足条件的最后一个元素的下标

let numbers = [1, 2, 3, 4, 5]
let index = numbers.lastIndex(where: { $0 > 2 })
print(index!) // 输出: 4

6、将数组按条件拆分

let numbers = [1, 2, 3, 4, 5]
let array = numbers.split(whereSeparator: { $0 > 3 || $0 < 2 } )
print(array) // 输出: [ArraySlice([2, 3])]

在泛型函数中的用法

可以用 where 关键字向泛型参数添加约束：

protocol Printable {
    func print()
}

func printBoth<P1: Printable, P2: Printable>(_ p1: P1, _ p2: P2) where P1 == P2 {
    p1.print()
    p2.print()
}

在协议声明中的使用

在协议的声明中，可以为关联类型（associatedtype）的类型增加约束：

protocol Stack {
  associatedtype Element where Element: CustomStringConvertible
  mutating func push (_ element: Element)
  mutating func pop () -> Element?
}

这里每天分享一个 iOS 的新知识，快来关注我吧