Swift 下标脚本

在 Swift 中，可以为自定义类型实现下标脚本，方便用户通过类似于数组的方式访问对象的成员变量或方法。

下标脚本语法

下标脚本语法采用 subscript 关键字，和函数语法类似，可以有多个参数和返回值类型。下面是一个简单的例子：

struct MyStruct {
    var data = [Int]()
    
    subscript(index: Int) -> Int {
        get {
            return data[index]
        }
        set(newValue) {
            data[index] = newValue
        }
    }
}

var myStruct = MyStruct()
myStruct[0] = 1
myStruct[1] = 2
print(myStruct[0]) // 1

上面的例子定义了一个 MyStruct 结构体，其中包含一个 data 成员变量和一个返回 Int 类型的下标脚本。下标脚本中的 get 和 set 方法可以通过下标语法来访问和修改 data 数组。

下标脚本的参数

下标脚本的参数可以是任意类型的。下面是一个例子：

class MyArray<T> {
    private var data = [T]()
    
    subscript(index: Int) -> T? {
        get {
            if index >= 0 && index < data.count {
                return data[index]
            }
            return nil
        }
        set(newValue) {
            if index >= 0 && index < data.count, let value = newValue {
                data[index] = value
            }
        }
    }
}

var myArray = MyArray<Int>()
myArray[0] = 1
myArray[1] = 2
print(myArray[0]) // Optional(1)

上面的例子中，定义了一个泛型类 MyArray，其中可以通过下标脚本访问和修改数组元素。注意，下标脚本的形参类型为 Int，返回值类型为可选类型 T?，在访问数组元素时需要进行判断，避免下标越界。

另外，下标脚本中也可以定义多个形参，例如：

struct MyStruct {
    private var data = [Int]()
    
    subscript(row: Int, col: Int) -> Int {
        get {
            let index = row * 2 + col
            return data[index]
        }
        set(newValue) {
            let index = row * 2 + col
            data[index] = newValue
        }
    }
}

var myStruct = MyStruct()
myStruct[0, 1] = 2
print(myStruct[0, 1]) // 2

上面的例子中，定义了一个二维数组，可以通过下标脚本访问和修改数组元素。注意，下标脚本的形参包括 row 和 col 两个参数，可以根据这两个参数计算出数组索引。

下标脚本的返回值

下标脚本可以返回任何类型的值，例如：

struct MyStruct {
    private var data = ["Hello", "World"]
    
    subscript(index: Int) -> String {
        return data[index]
    }
}

var myStruct = MyStruct()
print(myStruct[0]) // "Hello"

上面的例子中，下标脚本返回了一个 String 类型的值。

还可以返回元组类型的值，例如：

struct MyStruct {
    private var data = ["Apple", "Banana", "Orange"]
    
    subscript(index: Int) -> (String, Int) {
        return (data[index], data[index].count)
    }
}

var myStruct = MyStruct()
print(myStruct[0]) // ("Apple", 5)

上面的例子中，下标脚本返回了一个包含水果名称和名称字符数的元组类型值。

总结

下标脚本是 Swift 中非常灵活的一种语法，可以方便地为自定义类型实现类似于数组的访问方式。在实现下标脚本时，需要注意下标脚本的参数类型和返回值类型，以及避免下标越界的情况。

swift教程 Swift 函数