Question

我正在尝试将通用Array方法compactMap包装在Array扩展中，以使该方法的目的更具意义/可读性。我只是想采用一个Optional数组，并从中删除所有的nil值。

extension Array {
    public func removeNilElements() -> [Element] {
        let noNils = self.compactMap { $0 }
        return noNils // nil values still exist
    }
}

我遇到的问题是这里的compactMap无法正常工作。 nil值仍在结果数组noNils中。当我不使用此包装器而直接使用compactMap方法时，得到的是没有nil值的数组的预期结果。

let buttons = [actionMenuButton, createButton]   // [UIBarButtonItem?]
let nonNilButtons = buttons.compactMap { $0 }    // works correctly
let nonNilButtons2 = buttons.removeNilElements() // not working

我没有正确设计扩展方法吗？

Answer 1

您必须为 optional 元素数组定义方法，并将返回类型定义为非可选数组的对应数组。这可以通过通用函数完成：

extension Array {
    public func removeNilElements<T>() -> [T] where Element == T?    {
        let noNils = self.compactMap { $0 }
        return noNils
    }
}

示例：

let a = [1, 2, nil, 3, nil, 4]   // The type of a is [Int?]
let b = a.removeNilElements()    // The type of b is [Int]
print(b) // [1, 2, 3, 4]

在您的代码中，$0具有（非可选）类型Element，并且它只是由编译器包装为可选的，以匹配compactMap()的参数类型。

在类扩展中包装通用方法

1 个答案: