最近看 SwiftUI 的時(shí)候使用了 @State 和 @Binding, 在看API 的時(shí)候發(fā)現(xiàn), 他們的實(shí)現(xiàn)都是通過(guò) @propertyWrapper 實(shí)現(xiàn)的.

@propertyWrapper

它能干什么?
可以減少重復(fù)代碼

舉個(gè)例子:

extension UserDefaults {
    public enum Key: String {
        case someInt = "someInt"
    }
    var someInt: Int {
        get {
            return integer(forKey: Key.someInt.rawValue)
        }
        set {
            set(newValue, forKey: Key.someInt.rawValue)
        }
    }
}

之前使用 UserDefaults 通常采用上面的方式, 每個(gè) Key 都要實(shí)現(xiàn)一次 get set,
產(chǎn)生了大量的工作.

可以使用 @propertyWrapper 來(lái)替換這部分代碼, 實(shí)現(xiàn)如下:

@propertyWrapper
struct UserDefaultWrapper<T> {
    var key: String
    var defaultT: T!
    var wrappedValue: T! {
        get { (UserDefaults.standard.object(forKey: key) as? T) ?? defaultT }
        nonmutating set {
            if newValue == nil {
                UserDefaults.standard.removeObject(forKey: key)
            } else {
                UserDefaults.standard.set(newValue, forKey: key)
            }
        }
    }
    
    init(_ key: String, _ defaultT: T! = nil) {
        self.key = key
        self.defaultT = defaultT
    }
}
struct UserDefaultsUnit {
    @UserDefaultWrapper("test")
    static var test: String?
}

//使用
UserDefaultsUnit.test = "hahahha" // set
UserDefaultsUnit.test // get

看過(guò)上面代碼, 開始具體說(shuō)一下 @propertyWrapper.
簡(jiǎn)單的理解就是包裝屬性.
怎么包裝? 替換 set 和 get 方法
每個(gè) @propertyWrapper 都必須實(shí)現(xiàn)一個(gè)屬性 wrappedValue, 實(shí)現(xiàn) wrappedValue 的 set get 就相當(dāng)于替換掉原屬性的 set get.
如果你想實(shí)現(xiàn)被 @propertyWrapper 包裹的屬性 get set, 會(huì)產(chǎn)生錯(cuò)誤 Property wrapper cannot be applied to a computed property

@propertyWrapper 的思想是, 只可以通過(guò) wrappedValue 的 get set 方法來(lái)訪問(wèn)屬性, 所以它默認(rèn)的初始化方法是下面的方法, 給wrappedValue初始值, 這個(gè)是從Swift Property Wrappers 復(fù)制過(guò)來(lái)的時(shí)候發(fā)現(xiàn)的, 但你也可以不實(shí)現(xiàn)這個(gè)方法, 通過(guò)其他的方式初始化, 比如上面 UserDefaults 的實(shí)現(xiàn).

init(wrappedValue initialValue: T) {
        self.wrappedValue = initialValue
}

Codeale 使用

/// for codable
@available(swift 5.1)
@propertyWrapper
struct UserDefaultJsonWrapper<T: Codable> {
    var key: String
    var defaultT: T!
    var wrappedValue: T {
        get {
            guard let jsonString = UserDefaults.standard.string(forKey: key) else { return defaultT }
            guard let jsonData = jsonString.data(using: .utf8) else { return defaultT }
            guard let value = try? JSONDecoder().decode(T.self, from: jsonData) else { return defaultT }
            return value
        }
        set {
            let encoder = JSONEncoder()
            guard let jsonData = try? encoder.encode(newValue) else { return }
            let jsonString = String(bytes: jsonData, encoding: .utf8)
            UserDefaults.standard.set(jsonString, forKey: key)
        }
    }
    
    init(_ key: String, _ defaultT: T! = nil) {
        self.key = key
        self.defaultT = defaultT
    }
}

@propertyWrapper 可以使用的地方有很多, 一些通用的實(shí)現(xiàn)都可以使用它來(lái)減少重復(fù)代碼.
下面是 RGB 通用的簡(jiǎn)單實(shí)現(xiàn)

@propertyWrapper
struct RGBColorWrapper {
    private var r: CGFloat
    private var g: CGFloat
    private var b: CGFloat
    private var alpha: CGFloat
    
    var wrappedValue: UIColor { UIColor.init(red: r, green: g, blue: b, alpha: alpha) }
    
    init(_ r: CGFloat, _ g: CGFloat, _ b: CGFloat, _ alpha: CGFloat = 1) {
        self.r = r / 255
        self.g = g / 255
        self.b = b / 255
        self.alpha = alpha
    }
}

struct Color {
    @RGBColorWrapper(255, 0, 0)
    static var redRed: UIColor
}

// 使用
view.backgroundColor = Color.redRed

Reference: Swift Property Wrappers