在 GORM 中首次迁移时插入种子数据

Question

我想在数据库中创建表时插入种子数据。AutoMigrate当我执行 时，它不会返回与表创建相关的任何信息，因此我无法确认表是否已创建、更新或未执行任何操作。db.AutoMigrate(&User{})有没有办法从GORM知道表创建信息以插入种子数据？这样我就可以插入种子数据，例如：if err = db.AutoMigrate(&User{}); err != nil {    if db.CreatedFirstTime {        //Insert seed data    }}

Answer 1

根据文档，您无法从 中获取表创建信息。尝试使用与查询组合来获取表的信息。db.AutoMigrate(&User{})Migrator

例如：

if err = db.AutoMigrate(&User{}); err == nil && db.Migrator().HasTable(&User{}) {

    if err := db.First(&User{}).Error; errors.Is(err, gorm.ErrRecordNotFound) {

        //Insert seed data

    }

}