gohub/pkg/seed/seeder.go

// Package seed 存放数据库填充数据
package seed

import "gorm.io/gorm"

// 存放所有 Seeder
var seedrs []Seeder

// 按排序执行的 Seeder 数组
var orderedSeederNames []string

type SeederFunc func(*gorm.DB)

// Seeder 对应每一个 database/seeders 目录下的 Seeder 文件
type Seeder struct {
	Name string
	Func SeederFunc
}

// AddSeeder 添加一个 Seeder
func Add(name string, fn SeederFunc) {
	seedrs = append(seedrs, Seeder{
		Name: name,
		Func: fn,
	})
}

// SetRunOrder 设置 按顺序执行的 Seeder 数组
func SetRunOrder(names []string) {
	orderedSeederNames = names
}