acepanel/panel
2
0
Fork 0
mirror of https://github.com/acepanel/panel.git synced 2026-02-04 09:13:49 +08:00
Code Issues Packages Releases Activity

refactor: 应用支持依赖注入

Browse Source
This commit is contained in:
耗子
2024-12-16 02:01:31 +08:00
parent 1fdc86cc0a
commit a64660d0cb
57 changed files with 663 additions and 861 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
internal/data/database_user.go
View File

@@ -4,23 +4,28 @@ import (
"fmt"
"slices"
"github.com/TheTNB/panel/internal/app"
"gorm.io/gorm"
"github.com/TheTNB/panel/internal/biz"
"github.com/TheTNB/panel/internal/http/request"
"github.com/TheTNB/panel/pkg/db"
)
type databaseUserRepo struct {
db *gorm.DB
server biz.DatabaseServerRepo
}
func NewDatabaseUserRepo(server biz.DatabaseServerRepo) biz.DatabaseUserRepo {
return &databaseUserRepo{server: server}
func NewDatabaseUserRepo(db *gorm.DB, server biz.DatabaseServerRepo) biz.DatabaseUserRepo {
return &databaseUserRepo{
db: db,
server: server,
}
}
func (r databaseUserRepo) Count() (int64, error) {
var count int64
if err := app.Orm.Model(&biz.DatabaseUser{}).Count(&count).Error; err != nil {
if err := r.db.Model(&biz.DatabaseUser{}).Count(&count).Error; err != nil {
return 0, err
}
@@ -30,7 +35,7 @@ func (r databaseUserRepo) Count() (int64, error) {
func (r databaseUserRepo) List(page, limit uint) ([]*biz.DatabaseUser, int64, error) {
var user []*biz.DatabaseUser
var total int64
err := app.Orm.Model(&biz.DatabaseUser{}).Preload("Server").Order("id desc").Count(&total).Offset(int((page - 1) * limit)).Limit(int(limit)).Find(&user).Error
err := r.db.Model(&biz.DatabaseUser{}).Preload("Server").Order("id desc").Count(&total).Offset(int((page - 1) * limit)).Limit(int(limit)).Find(&user).Error
for u := range slices.Values(user) {
r.fillUser(u)
@@ -41,7 +46,7 @@ func (r databaseUserRepo) List(page, limit uint) ([]*biz.DatabaseUser, int64, er
func (r databaseUserRepo) Get(id uint) (*biz.DatabaseUser, error) {
user := new(biz.DatabaseUser)
if err := app.Orm.Preload("Server").Where("id = ?", id).First(user).Error; err != nil {
if err := r.db.Preload("Server").Where("id = ?", id).First(user).Error; err != nil {
return nil, err
}
@@ -97,14 +102,14 @@ func (r databaseUserRepo) Create(req *request.DatabaseUserCreate) error {
}
}
if err = app.Orm.FirstOrInit(user, user).Error; err != nil {
if err = r.db.FirstOrInit(user, user).Error; err != nil {
return err
}
user.Password = req.Password
user.Remark = req.Remark
return app.Orm.Save(user).Error
return r.db.Save(user).Error
}
func (r databaseUserRepo) Update(req *request.DatabaseUserUpdate) error {
@@ -156,7 +161,7 @@ func (r databaseUserRepo) Update(req *request.DatabaseUserUpdate) error {
user.Password = req.Password
user.Remark = req.Remark
return app.Orm.Save(user).Error
return r.db.Save(user).Error
}
func (r databaseUserRepo) UpdateRemark(req *request.DatabaseUserUpdateRemark) error {
@@ -167,7 +172,7 @@ func (r databaseUserRepo) UpdateRemark(req *request.DatabaseUserUpdateRemark) er
user.Remark = req.Remark
return app.Orm.Save(user).Error
return r.db.Save(user).Error
}
func (r databaseUserRepo) Delete(id uint) error {
@@ -198,7 +203,7 @@ func (r databaseUserRepo) Delete(id uint) error {
_ = postgres.UserDrop(user.Username)
}
return app.Orm.Where("id = ?", id).Delete(&biz.DatabaseUser{}).Error
return r.db.Where("id = ?", id).Delete(&biz.DatabaseUser{}).Error
}
func (r databaseUserRepo) DeleteByNames(serverID uint, names []string) error {
@@ -215,7 +220,7 @@ func (r databaseUserRepo) DeleteByNames(serverID uint, names []string) error {
}
defer mysql.Close()
users := make([]*biz.DatabaseUser, 0)
if err = app.Orm.Where("server_id = ? AND username IN ?", serverID, names).Find(&users).Error; err != nil {
if err = r.db.Where("server_id = ? AND username IN ?", serverID, names).Find(&users).Error; err != nil {
return err
}
for name := range slices.Values(names) {
@@ -239,7 +244,7 @@ func (r databaseUserRepo) DeleteByNames(serverID uint, names []string) error {
}
}
return app.Orm.Where("server_id = ? AND username IN ?", serverID, names).Delete(&biz.DatabaseUser{}).Error
return r.db.Where("server_id = ? AND username IN ?", serverID, names).Delete(&biz.DatabaseUser{}).Error
}
func (r databaseUserRepo) fillUser(user *biz.DatabaseUser) {