Swift 枚举类型，你知道几个？-51CTO.COM

本文我们将介绍在 Swift 中如何定义枚举、遍历枚举、枚举原始值、枚举关联值等相关的内容。如果你尚未安装 Xcode 和配置 Swift 开发环境，请您先阅读这篇文章。

接下来，我们启动 Xcode，然后选择 "File" > "New" > "Playground"。创建一个新的 Playground 并命名为 "Enumerations"。

在 Swift 中，枚举（Enum）是一种特殊的数据类型，它允许你定义一组相关的值。这些值是你在程序中会用到的一些具体选项。

定义一个枚举

在 Swift 中，我们使用 enum 关键字定义一个枚举，在枚举体内使用 case 关键字定义不同的情况，每个情况表示枚举的一个成员。

Swift Code

enum Color {
    case red
    case green
    case blue
}

let greenColor = Color.green

print(greenColor)
// Output: green1.
2.
3.
4.
5.
6.
7.
8.
9.
10.

在以上代码中，我们定义了一个名为 Color 的枚举，包含了三种颜色。

TypeScript Code

enum Color {
    Red,
    Green,
    Blue
}

let color: Color = Color.Green;

console.log(color);
// Output: 11.
2.
3.
4.
5.
6.
7.
8.
9.
10.

使用 switch 处理枚举

有了 Color 枚举后，我们可以使用 switch 语句来处理枚举。

Swift Code

enum Color {
    case red
    case green
    case blue
}

func describeColor(color: Color) {
    switch color {
    case .red:
        print("Color is red.")
    case .green:
        print("Color is green.")
    case .blue:
        print("Color is blue.")
    }
}

describeColor(color: .blue)

// Output: Color is blue.1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
16.
17.
18.
19.
20.

TypeScript Code

enum Color {
    Red,
    Green,
    Blue
}

function describeColor(color: Color): void {
    switch (color) {
        case Color.Red:
            console.log("Color is red.");
            break;
        case Color.Green:
            console.log("Color is green.");
            break;
        case Color.Blue:
            console.log("Color is blue.");
            break;
    }
}

describeColor(Color.Blue);

// Output: "Color is blue."1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
16.
17.
18.
19.
20.
21.
22.
23.

遍历枚举的成员

在 Swift 中，我们可以使用 CaseIterable 协议来使枚举遵循可迭代的协议，从而实现对枚举成员的遍历。

Swift Code

enum Color: CaseIterable {
    case red, green, blue
}

for color in Color.allCases {
    print(color)
}

/**
Output: 
red
green
blue
*/1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.

在上面的代码中，我们让 Color 枚举遵循 CaseIterable 协议，以便枚举该枚举的所有成员。

TypeScript Code

enum Color {
    Red,
    Green,
    Blue
}

for(let colorKey in Color) {
    console.log(colorKey)
}

/**
Output: 
"0" 
"1" 
"2" 
"Red" 
"Green" 
"Blue" 
*/1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
16.
17.
18.
19.

枚举原始值

Swift 中的枚举可以关联原始值，这些原始值可以是整数、浮点数、字符串等类型。枚举的原始值为每个成员提供了一个默认值，方便我们在不同的上下文中使用。

数值原始值

Swift Code

enum Weekday: Int {
    case sunday = 1
    case monday
    case tuesday
    case wednesday
    case thursday
    case friday
    case saturday
}

let today: Weekday = .tuesday
let rawValue: Int = today.rawValue

print(rawValue)
// Output: 31.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.

在以上代码中，我们定义了一个表示星期的枚举 Weekday，并为每个成员显式赋予了一个原始值。默认情况下，第一个成员的原始值为 1，后续成员的原始值递增。

TypeScript Code

enum Weekday {
    Sunday = 1,
    Monday,
    Tuesday,
    Wednesday,
    Thursday,
    Friday,
    Saturday
}

let today: Weekday = Weekday.Tuesday;
let rawValue: number = today;

console.log(rawValue);
// Output: 31.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.

在 TypeScript 中，数值枚举的原始值也是递增的，与 Swift 中的数值枚举相似。

字符串原始值

Swift Code

enum Direction: String {
    case up = "UP"
    case down = "DOWN"
    case left = "LEFT"
    case right = "RIGHT"
}

let move: Direction = .up
let directionString: String = move.rawValue

print(directionString)
// Output: UP1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.

在以上代码中，我们定义了一个字符串枚举 Direction，为每个成员显式赋予了一个字符串原始值。

TypeScript Code

enum Direction {
    Up = "UP",
    Down = "DOWN",
    Left = "LEFT",
    Right = "RIGHT"
}

