Swift中如何使用带指定键值类型的引用拷贝类集合？

核心问题解答 #

1. 为什么Swift中无法给NSMutableDictionary指定泛型

Objective-C的泛型属于轻量泛型，仅在编译期做类型检查，运行时没有泛型信息。Swift导入Objective-C的NSMutableDictionary时不会保留泛型约束，因此它在Swift中就是非泛型类型，直接写NSMutableDictionary<String, String>必然会报错。

2. 符合要求的替代方案（类型安全 + 引用语义）

最推荐的是用Swift原生泛型类包装Dictionary，既保留原生字典的所有特性、完全类型安全，又能实现引用语义，完全不需要依赖Objective-C的集合类：

final class RefDictionary<Key: Hashable, Value> {
    private var storage: [Key: Value]
    
    init(_ initial: [Key: Value] = [:]) {
        self.storage = initial
    }
    
    // 实现下标访问，用法和原生Dictionary完全一致
    subscript(key: Key) -> Value? {
        get { storage[key] }
        set { storage[key] = newValue }
    }
    
    // 按需扩展其他Dictionary的方法，比如迭代、count、keys、values等
    var count: Int { storage.count }
    var keys: Dictionary<Key, Value>.Keys { storage.keys }
    var values: Dictionary<Key, Value>.Values { storage.values }
}

使用的时候和原生字典没有差异，因为是类类型，赋值时只会拷贝引用，不会触发整个字典的内容拷贝：

let dict: RefDictionary<String, String> = .init()
dict["name"] = "test"
let copyDict = dict // 仅拷贝引用，没有内容拷贝
copyDict["age"] = "18"
print(dict["age"]) // 输出Optional("18")，符合引用语义的预期

3. 坚持使用NSMutableDictionary的封装方案

如果一定要用NSMutableDictionary，可以把强转逻辑封装在泛型类内部，对外暴露类型安全的接口，避免每次取值都手动强转：

final class TypedNSMutableDictionary<Key: NSCopying, Value> {
    private let storage = NSMutableDictionary()
    
    subscript(key: Key) -> Value? {
        get { storage[key] as? Value }
        set { storage[key] = newValue }
    }
    
    var allKeys: [Key] { storage.allKeys as! [Key] }
    var allValues: [Value] { storage.allValues as! [Value] }
}

关于内存问题的补充说明 #

你测试发现替换为NSMutableDictionary后内存没有明显差异是正常的：Swift原生Dictionary自带*写时复制（COW）*优化，只有当你对拷贝后的实例做修改时，才会触发实际的内容拷贝，单纯传递实例、读取操作不会产生额外内存开销。你遇到的内存问题大概率和值拷贝无关，建议排查其他迁移后的常见问题：

• 闭包循环引用
• Objective-C与Swift类型桥接时产生的不必要中间对象
• 自动释放池的使用和原Objective-C代码逻辑有差异

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