feat: 重构数据库1

internal/data/database.go

@@ -9,6 +9,7 @@ import (
 	"github.com/TheTNB/panel/internal/biz"
 	"github.com/TheTNB/panel/internal/http/request"
 	"github.com/TheTNB/panel/pkg/db"
+	"github.com/TheTNB/panel/pkg/types"
 )
 
 type databaseRepo struct{}
@@ -19,27 +20,52 @@ func NewDatabaseRepo() biz.DatabaseRepo {
 
 func (r databaseRepo) Count() (int64, error) {
 	var count int64
-	if err := app.Orm.Model(&biz.Database{}).Count(&count).Error; err != nil {
+	if err := app.Orm.Model(&types.Database{}).Count(&count).Error; err != nil {
 		return 0, err
 	}
 
 	return count, nil
 }
 
-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 (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
+func (r databaseRepo) List(page, limit uint) ([]types.Database, int64, error) {
+	var databaseServer []*biz.DatabaseServer
+	if err := app.Orm.Model(&biz.DatabaseServer{}).Order("id desc").Find(&databaseServer).Error; err != nil {
+		return nil, 0, err
 	}
 
-	return database, nil
+	database := make([]types.Database, 0)
+	for _, server := range databaseServer {
+		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 {
+				if databases, err := mysql.Databases(); err == nil {
+					for _, name := range databases {
+						database = append(database, types.Database{
+							Name:     name,
+							ServerID: server.ID,
+							Status:   types.DatabaseStatusValid,
+						})
+					}
+				}
+			}
+		case biz.DatabaseTypePostgresql:
+			postgres, err := db.NewPostgres(server.Username, server.Password, server.Host, server.Port)
+			if err == nil {
+				if databases, err := postgres.Databases(); err == nil {
+					for _, item := range databases {
+						database = append(database, types.Database{
+							Name:     item.Name,
+							ServerID: server.ID,
+							Status:   types.DatabaseStatusValid,
+						})
+					}
+				}
+			}
+		}
+	}
+
+	return database[(page-1)*limit:], int64(len(database)), nil
 }
 
 func (r databaseRepo) Create(req *request.DatabaseCreate) error {
@@ -79,41 +105,29 @@ func (r databaseRepo) Create(req *request.DatabaseCreate) error {
 		}
 	}
 
-	database := &biz.Database{
-		Name:     req.Name,
-		Username: req.Username,
-		Password: req.Password,
-		ServerID: req.ServerID,
-		Status:   biz.DatabaseStatusInvalid,
-		Remark:   req.Remark,
+	return nil
+}
+
+func (r databaseRepo) Delete(serverID uint, name string) error {
+	server, err := NewDatabaseServerRepo().Get(serverID)
+	if err != nil {
+		return err
 	}
 
-	return app.Orm.Create(database).Error
-}
-
-func (r databaseRepo) Update(req *request.DatabaseUpdate) error {
-	database := &biz.Database{
-		Name:     req.Name,
-		Username: req.Username,
-		Password: req.Password,
-		Remark:   req.Remark,
+	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
+		}
+		return mysql.DatabaseDrop(name)
+	case biz.DatabaseTypePostgresql:
+		postgres, err := db.NewPostgres(server.Username, server.Password, server.Host, server.Port)
+		if err != nil {
+			return err
+		}
+		return postgres.DatabaseDrop(name)
 	}
 
-	return app.Orm.Model(database).Where("id = ?", req.ID).Omit("ServerID").Updates(database).Error
-}
-
-func (r databaseRepo) Delete(id uint) error {
-	return app.Orm.Delete(&biz.Database{}, id).Error
-}
-
-func (r databaseRepo) Add(serverID uint, name string) error {
-	database := &biz.Database{
-		Name:     name,
-		Username: name,
-		ServerID: serverID,
-		Status:   biz.DatabaseStatusNormal,
-		Remark:   "sync from server",
-	}
-
-	return app.Orm.Create(database).Error
+	return nil
 }