seeder.tpl 809 B

12345678910111213141516171819202122232425262728293031
  1. package seeders
  2. import (
  3. "fmt"
  4. "github.com/runningwater/gohub/database/factories"
  5. "github.com/runningwater/gohub/pkg/console"
  6. "github.com/runningwater/gohub/pkg/logger"
  7. "github.com/runningwater/gohub/pkg/seed"
  8. "gorm.io/gorm"
  9. )
  10. func init() {
  11. // 添加 Seeder
  12. seed.Add("{{StructNamePlural}}TableSeeder", func(db *gorm.DB) {
  13. // 创建 10 个用户对象
  14. {{VariableNamePlural}} := factories.Make{{StructNamePlural}}(10)
  15. // 批量插入到数据库
  16. result := db.Table("{{TableName}}").Create(&{{VariableNamePlural}})
  17. if err := result.Error; err != nil {
  18. logger.LogIf(err)
  19. return
  20. }
  21. // 打印成功信息
  22. console.Success(fmt.Sprintf("Table [%v] %v rows seeded", result.Statement.Table, result.RowsAffected))
  23. })
  24. }