:art:服务端的初始化

repository/user.go

@@ -1,6 +1,7 @@
 package repository
 
 import (
+	"gorm.io/gorm"
 	"wireguard-dashboard/client"
 	"wireguard-dashboard/http/param"
 	"wireguard-dashboard/model/entity"
@@ -8,10 +9,14 @@ import (
 	"wireguard-dashboard/utils"
 )
 
-type user struct{}
+type user struct {
+	*gorm.DB
+}
 
 func User() user {
-	return user{}
+	return user{
+		client.DB,
+	}
 }
 
 // List
@@ -22,7 +27,7 @@ func User() user {
 // @return total
 // @return err
 func (r user) List(p param.UserList) (data []vo.User, total int64, err error) {
-	err = client.DB.Model(&entity.User{}).Scopes(utils.Page(p.Current, p.Size)).
+	err = r.Model(&entity.User{}).Scopes(utils.Page(p.Current, p.Size)).
 		Select("id", "created_at", "updated_at", "avatar", "name", "account", "is_admin", "status").Order("created_at DESC").
 		Find(&data).Offset(-1).Limit(-1).Count(&total).Error
 
@@ -36,7 +41,7 @@ func (r user) List(p param.UserList) (data []vo.User, total int64, err error) {
 // @return *entity.User
 // @return error
 func (r user) GetUserById(id string) (data *entity.User, err error) {
-	err = client.DB.Where("id = ?", id).First(&data).Error
+	err = r.Where("id = ?", id).First(&data).Error
 	return
 }
 
@@ -47,7 +52,7 @@ func (r user) GetUserById(id string) (data *entity.User, err error) {
 // @return data
 // @return err
 func (r user) GetUserByAccount(account string) (data *entity.User, err error) {
-	err = client.DB.Where("account = ?", account).First(&data).Error
+	err = r.Where("account = ?", account).First(&data).Error
 	return
 }
 
@@ -67,7 +72,7 @@ func (r user) Save(ent *entity.User) (err error) {
 			"status":   ent.Status,
 		}
 
-		return client.DB.Model(&entity.User{}).Where("id = ?", ent.Id).Updates(&updates).Error
+		return r.Model(&entity.User{}).Where("id = ?", ent.Id).Updates(&updates).Error
 	}
 
 	defaultPassword := utils.Password().GenerateHashPassword("admin123")
@@ -81,7 +86,7 @@ func (r user) Save(ent *entity.User) (err error) {
 	}
 
 	// 创建
-	return client.DB.Create(&ent).Error
+	return r.Create(&ent).Error
 }
 
 // ChangePassword
@@ -92,7 +97,7 @@ func (r user) Save(ent *entity.User) (err error) {
 // @return err
 func (r user) ChangePassword(p param.ChangePassword, userId string) (err error) {
 	password := utils.Password().GenerateHashPassword(p.NewPassword)
-	return client.DB.Model(&entity.User{}).Where("id = ?", userId).Update("password", password).Error
+	return r.Model(&entity.User{}).Where("id = ?", userId).Update("password", password).Error
 }
 
 // ChangeUserState
@@ -101,5 +106,5 @@ func (r user) ChangePassword(p param.ChangePassword, userId string) (err error)
 // @param p
 // @return err
 func (r user) ChangeUserState(p param.ChangeUserState) (err error) {
-	return client.DB.Model(&entity.User{}).Where("id = ?", p.ID).Update("status", p.Status).Error
+	return r.Model(&entity.User{}).Where("id = ?", p.ID).Update("status", p.Status).Error
 }