Swift 扩展：extension 为类型添加方法

Swift 扩展：extension 为类型添加方法

Swift 语言中的 extension（扩展）是一种在不修改原始类型源代码的情况下，为其添加新功能的机制。这种机制不仅适用于自定义的结构体、类和枚举，也适用于系统库中的基础类型（如 Int、String）。通过扩展，开发者可以将代码逻辑按功能模块划分，提高代码的可读性和可维护性。

1. 理解扩展的基本语法

扩展使用 extension 关键字声明。其基本结构遵循 Swift 的标准代码块规则。

打开 Xcode 代码编辑器。

输入 extension 关键字，后接要扩展的类型名称。

添加 一对大括号 {}，所有的扩展内容都必须写在这对大括号内部。

以下代码展示了为 Double 类型添加一个基础扩展的骨架：

extension Double {
    // 在这里添加新的方法或计算属性
}

2. 为值类型添加实例方法

值类型（如 struct 和 enum）在方法中修改自身属性时，需要特别注意。如果方法需要修改 self 或其属性，必须使用 mutating 关键字标记该方法。

2.1 扩展基础数值类型

假设我们需要为 Int 类型添加一个方法，用于判断该整数是否为偶数。

定义 扩展块，目标类型为 Int。

声明 一个名为 isEven 的方法，返回值为 Bool 类型。

实现 逻辑，判断该数字除以 2 的余数是否为 0。

extension Int {
    func isEven() -> Bool {
        return self % 2 == 0
    }
}

调用 该方法时，直接在整数字面量或变量后使用点语法。

let number = 10
if number.isEven() {
    print("这是一个偶数")
}

2.2 使用 mutating 修改自身

如果扩展方法需要修改值类型实例本身的值，必须显式声明为 mutating。

编写 一个将 Int 值平方的方法。

在 func 前添加 mutating 关键字。

将 计算结果赋值给 self。

extension Int {
    mutating func square() {
        self = self * self
    }
}

var myNumber = 5
myNumber.square()
// myNumber 现在是 25

3. 为引用类型添加方法

对于引用类型（即 class），扩展方法的编写更为直接。由于类是引用类型，方法内部可以直接修改属性，无需 mutating 关键字。

假设有一个简单的 User 类，我们需要通过扩展为其添加功能。

定义 目标类。

class User {
    var name: String
    var age: Int

    init(name: String, age: Int) {
        self.name = name
        self.age = age
    }
}

创建 扩展，添加一个打印用户信息的方法。

extension User {
    func printProfile() {
        print("用户名: \(name), 年龄: \(age)")
    }
}

实例化 类并调用扩展方法。

let currentUser = User(name: "张三", age: 28)
currentUser.printProfile()

4. 通过扩展添加计算属性

扩展虽然不能添加存储属性，但可以添加计算属性。这实际上等同于添加了一个不带参数的方法，常用于便捷的数据转换。

扩展 Double 类型。

添加 计算属性 km，用于将米转换为千米。

extension Double {
    var km: Double {
        return self / 1000.0
    }
}

let distance = 3000.0
print("\(distance.km) 公里") // 输出: 3.0 公里

5. 扩展协议与默认实现

Swift 扩展最强大的特性之一是为协议提供默认实现。这意味着只要类型遵循了该协议，就自动获得了方法功能，无需在每个类型中重复编写代码。

5.1 定义协议与默认实现

声明 一个协议 Identifiable，包含一个 id 属性和一个 describe 方法。

protocol Identifiable {
    var id: String { get }
    func describe() -> String
}

编写 协议扩展，为 describe 方法提供默认实现。

extension Identifiable {
    func describe() -> String {
        return "当前对象的 ID 是: \(id)"
    }
}

5.2 遵循协议并自动获得方法

定义 一个结构体 Item，使其遵循 Identifiable 协议。

实现 协议要求的 id 属性。

注意：不需要实现 describe 方法，因为扩展已经提供了默认实现。

struct Item: Identifiable {
    var id: String
}

let myItem = Item(id: "ITEM_001")
print(myItem.describe()) // 输出: 当前对象的 ID 是: ITEM_001

6. 扩展的约束条件

有时，我们只想为特定条件下的类型添加方法。例如，只想为数组元素为 Int 的数组添加求和方法。这需要使用 where 子句。

声明 对 Array 的扩展。

添加 约束条件 where Element == Int。

编写 求和逻辑。

extension Array where Element == Int {
    func sum() -> Int {
        return self.reduce(0, +)
    }
}

let scores = [90, 85, 95]
print("总分: \(scores.sum())") // 输出: 总分: 270

7. 扩展与继承的区别

为了更清晰地理解扩展的定位，可以通过下表对比其与继承的核心差异。

当只需要添加功能，而不需要改变对象本质或管理对象生命周期时，优先选择扩展。

8. 方法查找与冲突处理流程

当一个类型同时存在原生方法和扩展方法，或多个扩展定义了同名方法时，编译器遵循一套严格的查找逻辑。

如果在不同的扩展中定义了完全相同签名的方法，编译器会根据引入的文件顺序或命名空间产生歧义报错。此时必须通过显式指定类型调用，或者重命名方法以解决冲突。

9. 最佳实践建议

在使用扩展时，遵循以下原则可以保持代码库的整洁：

1. 划分功能模块：将不同的功能逻辑放入不同的 extension 块中，并使用 MARK: 注释进行标记。

   // MARK: - 网络请求相关
   extension User {
       func fetchDetails() { ... }
   }
   
   // MARK: - UI 展示相关
   extension User {
       func formatDisplayName() -> String { ... }
   }

2. 避免滥用：如果某个方法是对象核心生命周期或核心逻辑的一部分，应直接定义在主类型定义中，而非扩展中。扩展更适合添加“辅助性”或“边缘性”的功能。