🎨用户接口处理完毕

script/db_migrate.go

@@ -1,10 +1,11 @@
 package script
 
 import (
-	"go.uber.org/zap"
+	"gitee.ltd/lxh/logger/log"
 	"wireguard-dashboard/client"
 	"wireguard-dashboard/constant"
 	"wireguard-dashboard/model/entity"
+	"wireguard-dashboard/repository"
 	"wireguard-dashboard/utils"
 )
 
@@ -51,27 +52,28 @@ func (s Script) CreateSuperAdmin() error {
 	}
 
 	// 没有超管就创建一个
-	if count <= 0 {
-		avatarPath, err := utils.Avatar().GenerateAvatar()
-		if err != nil {
-			zap.S().Errorf("生成头像失败: %v", err.Error())
-			return err
-		}
-
-		if err = client.DB.Create(&entity.User{
-			Avatar:   avatarPath,
-			Name:     "超牛管理员",
-			Account:  "Admin",
-			Email:    "",
-			Password: utils.Password().GenerateHashPassword("admin123"),
-			IsAdmin:  constant.SuperAdmin,
-			Status:   constant.Normal,
-		}).Error; err != nil {
-			zap.S().Errorf("创建管理员失败: %v", err.Error())
-		}
-
+	if count > 0 {
 		return nil
 	}
 
+	// 生成一下头像
+	avatarPath, err := utils.Avatar().GenerateAvatar()
+	if err != nil {
+		log.Errorf("生成头像失败: %v", err.Error())
+		return err
+	}
+
+	if err = repository.User().Save(&entity.User{
+		Avatar:   avatarPath,
+		Name:     "超牛管理员",
+		Account:  "Admin",
+		Email:    "",
+		Password: utils.Password().GenerateHashPassword("admin123"),
+		IsAdmin:  constant.SuperAdmin,
+		Status:   constant.Normal,
+	}); err != nil {
+		return err
+	}
+
 	return nil
 }