let move: Direction = Direction.Up;
let directionString: string = move;

console.log(directionString);
// Output: "UP"1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.

字符串枚举的原始值在 TypeScript 中也是类似的，允许为每个成员指定字符串类型的原始值。

枚举关联值

Swift 中的枚举不仅可以有原始值，还可以携带关联值。关联值允许在定义枚举的时候指定每个成员携带的数据类型。这样，每个枚举成员都可以携带不同类型的数据。

Swift Code

import Foundation

// 定义 Shape 枚举描述不同的图形
enum Shape {
    case circle(radius: Double)
    case square(side: Double)
    case rectangle(width: Double, height: Double)
}

// 使用关联值创建不同的图形
let circle: Shape = .circle(radius: 3.0)
let square: Shape = .square(side: 4.0)
let rectangle: Shape = .rectangle(width: 3.0, height: 4.0)1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.

在以上代码中，我们定义了一个 Shape 枚举，其中的每个成员都可以携带不同类型的关联值，表示不同的图形。有了 Shape 枚举之后，我们可以创建一个 calculateArea 函数，来计算不同图形的面积。

Swift Code

func calculateArea(shape: Shape) -> Double {
    switch shape {
    case .circle(let radius):
        return Double.pi * pow(radius, 2)
    case .square(let side):
        return pow(side, 2)
    case .rectangle(let width, let height):
        return width * height
    }
}

// 计算不同图形的面积
let areaOfCircle = calculateArea(shape: circle) // 28.27433388230814
let areaOfSquare = calculateArea(shape: square) // 16
let areaOfRectangle = calculateArea(shape: rectangle) // 121.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.

在以上代码中，我们定义了一个函数 calculateArea，根据图形的类型计算图形的面积。通过关联值，我们可以轻松地提取不同图形的属性进行计算。在 TypeScript 中，由于并没有直接对应 Swift 枚举关联值的语法，我们可以使用 TypeScript 的联合类型来模拟这种行为。

TypeScript Code

interface Circle {
    kind: 'circle';
    radius: number;
}

interface Square {
    kind: 'square';
    side: number;
}

interface Rectangle {
    kind: 'rectangle';
    width: number;
    height: number;
}

// 使用联合类型表示不同的图形
type Shape = Circle | Square | Rectangle;1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
16.
17.
18.

在以上代码中，我们使用接口和联合类型来定义不同图形的数据结构。之后，我们也可以定义一个 calculateArea 函数来计算不同图形的面积。

TypeScript Code

function calculateArea(shape: Shape): number {
    switch (shape.kind) {
        case 'circle':
            return Math.PI * Math.pow(shape.radius, 2);
        case 'square':
            return Math.pow(shape.side, 2);
        case 'rectangle':
            return shape.width * shape.height;
        default:
            throw new Error('Invalid shape');
    }
}

const circle: Circle = { kind: "circle", radius: 3.0 }
const square: Square = { kind: "square", side: 4.0 }
const rectangle: Rectangle = { kind: "rectangle", width: 3.0, height: 4.0 }

// 计算不同图形的面积
const areaOfCircle = calculateArea(circle); // 28.274333882308138
const areaOfSquare = calculateArea(square); // 16
const areaOfRectangle = calculateArea(rectangle); // 121.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
16.
17.
18.
19.
20.
21.

枚举中定义计算属性

Swift Code

enum Color {
    case red, green, blue

    var hexValue: String {
        switch self {
        case .red:
            return "#FF0000"
        case .green:
            return "#00FF00"
        case .blue:
            return "#0000FF"
        }
    }
}

let greenColor = Color.green

print(greenColor.hexValue)
// Output: #00FF001.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
16.
17.
18.
19.

在以上代码中，我们为 Color 枚举增加了一个计算属性 hexValue，用于表示颜色的十六进制值。

枚举中定义方法

Swift Code

enum Color {
    case red, green, blue

    func description() -> String {
        switch self {
        case .red:
            return "Color is red."
        case .green:
            return "Color is green."
        case .blue:
            return "Color is blue."
        }
    }
}

let greenColor = Color.green

print(greenColor.description())
// Output: Color is green.1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
16.
17.
18.
19.

在以上代码中，我们在 Color 枚举中添加了一个 description 方法，用于返回颜色的描述信息。

本文我们介绍了在 Swift 中如何定义枚举、遍历枚举、枚举原始值、枚举关联值等相关的内容。通过与 TypeScript 语法的对比，希望能帮助您更好地理解 Swift 的相关特性。