map(keyPath) onde keyPath é uma variável

Sep 06 2020
let arr = [(1, 1), (2, 2), (3, 3), (4, 4), (5, 5)]
arr.map(\.0) // [1, 2, 3, 4, 5]

Funciona bem. Mas o código abaixo não compila:

let keyPath = \(Int, Int).0
arr.map(keyPath)

Não é possível converter o valor do tipo 'WritableKeyPath<(Int, Int), Int>' para o tipo de argumento esperado '((Int, Int)) throws -> T'.
O parâmetro genérico 'T' não pôde ser inferido.

Respostas

4 NewDev Sep 06 2020 at 20:41

Array.mapespera um encerramento com assinatura (Element) throws -> T.

No Swift 5.2, os caminhos-chave podiam ser passados ​​como funções/fechamento (aqui está uma proposta de evolução ), mas apenas como literais (pelo menos, de acordo com a proposta, diz "por enquanto", então talvez essa restrição fosse levantada ).

Para superar isso, você pode criar uma extensão Sequenceque aceite um caminho de chave:

extension Sequence {
   func map<T>(_ keyPath: KeyPath<Element, T>) -> [T] {
      return map { $0[keyPath: keyPath] }
   }
}

(crédito para:https://www.swiftbysundell.com/articles/the-power-of-key-paths-in-swift/)

Então você pode fazer o que você queria:

let keyPath = \(Int, Int).0
arr.map(keyPath)
Jessy Sep 06 2020 at 22:24

A proposta de evolução mostrou como fazer isso com um operador, mas você também pode usar a mesma sintaxe []ou ()parcialmente aplicada ou não, pois subscritos e funções não precisam de argumentos.

let oneTo5 = 1...5
let keyPath = \(Int, Int).0
XCTAssert(
  zip(oneTo5, oneTo5).map(keyPath[]).elementsEqual(oneTo5)
)
let keyPath = \Double.isZero
XCTAssertFalse(keyPath[1.0]())
public extension KeyPath {
  /// Convert a `KeyPath` to a partially-applied get accessor.
  subscript() -> (Root) -> Value {
    { $0[keyPath: self] } } /// Convert a `KeyPath` to a get accessor. subscript(root: Root) -> () -> Value { { root[keyPath: self] } } } public extension ReferenceWritableKeyPath { /// Convert a `KeyPath` to a partially-applied get/set accessor pair. subscript() -> (Root) -> Computed<Value> { { self[$0] }
  }

  /// Convert a `KeyPath` to a get/set accessor pair.
  subscript(root: Root) -> Computed<Value> {
    .init(
      get: self[root],
      set: { root[keyPath: self] = $0 }
    )
  }
}


/// A workaround for limitations of Swift's computed properties.
///
/// Limitations of Swift's computed property accessors:
/// 1. They are not mutable.
/// 2. They cannot be referenced as closures.
@propertyWrapper public struct Computed<Value> {
  public typealias Get = () -> Value
  public typealias Set = (Value) -> Void

  public init(
    get: @escaping Get,
    set: @escaping Set
  ) {
    self.get = get
    self.set = set
  }

  public var get: Get
  public var set: Set

  public var wrappedValue: Value {
    get { get() }
    set { set(newValue) }
  }

  public var projectedValue: Self {
    get { self }
    set { self = newValue }
  }
}

//MARK:- public
public extension Computed {
  init(
    wrappedValue: Value,
    get: @escaping Get = {
      fatalError("`get` must be assigned before accessing `wrappedValue`.")
    },
    set: @escaping Set
  ) {
    self.init(get: get, set: set)
    self.wrappedValue = wrappedValue
  }
}