GORM执行AutoMigrate首次建表时如何插入种子数据

GORM 本身未在 AutoMigrate 方法中直接暴露表操作状态的公开字段，你可以通过以下两种方式实现首次建表插入种子数据的需求：

方法1：前置判断表存在状态（推荐） #

GORM 内置的 Migrator 接口提供了 HasTable 方法，可以在执行迁移前先判断对应模型的表是否已存在，再根据判断结果处理种子数据，逻辑简单直观，适合绝大多数场景：

// 先判断User表是否已存在
hasUserTable := db.Migrator().HasTable(&User{})

// 执行正常的表结构迁移
if err := db.AutoMigrate(&User{}); err != nil {
    // 处理迁移错误逻辑
    panic(err)
}

// 表之前不存在，说明是本次迁移首次创建，插入种子数据
if !hasUserTable {
    // 此处写你的种子数据插入逻辑
    db.Create(&User{Username: "admin", Password: "加密后的默认密码"})
}

方法2：监听建表回调（适合多表批量处理场景） #

如果你有多个表都需要在首次创建时插入对应种子数据，可以注册 GORM 全局的建表回调，统一管理所有表的种子插入逻辑，避免重复写判断代码：

// 全局注册建表后回调，仅需在初始化DB时执行一次
db.Callback().CreateTable().After("gorm:create_table").Register("insert_seed_after_create", func(db *gorm.DB) {
    switch db.Statement.Schema.Table {
    case "users": // 匹配User表名
        db.Create(&User{Username: "admin", Password: "加密后的默认密码"})
    case "roles": // 可扩展匹配其他表
        db.Create(&Role{Name: "超级管理员", Perm: "all"})
    }
})

// 之后正常执行AutoMigrate即可，表首次创建时会自动触发对应种子插入逻辑
if err := db.AutoMigrate(&User{}, &Role{}); err != nil {
    panic(err)
}

内容的提问来源于stack exchange，提问作者Arif