使用紧凑映射的 Swift 语法解释

Question

我在检查代码时确实找到了以下代码：

override func layoutAttributesForElements(in rect: CGRect) -> [UICollectionViewLayoutAttributes]? {
        return super.layoutAttributesForElements(in: rect)?
            .compactMap { $0.copy() as? ParallaxLayoutAttributes }
            .compactMap(prepareAttributes)
    }

    private func prepareAttributes(attributes: ParallaxLayoutAttributes) -> ParallaxLayoutAttributes {
        // Lot of code doing stuff with attributes 
        return attributes
    }

所以，实际上我想问的是，compact 函数声明如下：

@inlinable public func compactMap<ElementOfResult>(_ transform: (Element) throws -> ElementOfResult?) rethrows -> [ElementOfResult]

在此示例中，我们只传递函数，不带参数：

.compactMap(prepareAttributes)

这完全颠覆了我的想法，因为，prepareAttributes 函数声明是这样的（你必须通过参数）：

private func prepareAttributes(attributes: ParallaxLayoutAttributes) -> ParallaxLayoutAttributes

那么，为什么上面的代码可以编译，.compactMap(prepareAttributes) 到底是怎么编译的？ 当您没有为 prepareAttributes 函数传递参数时运行？

Answer 1

在调用.compactMap(prepareAttributes) 中，您将函数prepareAttributes 作为闭包传递给compactMap。由于prepareAttributes 采用单个输入参数，其类型与compactMap 的闭包变量匹配，编译器可以自动推断它需要将$0 传递给prepareAttributes。

所以本质上，.compactMap(prepareAttributes) 是

的简写

.compactMap {prepareAttributes(attributes: $0) }

与map 相同的行为的一个简单示例是经常使用的map 类型，然后您将其传递给init，您可以将其写为.map { MyType(input: $0) } 或简化为.map(MyType.init)。

struct MyInt {
    let value: Int

    init(value: Int) {
        self.value = value
    }
}

let ints = [1,2,3]
let myInts = ints.map(MyInt.init) // same as `ints.map { MyInt(value: $0) }