如何使用GORM为PostgreSQL导入初始化种子数据

可行实现方案 #

原生GORM实现（无第三方依赖，匹配你的目录结构） #

这个实现修复了你之前参考教程中返回值缺失的问题，逻辑简单可控，完全适配你用的GORM v1版本。

1. 编写种子执行主逻辑

文件路径：pkg/seed/seed.go

package seed

import (
	"github.com/jinzhu/gorm"
	"替换成你自己的项目路径/pkg/seeds"
)

// SeedFunc 统一种子函数签名，所有种子方法都要遵循这个格式
type SeedFunc func(db *gorm.DB) error

// 注册所有种子，注意顺序：有外键依赖的数据要先放被依赖项（比如先写用户，再写关联的球员数据）
var seeds = []SeedFunc{
	seeds.UserSeed,
	seeds.PlayerSeed,
}

// Execute 执行全部已注册的种子数据
func Execute(db *gorm.DB) error {
	for _, s := range seeds {
		if err := s(db); err != nil {
			return err
		}
	}
	return nil
}

2. 编写具体业务种子文件

首先是用户种子，路径：pkg/seeds/user_seed.go

package seeds

import (
	"github.com/jinzhu/gorm"
	"替换成你自己项目的models包导入路径"
)

func UserSeed(db *gorm.DB) error {
	// 先判断数据是否已存在，避免重复导入
	var existCount int
	db.Model(&models.User{}).Where("email = ?", "admin@test.com").Count(&existCount)
	if existCount > 0 {
		return nil
	}

	// 写入种子数据
	users := []models.User{
		{
			Email:    "admin@test.com",
			Password: "替换成你加密后的管理员密码",
			UserType: "admin",
		},
		{
			Email:    "normal@test.com",
			Password: "替换成加密后的普通用户密码",
			UserType: "normal",
		},
	}
	return db.Create(&users).Error
}

然后是球员种子，路径：pkg/seeds/player_seed.go

package seeds

import (
	"github.com/jinzhu/gorm"
	"替换成你自己项目的models包导入路径"
)

func PlayerSeed(db *gorm.DB) error {
	var existCount int
	db.Model(&models.Player{}).Count(&existCount)
	if existCount > 0 {
		return nil
	}

	// 查询关联的用户ID，绑定外键
	var adminUser models.User
	db.Where("email = ?", "admin@test.com").First(&adminUser)

	players := []models.Player{
		{
			FirstName:       "Test",
			LastName:        "Player",
			Gender:          "male",
			PrimaryPosition: "SG",
			Email:           "player@test.com",
			UserID:          int(adminUser.ID),
			Team:            "TestTeam",
		},
	}
	return db.Create(&players).Error
}

3. 调用执行

在你项目数据库初始化完成的位置调用执行即可，比如在main.go中：

// db是你初始化完成的*gorm.DB实例
if err := seed.Execute(db); err != nil {
	panic("种子数据导入失败: " + err.Error())
}

如果需要单独触发种子导入，可以单独建一个cmd/seed/main.go入口，初始化db后调用seed.Execute就行，执行命令为go run cmd/seed/main.go。 #

注意事项 #

1. 你之前跑教程失败的核心原因就是种子函数没有按签名返回error类型值，只要每个种子函数在执行出错时返回对应error、执行成功时返回nil就不会有语法问题。
2. 你现有User模型的GORM标签有笔误：gorm:"typevarchar(100);unique_index;not null"少了冒号，要改成gorm:"type:varchar(100);unique_index;not null"，否则创建表时字段类型会不符合预期。
3. 如果需要更复杂的种子管理功能（比如指定执行单个种子、种子回滚、分环境种子），可以使用成熟的开源种子包，你当前的简单场景用上面的原生实现足够，没有额外依赖，排查问题更方便。

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