Swift 扩展

扩展允许你在不修改源代码的情况下向现有的类、结构体或枚举类型添加功能。这包括但不限于方法的添加、计算属性的定义以及下标操作符的重载等。

添加方法

扩展最常见的一种用途是为现有类型添加新的方法。这可以让你在不改变原有类、结构体或枚举代码的情况下为其增加新的行为。

extension Int {
    func squared() -> Int {
        return self * self
    }
}

let number = 5
print(number.squared()) // 输出：25

这里我们为 Int 类型添加了一个名为 squared 的方法，用于计算一个整数的平方。

添加计算属性

扩展不仅可以添加方法，还可以添加计算属性。需要注意的是，扩展不能添加存储属性，也不能重写现有的属性。

// www.javascriptcn.com code example
extension Double {
    var kilometers: Double { return self / 1000 }
    var meters: Double { return self }
    var centimeters: Double { return self * 100 }
}

let distance = 12345.0
print(distance.kilometers) // 输出：12.345
print(distance.meters)     // 输出：12345.0
print(distance.centimeters) // 输出：1234500.0

在上面的例子中，我们为 Double 类型添加了几个计算属性，用于方便地将距离转换成不同的单位。

提供新构造器

扩展也可以为现有的类型提供新的初始化方法（构造器）。这可以用于简化对象的创建过程，或者提供一些特定场景下的初始化方式。

// www.javascriptcn.com code example
extension String {
    init?(numeric value: Int) {
        if value >= 0 && value <= 9 {
            self.init(value.description)
        } else {
            return nil
        }
    }
}

if let singleDigit = String(numeric: 7) {
    print("Single digit: \(singleDigit)") // 输出：Single digit: 7
}

这个例子中，我们为 String 类型添加了一个新的初始化方法，它接受一个整数值，并将其转换为对应的单个字符的字符串形式。如果输入的值不在有效范围内，则返回 nil。

使用协议扩展实现默认行为

通过扩展协议，你可以为协议的所有采纳者提供默认实现。这可以极大地减少重复代码，并且使得遵循该协议的类型能够继承这些默认的行为。

// www.javascriptcn.com code example
protocol VolumeConvertible {
    var volume: Double { get set }
}

extension VolumeConvertible {
    func showVolume() {
        print("Volume is \(volume)")
    }
}

struct Cube: VolumeConvertible {
    var volume: Double
}

let cube = Cube(volume: 123.45)
cube.showVolume() // 输出：Volume is 123.45

在这个例子中，我们定义了一个名为 VolumeConvertible 的协议，并为其添加了一个名为 showVolume 的方法。然后，我们为这个协议提供了一个默认实现，这样任何遵循此协议的类型都可以直接使用这个方法，而无需再次实现它。

为已有类型添加协议遵守

扩展不仅限于添加新的方法和属性，它们还可以用来使现有的类型遵守特定的协议。这对于那些无法直接修改以遵守协议的类型特别有用。

// www.javascriptcn.com code example
struct Rectangle {
    var width: Double
    var height: Double
}

extension Rectangle: CustomStringConvertible {
    var description: String {
        return "Rectangle with width \(width) and height \(height)"
    }
}

let rectangle = Rectangle(width: 10, height: 20)
print(rectangle) // 输出：Rectangle with width 10.0 and height 20.0

在这个例子中，我们让 Rectangle 结构体遵守了 CustomStringConvertible 协议，从而可以自定义其字符串表示形式。

以上就是关于 Swift 扩展的一些基本概念及使用方法。通过合理利用扩展，我们可以有效地增强现有类型的可用性，同时保持代码的整洁与可维护性。