可解码属性包装器的自动可解码合成

Question

假设我有可解码的属性包装器：

@propertyWrapper
struct OptionalDecodable<Value: Decodable>: Decodable {
  var wrappedValue: Value?
}

编译器会为以下内容合成 init

struct Model: Decodable {
  @OptionalDecodable private(set) var string: String?
}

为了测试这是否可行，我只是尝试解码空 JSON（即“{}”）

但是，string 属性不被视为可选，即当没有 string 键时，我收到一个错误，即找不到键。

有解决办法吗？

Answer 1

我不确定这是否是最好的方法，但问题是属性包装器的wrappedValue 类型必须匹配属性的类型，而String 与String? 不同。

解决此问题的一种方法是使属性包装器通用，但以允许您从 String 或 Int 初始化类型的方式进行约束：

protocol ExpressibleByString {
    init(fromString: String)
}
extension String: ExpressibleByString {
    init(fromString: String) { self = fromString }
}
extension Optional: ExpressibleByString where Wrapped == String {
    init(fromString: String) { self = fromString }
}

@propertyWrapper
struct IntOrString<V: ExpressibleByString & Decodable>: Decodable {
    var wrappedValue: V
}

extension IntOrString {
    init(from decoder: Decoder) throws {
        let container = try decoder.singleValueContainer()
        
        do {
            let int = try container.decode(Int.self)
            wrappedValue = .init(fromString: int.description)
        } catch DecodingError.typeMismatch {
            wrappedValue = try .init(fromString: container.decode(String.self))
        }
   }
}

extension KeyedDecodingContainer {
    func decode<V: ExpressibleByNilLiteral>(_ t: IntOrString<V>.Type, forKey key: K) throws -> IntOrString<V> {
        if let v = try decodeIfPresent(t, forKey: key) {
            return v
        }
        return IntOrString(wrappedValue: nil)
    }
}

然后您可以在可选和非可选String 上使用它：

struct Foo: Decodable {
    @IntOrString
    var p1: String?

    @IntOrString
    var p2: String
}