65 lines
1.2 KiB
Go
Executable File
65 lines
1.2 KiB
Go
Executable File
package seeders
|
|
|
|
import (
|
|
"atom/container"
|
|
"atom/contracts"
|
|
"atom/database/models"
|
|
"log"
|
|
|
|
"github.com/brianvoe/gofakeit/v6"
|
|
"go.uber.org/dig"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func init() {
|
|
if err := container.Container.Provide(NewSysRoleSeeder, dig.Group("seeders")); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
type SysRoleSeeder struct {
|
|
}
|
|
|
|
func NewSysRoleSeeder() contracts.Seeder {
|
|
return &SysRoleSeeder{}
|
|
}
|
|
|
|
func (s *SysRoleSeeder) Run(faker *gofakeit.Faker, db *gorm.DB) {
|
|
times := 50
|
|
for i := 0; i < times; i++ {
|
|
data := s.Generate(faker, i)
|
|
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 *SysRoleSeeder) Generate(faker *gofakeit.Faker, idx int) models.SysRole {
|
|
parentID := 0
|
|
if idx > 10 {
|
|
parentID = faker.IntRange(1, 10)
|
|
}
|
|
|
|
if idx > 20 {
|
|
parentID = faker.IntRange(10, 20)
|
|
}
|
|
|
|
if idx > 30 {
|
|
parentID = faker.IntRange(20, 30)
|
|
}
|
|
|
|
if idx > 40 {
|
|
parentID = faker.IntRange(30, 40)
|
|
}
|
|
|
|
return models.SysRole{
|
|
UUID: faker.UUID(),
|
|
Name: faker.Name(),
|
|
ParentID: uint64(parentID),
|
|
DefaultRouter: faker.Name(),
|
|
}
|
|
}
|