Swift中适配CoreData perform方法的闭包传参替代方案

核心问题是NSManagedObjectContext的perform方法要求闭包签名为() throws -> T，但我们希望包装器的read方法能接收带NSManagedObjectContext参数的闭包(NSManagedObjectContext) throws -> T，这里有两种简洁的实现方案，且都能保证并发安全：

方案一：在perform闭包内直接传递MOC #

直接在perform的无参闭包里，把当前的viewContext传给用户定义的带参闭包，完成签名适配：

class Persistence {
    let container: NSPersistentContainer
    
    init(container: NSPersistentContainer) {
        self.container = container
    }
    
    func read<T>(_ block: @escaping (NSManagedObjectContext) throws -> T) async rethrows -> T {
        return try await container.viewContext.perform {
            // 直接把当前viewContext传入用户闭包
            try block(self.container.viewContext)
        }
    }
}

这样调用时就能按预期传入MOC：

try await persistence.read { moc in
    let repo = Repository(context: moc)
    // 执行CoreData读取操作
}

方案二：给NSManagedObjectContext扩展带参perform重载 #

如果需要在多个地方复用这种带参闭包的调用方式，可以给NSManagedObjectContext扩展一个自定义的perform方法，封装签名转换逻辑：

extension NSManagedObjectContext {
    func perform<T>(_ block: @escaping (NSManagedObjectContext) throws -> T) async rethrows -> T {
        return try await self.perform {
            try block(self)
        }
    }
}

// 此时Persistence类的read方法可以简化为：
class Persistence {
    let container: NSPersistentContainer
    
    init(container: NSPersistentContainer) {
        self.container = container
    }
    
    func read<T>(_ block: @escaping (NSManagedObjectContext) throws -> T) async rethrows -> T {
        return try await container.viewContext.perform(block)
    }
}

并发安全说明 #

这两种方案都基于CoreData原生的perform方法，该方法会自动将闭包调度到与当前MOC关联的串行队列上执行，完全符合CoreData的并发访问规则，不存在多线程直接操作MOC的风险，是安全的实现方式。

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