complete role crud

This commit is contained in:
yanghao05
2023-02-04 19:54:29 +08:00
parent 2fb6bba903
commit 285e1f1c51
13 changed files with 270 additions and 19 deletions

47
database/seeders/sys_role.go Executable file
View File

@@ -0,0 +1,47 @@
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 := 10
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 {
return models.SysRole{
UUID: faker.UUID(),
Name: faker.Name(),
ParentID: uint64(faker.IntRange(1, 100)),
DefaultRouter: faker.Name(),
}
}