forked from zelo72/mastodon-ios
50 lines
1.3 KiB
Swift
50 lines
1.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()
|
||
|
|
|
||
|
|
os_log("%{public}s[%{public}ld], %{public}s: %s", ((#file as NSString).lastPathComponent), #line, #function, error.localizedDescription)
|
||
|
|
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)))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|