GRDB.swift中saved()方法未返回更新记录的解决方案

问题描述

在使用GRDB.swift库时，开发者可能会遇到saved()方法未能正确返回包含更新ID的记录的情况。具体表现为：当调用saved()方法保存一个记录后，返回的记录中ID字段仍然为nil，而实际上数据库已经成功插入并生成了ID。

问题分析

这个问题源于GRDB.swift的设计机制。GRDB.swift是一个Swift语言的SQLite工具库，它提供了两种保存记录的方式：

1. save()方法：这是一个mutating方法，会直接修改原记录
2. saved()方法：这是一个非mutating方法，返回修改后的新记录

关键在于，GRDB.swift不会自动更新记录的ID字段。无论是使用save()还是saved()，都需要开发者自行实现didInsert方法来处理插入成功后的ID赋值。

解决方案

要让saved()方法正确返回包含ID的记录，需要在记录类型中实现didInsert方法。以下是一个完整的示例：

struct Player: Codable, MutablePersistableRecord {
    var id: Int64?
    var name: String
    
    // 必须实现这个方法来处理插入后的ID赋值
    mutating func didInsert(_ inserted: InsertionSuccess) {
        id = inserted.rowID
    }
}

// 使用示例
let dbQueue = try DatabaseQueue()
try dbQueue.write { db in
    try db.create(table: "player") { t in
        t.autoIncrementedPrimaryKey("id")
        t.column("name", .text)
    }
    
    let alice = Player(id: nil, name: "Alice")
    let savedAlice = try alice.saved(db)
    
    // savedAlice现在包含正确的ID
    print(savedAlice.id) // 不为nil
}

两种保存方式的区别

1. save()方法：
   • 会直接修改原记录
   • 需要在可变(var)变量上调用
   • 适用于需要就地更新的场景

var bob = Player(id: nil, name: "Bob")
try bob.save(db)
// bob现在包含ID

2. saved()方法：
   • 返回修改后的新记录
   • 原记录保持不变
   • 适用于需要保留原记录的场景

let alice = Player(id: nil, name: "Alice")
let savedAlice = try alice.saved(db)
// alice保持原样，savedAlice包含ID

最佳实践

1. 始终为需要自增ID的记录实现didInsert方法
2. 根据场景选择合适的保存方法：
   • 需要继续使用更新后记录时用save()
   • 需要保留原记录时用saved()
3. 确保数据库表设置了自增主键

通过正确实现didInsert方法，可以确保GRDB.swift的保存操作能够正确更新记录的ID字段，无论是使用save()还是saved()方法。