是否可以中止 Swift 集合上的地图功能？

Question

我们有一个案例，我们收到了一个Array<Any> 类型的对象，我们需要将其转换为Array<Codable>。如果原始数组中的任何一项不符合Codable，那么我们希望整个事情中止并返回nil。

或者当前的方法是手动循环遍历所有内容，一路测试，就像这样......

func makeCodable(sourceArray:Array<Any>) -> Array<Codable>?{

    var codableArray = Array<Codable>()

    for item in sourceArray{

        guard let codableItem = item as? Codable else {
            return nil
        }

        codableArray.append(codableItem)
    }

    return codableArray
}

但是，我想知道是否有更简单的方法可以使用 map 命令执行此操作，但如果无法映射任何元素，则需要将其短路。这就是我不确定是否可能。

比如这个伪代码……

func makeCodable(sourceArray:Array<Any>) -> Array<Codable>?{

    return sourceArray.map({ $0 as? Codable});
}

这可能吗，还是我们原来的方式是正确/唯一的方式？

Answer 1

这是使用map 和throws 的一种解决方案。

func makeCodable(sourceArray: [Any]) -> [Codable]? {
    enum CodableError: Error {
        case notCodable
    }

    let res: [Codable]? = try? sourceArray.map {
        guard let codable = $0 as? Codable else {
            throw CodableError.notCodable
        }

        return codable
    }

    return res
}

let res = makeCodable2(sourceArray: [5, 6.5, "Hi", UIView()])
print(res) // nil

这是一个使makeCodable 抛出并返回非可选数组的变体：

enum CodableError: Error {
    case notCodable
}

func makeCodable(sourceArray: [Any]) throws -> [Codable] {
    let res: [Codable] = try sourceArray.map {
        guard let cod = $0 as? Codable else {
            throw CodableError.notCodable
        }

        return cod
    }

    return res
}

do {
    let res = try makeCodable(sourceArray: [5, 6.5, "Hi"])
    print(res) // prints array
    let bad = try makeCodable(sourceArray: [5, 6.5, "Hi", UIView()])
    print(bad)
} catch {
    print(error) // goes here on 2nd call
}

Answer 2

作为@rmaddy shows，您可以利用map(_:) 可以接受抛出闭包这一事实，并在抛出错误时停止映射，将抛出的错误传播回调用者（然后您可以使用@987654324 吸收@)。

对此的一个细微变化是定义您自己的抛出 cast(_:to:) 函数来调用转换闭包：

struct TypeMismatchError : Error {
  var expected: Any.Type
  var actual: Any.Type
}

func cast<T, U>(_ x: T, to _: U.Type) throws -> U {
  guard let casted = x as? U else {
    throw TypeMismatchError(expected: U.self, actual: type(of: x))
  }
  return casted
}

func makeCodable(sourceArray: [Any]) -> [Codable]? {
  return try? sourceArray.map { try cast($0, to: Codable.self) }
}

虽然我们完全忽略了在这种情况下抛出的错误，但我发现在其他情况下，偶尔有一个抛出转换函数很有用（你当然也可以通过将 makeCodable 设置为抛出函数和使用try)。

但是，话虽如此，请注意，您生成的 [Codable]? 在当前形式下确实不太有用。您无法从中解码内容，因为您手头没有任何具体类型，也无法直接将其编码为protocols don't conform to themselves（即Codable 不符合Encodable，因此您可以）不要只是将Codable 或[Codable] 交给JSONEncoder）。

如果您真的想对 [Codable] 进行一些编码，则需要将每个元素包装在符合 Encodable 的包装器中，例如：

struct AnyEncodable : Encodable {

  var base: Encodable

  init(_ base: Encodable) {
    self.base = base
  }

  func encode(to encoder: Encoder) throws {
    try base.encode(to: encoder)
  }
}

func makeEncodable(sourceArray: [Any]) -> [AnyEncodable]? {
  return try? sourceArray.map {
    AnyEncodable(try cast($0, to: Encodable.self))
  }
}

现在[AnyEncodable] 是您可以传递给的东西，例如JSONEncoder。