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

feat: 数据库管理提交1

Browse Source
This commit is contained in:
耗子
2024-11-22 02:58:31 +08:00
parent 39d2eee676
commit a96bbf3dfb
17 changed files with 764 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

63
internal/data/database.go
View File

@@ -1,18 +1,25 @@
package data
import (
"fmt"
"github.com/TheTNB/panel/internal/app"
"github.com/TheTNB/panel/internal/biz"
"github.com/TheTNB/panel/internal/http/request"
"github.com/TheTNB/panel/pkg/db"
)
type databaseRepo struct{}
func NewDatabaseRepo() biz.DatabaseRepo {
return &databaseRepo{}
type databaseRepo struct {
databaseServer biz.DatabaseServerRepo
}
func (d databaseRepo) Count() (int64, error) {
func NewDatabaseRepo() biz.DatabaseRepo {
return &databaseRepo{
databaseServer: NewDatabaseServerRepo(),
}
}
func (r databaseRepo) Count() (int64, error) {
var count int64
if err := app.Orm.Model(&biz.Database{}).Count(&count).Error; err != nil {
return 0, err
@@ -21,14 +28,14 @@ func (d databaseRepo) Count() (int64, error) {
return count, nil
}
func (d databaseRepo) List(page, limit uint) ([]*biz.Database, int64, error) {
func (r databaseRepo) List(page, limit uint) ([]*biz.Database, int64, error) {
var database []*biz.Database
var total int64
err := app.Orm.Model(&biz.Database{}).Order("id desc").Count(&total).Offset(int((page - 1) * limit)).Limit(int(limit)).Find(&database).Error
return database, total, err
}
func (d databaseRepo) Get(id uint) (*biz.Database, error) {
func (r databaseRepo) Get(id uint) (*biz.Database, error) {
database := new(biz.Database)
if err := app.Orm.Where("id = ?", id).First(database).Error; err != nil {
return nil, err
@@ -37,7 +44,43 @@ func (d databaseRepo) Get(id uint) (*biz.Database, error) {
return database, nil
}
func (d databaseRepo) Create(req *request.DatabaseCreate) error {
func (r databaseRepo) Create(req *request.DatabaseCreate) error {
server, err := r.databaseServer.Get(req.ServerID)
if err != nil {
return err
}
switch server.Type {
case biz.DatabaseTypeMysql:
mysql, err := db.NewMySQL(server.Username, server.Password, fmt.Sprintf("%s:%d", server.Host, server.Port))
if err != nil {
return err
}
if err = mysql.UserCreate(req.Username, req.Password); err != nil {
return err
}
if err = mysql.DatabaseCreate(req.Name); err != nil {
return err
}
if err = mysql.PrivilegesGrant(req.Username, req.Name); err != nil {
return err
}
case biz.DatabaseTypePostgresql:
postgres, err := db.NewPostgres(server.Username, server.Password, server.Host, server.Port)
if err != nil {
return err
}
if err = postgres.UserCreate(req.Username, req.Password); err != nil {
return err
}
if err = postgres.DatabaseCreate(req.Name); err != nil {
return err
}
if err = postgres.PrivilegesGrant(req.Username, req.Name); err != nil {
return err
}
}
database := &biz.Database{
Name: req.Name,
Username: req.Username,
@@ -50,7 +93,7 @@ func (d databaseRepo) Create(req *request.DatabaseCreate) error {
return app.Orm.Create(database).Error
}
func (d databaseRepo) Update(req *request.DatabaseUpdate) error {
func (r databaseRepo) Update(req *request.DatabaseUpdate) error {
database := &biz.Database{
Name: req.Name,
Username: req.Username,
@@ -61,6 +104,6 @@ func (d databaseRepo) Update(req *request.DatabaseUpdate) error {
return app.Orm.Model(database).Where("id = ?", req.ID).Omit("ServerID").Updates(database).Error
}
func (d databaseRepo) Delete(id uint) error {
func (r databaseRepo) Delete(id uint) error {
return app.Orm.Delete(&biz.Database{}, id).Error
}