关注我,每天分享一个关于 iOS 的新知识
前言
where
是 Swift 中一个强大的关键字,可以轻松过滤掉一些值。它可以用于许多不同的表达式中,今天就来盘点一下。
在 switch 中使用
在 switch
语句中的 case
中,可以用 where
关键字做二次筛选,举个例子:
swift
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 子句来添加条件判断:
typescript
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
子句中添加约束:
scss
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、查找满足条件的第一个元素
typescript
let numbers = [1, 2, 3, 4, 5]
let number = numbers.first(where: { $0 > 2 })
print(number!) // 输出: 3
2、查找满足条件的最后一个元素
typescript
let numbers = [1, 2, 3, 4, 5]
let number = numbers.last(where: { $0 > 2 })
print(number!) // 输出: 5
3、判断是否包含满足条件的元素
bash
let numbers = [1, 2, 3, 4, 5]
let isContains = numbers.contains(where: { $0 > 2 })
print(isContains) // 输出: true
4、查找满足条件的第一个元素的下标
bash
let numbers = [1, 2, 3, 4, 5]
let index = numbers.firstIndex(where: { $0 > 2 })
print(index!) // 输出: 2
5、查找满足条件的最后一个元素的下标
bash
let numbers = [1, 2, 3, 4, 5]
let index = numbers.lastIndex(where: { $0 > 2 })
print(index!) // 输出: 4
6、将数组按条件拆分
ini
let numbers = [1, 2, 3, 4, 5]
let array = numbers.split(whereSeparator: { $0 > 3 || $0 < 2 } )
print(array) // 输出: [ArraySlice([2, 3])]
在泛型函数中的用法
可以用 where
关键字向泛型参数添加约束:
swift
protocol Printable {
func print()
}
func printBoth<P1: Printable, P2: Printable>(_ p1: P1, _ p2: P2) where P1 == P2 {
p1.print()
p2.print()
}
在协议声明中的使用
在协议的声明中,可以为关联类型 (associatedtype
) 的类型增加约束:
swift
protocol Stack {
associatedtype Element where Element: CustomStringConvertible
mutating func push (_ element: Element)
mutating func pop () -> Element?
}
这里每天分享一个 iOS 的新知识,快来关注我吧
本文同步自微信公众号 "iOS新知",每天准时分享一个新知识,这里只是同步,想要及时学到就来关注我吧!