85 lines
2.3 KiB
Swift
85 lines
2.3 KiB
Swift
//
|
|
// NSManagedObjectContext.swift
|
|
// CoreDataStack
|
|
//
|
|
// Created by Cirno MainasuK on 2020-8-10.
|
|
//
|
|
|
|
import os
|
|
import Foundation
|
|
import Combine
|
|
import CoreData
|
|
|
|
extension NSManagedObjectContext {
|
|
public func insert<T: NSManagedObject>() -> T where T: Managed {
|
|
guard let object = NSEntityDescription.insertNewObject(forEntityName: T.entityName, into: self) as? T else {
|
|
fatalError("cannot insert object: \(T.self)")
|
|
}
|
|
|
|
return object
|
|
}
|
|
|
|
public func saveOrRollback() throws {
|
|
do {
|
|
guard hasChanges else {
|
|
return
|
|
}
|
|
try save()
|
|
} catch {
|
|
rollback()
|
|
|
|
throw error
|
|
}
|
|
}
|
|
|
|
public func performChanges(block: @escaping () -> Void) -> Future<Result<Void, Error>, Never> {
|
|
Future { promise in
|
|
self.perform {
|
|
block()
|
|
do {
|
|
try self.saveOrRollback()
|
|
promise(.success(Result.success(())))
|
|
} catch {
|
|
promise(.success(Result.failure(error)))
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
extension NSManagedObjectContext {
|
|
public func perform<T>(block: @escaping () throws -> T) async throws -> T {
|
|
return try await perform(schedule: .enqueued) {
|
|
try block()
|
|
}
|
|
}
|
|
|
|
public func performChanges<T>(block: @escaping () throws -> T) async throws -> T {
|
|
return try await perform(schedule: .enqueued) {
|
|
let value = try block()
|
|
try self.saveOrRollback()
|
|
return value
|
|
}
|
|
} // end func
|
|
}
|
|
|
|
extension NSManagedObjectContext {
|
|
static let objectCacheKey = "ObjectCacheKey"
|
|
private typealias ObjectCache = [String: NSManagedObject]
|
|
|
|
public func cache(
|
|
_ object: NSManagedObject?,
|
|
key: String
|
|
) {
|
|
var cache = userInfo[NSManagedObjectContext.objectCacheKey] as? ObjectCache ?? [:]
|
|
cache[key] = object
|
|
userInfo[NSManagedObjectContext.objectCacheKey] = cache
|
|
}
|
|
|
|
public func cache(froKey key: String) -> NSManagedObject? {
|
|
guard let cache = userInfo[NSManagedObjectContext.objectCacheKey] as? ObjectCache
|
|
else { return nil }
|
|
return cache[key]
|
|
}
|
|
}
|