Swift泛型：如何返回任意元素类型的空数组与空字典

核心思路 #

放弃字符串类型检查的方案，直接利用Swift原生的类型判断（is/as）和泛型约束，精准识别可选类型、数组、字典，返回对应的空容器，同时兼容任意元素类型的数组/字典（包括嵌套结构、元组数组等）。

实现泛型函数findOut #

以下实现支持任意元素类型的数组、字典，当元素/值为可选类型时返回空容器，同时兼容非可选元素场景：

func findOut<T>(_ type: T.Type) -> T {
    // 处理数组类型：元素为可选类型时返回空数组
    if let arrayType = T.self as? Array<Any>.Type {
        let elementType = arrayType.elementType
        if elementType is OptionalProtocol.Type {
            return [] as! T
        }
        // 非可选元素的数组，可根据需求调整逻辑，这里默认返回空数组
        return [] as! T
    }
    
    // 处理字典类型：值为可选类型时返回空字典
    if let dictType = T.self as? Dictionary<AnyHashable, Any>.Type {
        let valueType = dictType.valueType
        if valueType is OptionalProtocol.Type {
            return [:] as! T
        }
        // 非可选值的字典，同理可调整逻辑
        return [:] as! T
    }
    
    // 其他类型的默认处理，可根据需求改为返回默认值或抛出错误
    fatalError("Unsupported type: \(T.self)")
}

// 辅助协议：统一识别任意可选类型
private protocol OptionalProtocol {
    static var wrappedType: Any.Type { get }
}

extension Optional: OptionalProtocol {
    static var wrappedType: Any.Type { Wrapped.self }
}

关键细节说明 #

• OptionalProtocol辅助协议：通过给Optional扩展实现该协议，无需逐个判断Optional<Wrapped>的具体类型，就能统一识别所有可选类型。
• 数组/字典类型判断：直接通过T.self as? Array<Any>.Type识别数组类型，再通过elementType获取数组元素类型；字典则通过valueType获取值类型，逻辑清晰且类型安全。
• 强制转换安全性：as! T的使用是安全的，因为我们已经通过类型判断确认了当前T的类型是数组或字典。

使用示例 #

// 元素为可选字符串的数组
let emptyOptionalStringArray: [String?] = findOut([String?].self)
print(emptyOptionalStringArray) // 输出：[]

// 嵌套字典数组
let emptyNestedDictArray: [[String: Any]] = findOut([[String: Any]].self)
print(emptyNestedDictArray) // 输出：[]

// 值为可选整数的字典
let emptyOptionalIntDict: [String: Int?] = findOut([String: Int?].self)
print(emptyOptionalIntDict) // 输出：[:]

// 元组数组
let emptyTupleArray: [(String, Int)] = findOut([(String, Int)].self)
print(emptyTupleArray) // 输出：[]

扩展：支持更多容器类型 #

如果需要支持Set等其他容器，只需添加对应的类型判断分支：

if let setType = T.self as? Set<Any>.Type {
    let elementType = setType.elementType
    if elementType is OptionalProtocol.Type {
        return [] as! T
    }
    return [] as! T
}

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