一起养成写作习惯！这是我参与「掘金日新计划 · 4 月更文挑战」的第22天，点击查看活动详情。
今日职言：很多时候，失败并非自身能力不足，而是负面情绪太多。情绪稳定，是一个人最好的修养。
承接上一章的内容，我们继续完成使用CoreData框架搭建一个简单的ToDo待办事项App。
这一章节，我们正式进入使用CoreData框架实现数据持久化。
之前我们是通过创建项目的时候勾选使用CoreData框架，系统给我们创建了需要的文件和模型。那么这一章节，我们尝试使用CoreData框架达到数据持久化的目的。
Persistence.swift文件
Persistence.swift文件是数据保存到持久存储区的文件，通过将管理对象上下文注入到环境中，来实现在任何视图都可以检索上下文，并且能够管理数据。
我们删除多余的示例数据代码。
import Foundation
import CoreData
struct PersistenceController {
static let shared = PersistenceController()
let container: NSPersistentContainer
init(inMemory: Bool = false) {
container = NSPersistentContainer(name: "SwiftUICoreData")
if inMemory { container.persistentStoreDescriptions.first!.url = URL(fileURLWithPath: "/dev/null")
}
container.loadPersistentStores(completionHandler: { storeDescription, error in
if let error = error as NSError? {
fatalError("Unresolved error \(error), \(error.userInfo)")
}
})
}
}
SwiftUICoreDataApp.swift文件
我们回到SwiftUICoreDataApp.swift文件，我们需要将托管对象上下文注入到环境中，这样我们就能够方便地在内容视图中访问上下文，并且管理数据库中的数据。
import SwiftUI
@main
struct SwiftUICoreDataApp: App {
let persistenceController = PersistenceController.shared
var body: some Scene {
WindowGroup {
ContentView()
.environment(\.managedObjectContext, persistenceController.container.viewContext)
}
}
}
SwiftUICoreData.xcdatamodeld模型
模型创建完成后，我们需要创建一个新的实体ToDoItem来存储我们需要用到的参数，我们将原来的实体Item重新命名为ToDoItem。
在ToDoItem实体中，我们需要定义好项目需要的属性。
id:UUID
name:String
priorityNum:Integer32
isCompleted:Boolean
由于priority优先级属性是一个Enum枚举类型，为了将Enum枚举类型保存到数据库中，我们必须存储它的原始值，即Int整数，因此需要使用Integer 32类型，而且为了避免命名冲突，我们将属性命名为priorityNum。
这里再科普一个知识点。
由于ToDoItem实体我们重新定义了，那么要保证Module模块要选择CurrrentProductModule当前产品的模型，Codegen代码基因要选择Manual/None，不然我们在项目中引用模型的时候可能会找不到我们定义的ToDoItem实体，这非常重要！非常重要！非常重要！重要的事情说三遍。
CoreData模型类
在CoreData框架中，每个实体都与一个模型类配对，我们需要在ToDoItem.swift文件定义模型类。
这里CoreData的模型类继承自NSManagedObject协议，每个属性都使用@NSManaged进行注释，并且对应我们ToDoItem创建的属性id、name、priorityNum、isCompleted。
import Foundation
import CoreData
enum Priority: Int {
case low = 0
case normal = 1
case high = 2
}
public class ToDoItem: NSManagedObject {
@NSManaged public var id: UUID
@NSManaged public var name: String
@NSManaged public var priorityNum: Int32
@NSManaged public var isCompleted: Bool
}
extension ToDoItem: Identifiable {
var priority: Priority {
get {
return Priority(rawValue: Int(priorityNum)) ?? .normal
}
set {
self.priorityNum = Int32(newValue.rawValue)
}
}
}
ContentView.swift文件
定义好CoreData模型类后，我们需要在ContentView主视图中使用它，我们使用@FetchRequest属性包装器从数据库加载数据，我们替换之前使用@State标记的ToDoItem数组数据来源。
@FetchRequest( entity: ToDoItem.entity(),
sortDescriptors: [ NSSortDescriptor(keyPath: \ToDoItem.priorityNum, ascending: false) ])
var todoItems: FetchedResults<ToDoItem>
我们在环境中注入了托管对象上下文，获取所需的数据。
由于我们的List列表数据来源于新定义的todoItems，我们把整个ToDoListView的body内容拆回到ContentView主视图，这样我们就可以直接遍历循环todoItems的内容。
List {
ForEach(todoItems) { todoItem in
ToDoListRow(todoItem: todoItem)
}
NewToDoView.swift文件
紧接着，我们需要同步更新NewToDoView.swift的代码。要将一个新任务保存到数据库中，首先需要从环境中获取管理对象上下文。
@Environment(\.managedObjectContext) var context
然后，我们在NewToDoView视图和saveButton保存按钮视图就不需要再@Binding绑定ToDoItem数组了。
@Binding var todoItems: [ToDoItem]
然后，我们再更新下addTask添加新事项的方法。
//添加新事项方法
private func addTask(name: String, priority: Priority, isCompleted: Bool = false) {
let task = ToDoItem(context: context)
task.id = UUID()
task.name = name
task.priority = priority
task.isCompleted = isCompleted
do {
try context.save()
} catch {
print(error)
}
}
我们要将新事项插入数据库中，需要使用托管上下文创建ToDoItem，然后调用上下文的save函数来提交更改。
因为我们删除了todoItems绑定，所以还需要调整NewToDoView_Previews预览视图。
NewToDoView(name: "", priority: .normal, showNewTask: .constant(true))
由于我们NewToDoView新建事项视图发现变化，我们回到ContentView首页视图，重新修改下绑定关系。
NewToDoView(name: "", priority: .normal, showNewTask: $showNewTask)
同样，我们在ContentView首页视图也需要从环境中获取管理对象上下文。
@Environment(\.managedObjectContext) var context
ToDoListRow视图
与添加新事项类似，我们需要获取用于记录更新的托管对象上下文。对于Toggle优先级选择视图，每当切换发生更改时，todoItem的isCompleted属性将被更新。我们可以附加onReceive修饰符，onReceive修饰符可以监听isCompleted属性和其他我们定义好的属性的更改，并通过调用上下文的save函数将它们保存到持久存储区。
//监听todoItem数组参数变化并保存
.onReceive(todoItem.objectWillChange, perform: { _ in
if self.context.hasChanges {
try? self.context.save()
}
})
deleteTask删除事项方法
上面我们既然完成了addTask新增事项方法，那么顺便完成deleteTask删除事项的交互。
//删除事项方法
private func deleteTask(indexSet: IndexSet) {
for index in indexSet {
let itemToDelete = todoItems[index]
context.delete(itemToDelete)
}
DispatchQueue.main.async {
do {
try context.save()
} catch {
print(error)
}
}
}
deleteTask删除事项方法接收一个存储要删除的项的索引集，我们只需要调用上下文的delete函数，并指定要删除的项，然后调用save函数来提交更新。
我们把这个方法加到List里，就可以实现滑动删除的操作。
List {
ForEach(todoItems) { todoItem in
ToDoListRow(todoItem: todoItem)
}.onDelete(perform: deleteTask)
}
preview预览效果
我们运行模拟器的时候发现报错了，这是因为我们没有在contentview_preview结构体中注入托管对象上下文。我们需要创建一个内存中的数据存储，并准备一些测试数据。
我们在Persistence.swift文件创建了一个PersistenceController实例，并将inMemory参数设置为true，然后我们添加10个示例待办事项，并将它们保存到数据存储中。
import CoreData
struct PersistenceController {
static let shared = PersistenceController()
let container: NSPersistentContainer
static var preview: PersistenceController = {
let result = PersistenceController(inMemory: true)
let viewContext = result.container.viewContext
for index in 0..<10 {
let newItem = ToDoItem(context: viewContext)
newItem.id = UUID()
newItem.name = "待办事项\(index)"
newItem.priority = .normal
newItem.isCompleted = false
}
do {
try viewContext.save()
} catch {
let nsError = error as NSError
fatalError("Unresolved error \(nsError), \(nsError.userInfo)")
}
return result
}()
init(inMemory: Bool = false) {
container = NSPersistentContainer(name: "SwiftUICoreData")
if inMemory {
container.persistentStoreDescriptions.first!.url = URL(fileURLWithPath: "/dev/null")
}
container.loadPersistentStores(completionHandler: { (storeDescription, error) in
if let error = error as NSError? {
fatalError("Unresolved error \(error), \(error.userInfo)")
}
})
}
}
之后我们如果只需要在ContentView_Previews预览视图的上下文注入到内容视图的环境中，就可以看到示例的数据啦～
struct ContentView_Previews: PreviewProvider {
static var previews: some View {
ContentView().environment(\.managedObjectContext, PersistenceController.preview.container.viewContext)
}
}
快来动手试试吧！
如果本专栏对你有帮助，不妨点赞、评论、关注～
