如何实现调用Set的.insert()/.remove(at:)时自动更新UserDefaults？

嘿，这个需求我之前做项目的时候刚好碰到过！每次修改Set后手动序列化保存到UserDefaults确实繁琐，其实我们可以借助Swift的@propertyWrapper特性，封装一套自动同步的逻辑，让你调用insert()、remove()等方法时，自动把修改后的Set同步到UserDefaults里。

核心思路 #

因为Set是值类型，调用它的mutating方法（比如insert、remove）时，本质是生成了一个新的Set实例。我们可以用@propertyWrapper包装这个Set，在它的setter里自动完成序列化和持久化操作，这样每次修改Set都会触发自动保存。

步骤1：确保元素类型支持序列化 #

首先，你的SomeObject必须同时遵循Codable（用于转成Data存储到UserDefaults）和Hashable（Set的元素强制要求）：

struct SomeObject: Codable, Hashable {
    let id: String
    let name: String
    
    // 如果属性都是基础类型，Hashable会自动合成；如果有自定义属性，需要手动实现hash(into:)
}

步骤2：实现自定义@propertyWrapper #

这个包装器会负责从UserDefaults读取、写入Set的逻辑：

@propertyWrapper
struct PersistentSet<Element: Codable & Hashable> {
    private let key: String
    private let defaultValue: Set<Element>
    private let userDefaults: UserDefaults
    
    // 初始化方法，支持自定义UserDefaults实例（比如App Group共享数据）
    init(key: String, defaultValue: Set<Element>, userDefaults: UserDefaults = .standard) {
        self.key = key
        self.defaultValue = defaultValue
        self.userDefaults = userDefaults
    }
    
    var wrappedValue: Set<Element> {
        get {
            // 从UserDefaults解码Set
            guard let data = userDefaults.data(forKey: key) else {
                return defaultValue
            }
            do {
                return try JSONDecoder().decode(Set<Element>.self, from: data)
            } catch {
                print("解码Set失败：\(error)")
                return defaultValue
            }
        }
        set {
            // 将新的Set编码后写入UserDefaults
            do {
                let data = try JSONEncoder().encode(newValue)
                userDefaults.set(data, forKey: key)
            } catch {
                print("编码Set失败：\(error)")
            }
        }
    }
}

步骤3：使用自动持久化的Set #

现在你只需要用@PersistentSet标记你的Set变量，之后所有修改操作都会自动同步到UserDefaults：

class MyViewModel {
    // 定义持久化的Set，指定存储key和默认值
    @PersistentSet(key: "SavedSomeObjects", defaultValue: []) var mySet: Set<SomeObject>
    
    func addNewObject(_ obj: SomeObject) {
        // 直接调用insert，自动同步到UserDefaults
        mySet.insert(obj)
    }
    
    func deleteObject(_ obj: SomeObject) {
        // 直接调用remove，同样自动同步
        mySet.remove(obj)
    }
    
    func clearAll() {
        // 直接清空Set，也会自动保存
        mySet.removeAll()
    }
}

注意事项 #

• Hashable稳定性：确保SomeObject的hash(into:)实现稳定，否则可能导致Set序列化/反序列化后元素丢失或重复。
• 错误处理：示例中用print处理错误，实际项目可以根据需求改成抛出错误、记录日志或降级处理。
• 自定义UserDefaults：如果需要在App Group间共享数据，可以在初始化PersistentSet时传入自定义的UserDefaults实例（比如UserDefaults(suiteName: "your.group.id")）。

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