add seeder demo
This commit is contained in:
@@ -29,7 +29,6 @@ var seedCmd = &cobra.Command{
|
|||||||
}
|
}
|
||||||
|
|
||||||
for _, seeder := range seedersContainer.Seeders {
|
for _, seeder := range seedersContainer.Seeders {
|
||||||
log.Printf("seed table (%s) for %d times", seeder.Table(), seeder.Times())
|
|
||||||
seeder.Run(seedersContainer.DB)
|
seeder.Run(seedersContainer.DB)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
|
|||||||
@@ -4,7 +4,5 @@ import "gorm.io/gorm"
|
|||||||
|
|
||||||
// Migration route interface
|
// Migration route interface
|
||||||
type Seeder interface {
|
type Seeder interface {
|
||||||
Times() uint
|
|
||||||
Table() string
|
|
||||||
Run(*gorm.DB)
|
Run(*gorm.DB)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -33,23 +33,20 @@ type User struct {
|
|||||||
Status bool
|
Status bool
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *UserSeeder) Times() uint {
|
|
||||||
return 10
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *UserSeeder) Run(db *gorm.DB) {
|
func (s *UserSeeder) Run(db *gorm.DB) {
|
||||||
var i uint
|
times := 10
|
||||||
for i = 0; i < s.Times(); i++ {
|
for i := 0; i < times; i++ {
|
||||||
data := s.Generate(i)
|
data := s.Generate(i)
|
||||||
db.Table(s.Table()).Create(&data)
|
if i == 0 {
|
||||||
|
stmt := &gorm.Statement{DB: db}
|
||||||
|
_ = stmt.Parse(&data)
|
||||||
|
log.Printf("seeding %s for %d times", stmt.Schema.Table, times)
|
||||||
|
}
|
||||||
|
db.Create(&data)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *UserSeeder) Table() string {
|
func (s *UserSeeder) Generate(idx int) User {
|
||||||
return "users"
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *UserSeeder) Generate(idx uint) User {
|
|
||||||
return User{
|
return User{
|
||||||
Username: fmt.Sprintf("username: %d", idx),
|
Username: fmt.Sprintf("username: %d", idx),
|
||||||
Password: fmt.Sprintf("password: %d", idx),
|
Password: fmt.Sprintf("password: %d", idx),
|
||||||
|
|||||||
Reference in New Issue
Block a user