最近看 SwiftUI 的時候使用了 @State 和 @Binding, 在看API 的時候發(fā)現(xiàn), 他們的實現(xiàn)都是通過 @propertyWrapper 實現(xiàn)的.
@propertyWrapper
它能干什么?
可以減少重復代碼
舉個例子:
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 通常采用上面的方式, 每個 Key 都要實現(xiàn)一次 get set,
產生了大量的工作.
可以使用 @propertyWrapper 來替換這部分代碼, 實現(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
看過上面代碼, 開始具體說一下 @propertyWrapper.
簡單的理解就是包裝屬性.
怎么包裝? 替換 set 和 get 方法
每個 @propertyWrapper 都必須實現(xiàn)一個屬性 wrappedValue, 實現(xiàn) wrappedValue 的 set get 就相當于替換掉原屬性的 set get.
如果你想實現(xiàn)被 @propertyWrapper 包裹的屬性 get set, 會產生錯誤 Property wrapper cannot be applied to a computed property
@propertyWrapper 的思想是, 只可以通過 wrappedValue 的 get set 方法來訪問屬性, 所以它默認的初始化方法是下面的方法, 給wrappedValue初始值, 這個是從Swift Property Wrappers 復制過來的時候發(fā)現(xiàn)的, 但你也可以不實現(xiàn)這個方法, 通過其他的方式初始化, 比如上面 UserDefaults 的實現(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 可以使用的地方有很多, 一些通用的實現(xiàn)都可以使用它來減少重復代碼.
下面是 RGB 通用的簡單實現(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
