Swift中为可选元素数组创建返回非可选类型方法的扩展可行吗？

你想要的需求完全可以实现，不过要注意Swift里Optional是一个枚举类型，不是协议，所以不能直接用Element: Optional这种约束。我们可以通过两种方式来实现你的目标：

方法一：利用泛型约束直接关联Optional的Wrapped类型 #

这种方法不需要额外定义协议，直接在方法层面添加泛型约束，明确Element就是Optional<Wrapped>：

extension Array {
    /// 将可选元素数组转换为非可选元素数组（自动过滤nil）
    func foo() -> [Wrapped] where Element == Optional<Wrapped> {
        // 这里可以根据你的需求实现逻辑，比如过滤nil返回非可选数组
        return compactMap { $0 }
    }
}

使用示例： #

let optionalStrings: [String?] = ["Swift", nil, "Extension", nil]
let nonOptionalStrings = optionalStrings.foo()
// nonOptionalStrings 的类型是 [String]，值为 ["Swift", "Extension"]

方法二：通过自定义协议统一约束Optional类型 #

如果你希望扩展本身就只作用于元素是可选类型的数组，可以先定义一个能提取Wrapped类型的协议，再让Optional遵循它：

// 定义协议，用于提取Optional的Wrapped类型
protocol OptionalType {
    associatedtype Wrapped
    var wrappedValue: Wrapped? { get }
}

// 让原生Optional遵循这个协议
extension Optional: OptionalType {
    var wrappedValue: Wrapped? { self }
}

// 扩展只作用于元素遵循OptionalType的数组
extension Array where Element: OptionalType {
    func foo() -> [Element.Wrapped] {
        return compactMap { $0.wrappedValue }
    }
}

这种写法的好处是扩展的约束更直观，方法不需要额外的泛型参数，调用起来和普通方法一样。

核心思路说明 #

你的伪代码里extension Array where Element: Optional的问题在于Optional是枚举而非协议，所以不能用:来做约束。我们的核心思路是通过泛型关联或者自定义协议，获取Optional的Wrapped关联类型，这样就能把返回类型指定为非可选的Wrapped类型。

内容的提问来源于stack exchange，提问作者Balazs Szamody