为什么通过扩展使用的 flatMap 与直接调用时返回的结果不同？

Question

考虑这段代码...

import Foundation

let source = ["A", "B", nil, "D"]
print(type(of:source))

let result1 = source.flatMap{ $0 }
print(type(of:result1))
print(result1)

extension Array
{
    func sameThing() -> Array
    {
        return self.flatMap{ $0 }
    }
}

let result2 = source.sameThing()
print(type(of:result2))
print(result2)

result1 是 Array<String>，而 result2 是 Array<Optional<String>>。但为什么呢？

我尝试过使用序列而不是数组，但也没有运气。

Answer 1

您的返回类型保留泛型类型

func sameThing(separator:String = " ") -> Array

因此，此操作的可选元素签名等价于[T?] -> [T?]

Sequence.flatMap 有两个带有特制签名的重载

func flatMap<SegmentOfResult : Sequence>(
    _ transform: (${GElement}) throws -> SegmentOfResult
) rethrows -> [SegmentOfResult.${GElement}]

func flatMap<ElementOfResult>(
    _ transform: (${GElement}) throws -> ElementOfResult?
) rethrows -> [ElementOfResult] {

第二个获胜。但是由于您要求提供选项数组，ElementOfResult 变为 Optional<String>，transform 变为 Optional<String> -> Optional<Optional<String>>。

因此，当闭包{ $0 } 返回nil 时，它被提升到Optional(nil)，即具有.some(nil) 值，而不是.none。然后它将通过 nil 守卫并出现在结果中。