mirror of
https://github.com/acepanel/panel.git
synced 2026-02-04 04:22:33 +08:00
89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
package biz
|
|
|
|
import (
|
|
"errors"
|
|
"time"
|
|
|
|
"github.com/go-rat/utils/crypt"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/TheTNB/panel/internal/app"
|
|
"github.com/TheTNB/panel/internal/http/request"
|
|
)
|
|
|
|
type DatabaseType string
|
|
|
|
const (
|
|
DatabaseTypeMysql DatabaseType = "mysql"
|
|
DatabaseTypePostgresql DatabaseType = "postgresql"
|
|
DatabaseTypeRedis DatabaseType = "redis"
|
|
)
|
|
|
|
type DatabaseServer struct {
|
|
ID uint `gorm:"primaryKey" json:"id"`
|
|
Name string `gorm:"not null;unique" json:"name"`
|
|
Type DatabaseType `gorm:"not null" json:"type"`
|
|
Host string `gorm:"not null" json:"host"`
|
|
Port uint `gorm:"not null" json:"port"`
|
|
Username string `gorm:"not null" json:"username"`
|
|
Password string `gorm:"not null" json:"password"`
|
|
Remark string `gorm:"not null" json:"remark"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
|
|
Databases []*Database `gorm:"foreignKey:ServerID" json:"-"`
|
|
}
|
|
|
|
func (r *DatabaseServer) BeforeSave(tx *gorm.DB) error {
|
|
crypter, err := crypt.NewXChacha20Poly1305([]byte(app.Key))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
r.Password, err = crypter.Encrypt([]byte(r.Password))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
func (r *DatabaseServer) AfterFind(tx *gorm.DB) error {
|
|
crypter, err := crypt.NewXChacha20Poly1305([]byte(app.Key))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
password, err := crypter.Decrypt(r.Password)
|
|
if err == nil {
|
|
r.Password = string(password)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// TODO 检查放到业务层
|
|
func (r *DatabaseServer) BeforeDelete(tx *gorm.DB) error {
|
|
if r.Name == "local_mysql" && !app.IsCli {
|
|
return errors.New("can't delete local_mysql, if you must delete it, please uninstall mysql")
|
|
}
|
|
if r.Name == "local_postgresql" && !app.IsCli {
|
|
return errors.New("can't delete local_postgresql, if you must delete it, please uninstall postgresql")
|
|
}
|
|
if r.Name == "local_redis" && !app.IsCli {
|
|
return errors.New("can't delete local_redis, if you must delete it, please uninstall redis")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
type DatabaseServerRepo interface {
|
|
Count() (int64, error)
|
|
List(page, limit uint) ([]*DatabaseServer, int64, error)
|
|
Get(id uint) (*DatabaseServer, error)
|
|
Create(req *request.DatabaseServerCreate) error
|
|
Update(req *request.DatabaseServerCreate) error
|
|
Delete(id uint) error
|
|
}
|