mirror of
https://github.com/mastodon/mastodon-ios.git
synced 2024-12-08 14:46:42 +01:00
138 lines
6.4 KiB
Swift
138 lines
6.4 KiB
Swift
//
|
|
// CoreDataStack.swift
|
|
// CoreDataStack
|
|
//
|
|
// Created by Cirno MainasuK on 2021-1-27.
|
|
//
|
|
|
|
import os
|
|
import Foundation
|
|
import Combine
|
|
@_exported import CoreData
|
|
import MastodonCommon
|
|
|
|
public final class CoreDataStack {
|
|
|
|
static let logger = Logger(subsystem: "CoreDataStack", category: "DB")
|
|
|
|
private(set) var storeDescriptions: [NSPersistentStoreDescription]
|
|
public let didFinishLoad = CurrentValueSubject<Bool, Never>(false)
|
|
|
|
init(persistentStoreDescriptions storeDescriptions: [NSPersistentStoreDescription]) {
|
|
self.storeDescriptions = storeDescriptions
|
|
}
|
|
|
|
public convenience init(databaseName: String = "shared") {
|
|
let storeURL = URL.storeURL(for: AppName.groupID, databaseName: databaseName)
|
|
let storeDescription = NSPersistentStoreDescription(url: storeURL)
|
|
self.init(persistentStoreDescriptions: [storeDescription])
|
|
}
|
|
|
|
public private(set) lazy var persistentContainer: NSPersistentContainer = {
|
|
/*
|
|
The persistent container for the application. This implementation
|
|
creates and returns a container, having loaded the store for the
|
|
application to it. This property is optional since there are legitimate
|
|
error conditions that could cause the creation of the store to fail.
|
|
*/
|
|
let container = CoreDataStack.persistentContainer()
|
|
CoreDataStack.configure(persistentContainer: container, storeDescriptions: storeDescriptions)
|
|
CoreDataStack.load(persistentContainer: container) { [weak self] in
|
|
guard let self = self else { return }
|
|
self.didFinishLoad.value = true
|
|
}
|
|
|
|
return container
|
|
}()
|
|
|
|
static func persistentContainer() -> NSPersistentContainer {
|
|
let bundles = [Bundle.module] // .module required for package in the SwiftPM
|
|
guard let managedObjectModel = NSManagedObjectModel.mergedModel(from: bundles) else {
|
|
fatalError("cannot locate bundles")
|
|
}
|
|
|
|
let container = NSPersistentContainer(name: "CoreDataStack", managedObjectModel: managedObjectModel)
|
|
return container
|
|
}
|
|
|
|
static func configure(persistentContainer container: NSPersistentContainer, storeDescriptions: [NSPersistentStoreDescription]) {
|
|
container.persistentStoreDescriptions = storeDescriptions
|
|
}
|
|
|
|
static func load(persistentContainer container: NSPersistentContainer, callback: @escaping () -> Void) {
|
|
container.loadPersistentStores(completionHandler: { (storeDescription, error) in
|
|
if let error = error as NSError? {
|
|
// Replace this implementation with code to handle the error appropriately.
|
|
// fatalError() causes the application to generate a crash log and terminate. You should not use this function in a shipping application, although it may be useful during development.
|
|
|
|
/*
|
|
Typical reasons for an error here include:
|
|
* The parent directory does not exist, cannot be created, or disallows writing.
|
|
* The persistent store is not accessible, due to permissions or data protection when the device is locked.
|
|
* The device is out of space.
|
|
* The store could not be migrated to the current model version.
|
|
Check the error message to determine what the actual problem was.
|
|
*/
|
|
if let reason = error.userInfo["reason"] as? String,
|
|
(reason == "Can't find mapping model for migration" || reason == "Persistent store migration failed, missing mapping model.") {
|
|
if let storeDescription = container.persistentStoreDescriptions.first, let url = storeDescription.url {
|
|
try? container.persistentStoreCoordinator.destroyPersistentStore(at: url, ofType: NSSQLiteStoreType, options: nil)
|
|
os_log("%{public}s[%{public}ld], %{public}s: cannot migrate model. rebuild database…", ((#file as NSString).lastPathComponent), #line, #function)
|
|
} else {
|
|
assertionFailure()
|
|
}
|
|
}
|
|
|
|
fatalError("Unresolved error \(error), \(error.userInfo)")
|
|
}
|
|
|
|
container.viewContext.mergePolicy = NSMergePolicy.mergeByPropertyObjectTrump
|
|
|
|
// it's looks like the remote notification only trigger when app enter and leave background
|
|
container.viewContext.automaticallyMergesChangesFromParent = true
|
|
|
|
os_log("%{public}s[%{public}ld], %{public}s: %s", ((#file as NSString).lastPathComponent), #line, #function, storeDescription.debugDescription)
|
|
|
|
callback()
|
|
|
|
#if DEBUG
|
|
do {
|
|
let storeURL = URL.storeURL(for: AppName.groupID, databaseName: "shared")
|
|
let data = try Data(contentsOf: storeURL)
|
|
let formatter = ByteCountFormatter()
|
|
formatter.allowedUnits = [.useMB]
|
|
formatter.countStyle = .file
|
|
let size = formatter.string(fromByteCount: Int64(data.count))
|
|
CoreDataStack.logger.log(level: .debug, "\((#file as NSString).lastPathComponent, privacy: .public)[\(#line, privacy: .public)], \(#function, privacy: .public): Database size: \(size)")
|
|
} catch {
|
|
CoreDataStack.logger.log(level: .debug, "\((#file as NSString).lastPathComponent, privacy: .public)[\(#line, privacy: .public)], \(#function, privacy: .public): Cannot get database size")
|
|
}
|
|
#endif
|
|
})
|
|
}
|
|
|
|
}
|
|
|
|
extension CoreDataStack {
|
|
public func newTaskContext() -> NSManagedObjectContext {
|
|
let taskContext = persistentContainer.newBackgroundContext()
|
|
taskContext.mergePolicy = NSMergeByPropertyObjectTrumpMergePolicy
|
|
taskContext.undoManager = nil
|
|
return taskContext
|
|
}
|
|
}
|
|
|
|
extension CoreDataStack {
|
|
|
|
public func rebuild() {
|
|
let oldStoreURL = persistentContainer.persistentStoreCoordinator.url(for: persistentContainer.persistentStoreCoordinator.persistentStores.first!)
|
|
try! persistentContainer.persistentStoreCoordinator.destroyPersistentStore(at: oldStoreURL, ofType: NSSQLiteStoreType, options: nil)
|
|
|
|
CoreDataStack.load(persistentContainer: persistentContainer) { [weak self] in
|
|
guard let self = self else { return }
|
|
self.didFinishLoad.value = true
|
|
}
|
|
}
|
|
|
|
}
|