update

internal/erpserver/repository/seed/seed.go

@@ -2,43 +2,63 @@ package seed
 
 import (
 	"context"
+	"time"
 
 	"management/internal/erpserver/model/system"
 )
 
-func Init(
+type Seed struct {
+	configRepository     system.ConfigRepository
+	departmentRepository system.DepartmentRepository
+	roleRepository       system.RoleRepository
+	userRepository       system.UserRepository
+	menuRepository       system.MenuRepository
+}
+
+func NewSeed(
 	configRepository system.ConfigRepository,
 	departmentRepository system.DepartmentRepository,
 	roleRepository system.RoleRepository,
 	userRepository system.UserRepository,
 	menuRepository system.MenuRepository,
-) error {
-	ctx := context.Background()
+) *Seed {
+	return &Seed{
+		configRepository:     configRepository,
+		departmentRepository: departmentRepository,
+		roleRepository:       roleRepository,
+		userRepository:       userRepository,
+		menuRepository:       menuRepository,
+	}
+}
+
+func (s *Seed) Run() error {
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
 
 	// 后台pear配置
-	if err := configRepository.Initialize(ctx); err != nil {
+	if err := s.configRepository.Initialize(ctx); err != nil {
 		return err
 	}
 
 	// 部门
-	err := departmentRepository.Initialize(ctx)
+	err := s.departmentRepository.Initialize(ctx)
 	if err != nil {
 		return err
 	}
 
 	// 角色
-	role, err := roleRepository.Initialize(ctx)
+	role, err := s.roleRepository.Initialize(ctx)
 	if err != nil {
 		return err
 	}
 
 	// 用户
-	if err := userRepository.Initialize(ctx, 0, role.ID); err != nil {
+	if err := s.userRepository.Initialize(ctx, 0, role.ID); err != nil {
 		return err
 	}
 
 	// 菜单
-	if err = menuRepository.Initialize(ctx); err != nil {
+	if err = s.menuRepository.Initialize(ctx); err != nil {
 		return err
 	}