在 Swift 中使用 CaseIterable

在 Swift 中，有时需要在 switch 语句中使用全部枚举值，或者需要在 UICollectionView 或 UIPickerView 中使用全部枚举值来填充数据源。CaseIterable 协议解决了这个问题。

介绍

CaseIterable 是一个将枚举声明为可迭代（iterable）的协议。遵循此协议的枚举可通过枚举的 allCases 属性访问所有可能的情况。

接下来，我们将使用一个示例枚举来演示如何使用 CaseIterable 协议。

enum Direction {
    case north
    case south
    case east
    case west
}

使用

为了使 Direction 枚举可迭代，我们需要扩展 Direction 并让其遵循 CaseIterable 协议：

extension Direction: CaseIterable {}

现在，我们可以通过 allCases 属性访问 Direction 中的所有情况：

let allDirections = Direction.allCases 
// [Direction.north, Direction.south, Direction.east, Direction.west]

我们也可以在 switch 语句中使用 allCases：

for direction in Direction.allCases {
    switch direction {
    case .north:
        print("向北")
    case .south:
        print("向南")
    case .east:
        print("向东")
    case .west:
        print("向西")
    }
}

还可以将 allCases 用于 UICollectionView 和 UIPickerView 的数据源：

func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
    return Direction.allCases.count
}

func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {
    let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "DirectionCell", for: indexPath) as! DirectionCell
    
    let direction = Direction.allCases[indexPath.row]
    cell.directionLabel.text = direction.rawValue.capitalized // "North", "South", "East", "West"
    
    return cell
}

总结

CaseIterable 是一个非常有用的协议，可使遵循协议的枚举变得可迭代，并允许您使用 allCases 访问所有情况。使用 CaseIterable 协议可改善代码的可读性，并使维护更加容易。