mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-04 08:20:25 +08:00
129 lines
4.3 KiB
Go
129 lines
4.3 KiB
Go
package api
|
||
|
||
import (
|
||
"mayfly-go/internal/db/api/form"
|
||
"mayfly-go/internal/db/api/vo"
|
||
"mayfly-go/internal/db/application"
|
||
"mayfly-go/internal/db/domain/entity"
|
||
"mayfly-go/pkg/biz"
|
||
"mayfly-go/pkg/ginx"
|
||
"mayfly-go/pkg/req"
|
||
"mayfly-go/pkg/utils/cryptox"
|
||
"strconv"
|
||
"strings"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
type Instance struct {
|
||
InstanceApp application.Instance
|
||
DbApp application.Db
|
||
}
|
||
|
||
// Instances 获取数据库实例信息
|
||
// @router /api/instances [get]
|
||
func (d *Instance) Instances(rc *req.Ctx) {
|
||
queryCond, page := ginx.BindQueryAndPage[*entity.InstanceQuery](rc.GinCtx, new(entity.InstanceQuery))
|
||
res, err := d.InstanceApp.GetPageList(queryCond, page, new([]vo.InstanceListVO))
|
||
biz.ErrIsNil(err)
|
||
rc.ResData = res
|
||
}
|
||
|
||
func (d *Instance) TestConn(rc *req.Ctx) {
|
||
form := &form.InstanceForm{}
|
||
instance := ginx.BindJsonAndCopyTo[*entity.DbInstance](rc.GinCtx, form, new(entity.DbInstance))
|
||
|
||
// 密码解密，并使用解密后的赋值
|
||
originPwd, err := cryptox.DefaultRsaDecrypt(form.Password, true)
|
||
biz.ErrIsNilAppendErr(err, "解密密码错误: %s")
|
||
instance.Password = originPwd
|
||
|
||
biz.ErrIsNil(d.InstanceApp.TestConn(instance))
|
||
}
|
||
|
||
// SaveInstance 保存数据库实例信息
|
||
// @router /api/instances [post]
|
||
func (d *Instance) SaveInstance(rc *req.Ctx) {
|
||
form := &form.InstanceForm{}
|
||
instance := ginx.BindJsonAndCopyTo[*entity.DbInstance](rc.GinCtx, form, new(entity.DbInstance))
|
||
|
||
// 密码解密，并使用解密后的赋值
|
||
originPwd, err := cryptox.DefaultRsaDecrypt(form.Password, true)
|
||
biz.ErrIsNilAppendErr(err, "解密密码错误: %s")
|
||
instance.Password = originPwd
|
||
|
||
// 密码脱敏记录日志
|
||
form.Password = "****"
|
||
rc.ReqParam = form
|
||
biz.ErrIsNil(d.InstanceApp.Save(rc.MetaCtx, instance))
|
||
}
|
||
|
||
// GetInstance 获取数据库实例密码，由于数据库是加密存储，故提供该接口展示原文密码
|
||
// @router /api/instances/:instance [GET]
|
||
func (d *Instance) GetInstance(rc *req.Ctx) {
|
||
dbId := getInstanceId(rc.GinCtx)
|
||
dbEntity, err := d.InstanceApp.GetById(new(entity.DbInstance), dbId)
|
||
biz.ErrIsNil(err, "获取数据库实例错误")
|
||
dbEntity.Password = ""
|
||
rc.ResData = dbEntity
|
||
}
|
||
|
||
// GetInstancePwd 获取数据库实例密码，由于数据库是加密存储，故提供该接口展示原文密码
|
||
// @router /api/instances/:instance/pwd [GET]
|
||
func (d *Instance) GetInstancePwd(rc *req.Ctx) {
|
||
instanceId := getInstanceId(rc.GinCtx)
|
||
instanceEntity, err := d.InstanceApp.GetById(new(entity.DbInstance), instanceId, "Password")
|
||
biz.ErrIsNil(err, "获取数据库实例错误")
|
||
instanceEntity.PwdDecrypt()
|
||
rc.ResData = instanceEntity.Password
|
||
}
|
||
|
||
// DeleteInstance 删除数据库实例信息
|
||
// @router /api/instances/:instance [DELETE]
|
||
func (d *Instance) DeleteInstance(rc *req.Ctx) {
|
||
idsStr := ginx.PathParam(rc.GinCtx, "instanceId")
|
||
rc.ReqParam = idsStr
|
||
ids := strings.Split(idsStr, ",")
|
||
|
||
for _, v := range ids {
|
||
value, err := strconv.Atoi(v)
|
||
biz.ErrIsNilAppendErr(err, "string类型转换为int异常: %s")
|
||
instanceId := uint64(value)
|
||
if d.DbApp.Count(&entity.DbQuery{InstanceId: instanceId}) != 0 {
|
||
instance, err := d.InstanceApp.GetById(new(entity.DbInstance), instanceId, "name")
|
||
biz.ErrIsNil(err, "获取数据库实例错误，数据库实例ID为: %d", instance.Id)
|
||
biz.IsTrue(false, "不能删除数据库实例【%s】，请先删除其关联的数据库资源。", instance.Name)
|
||
}
|
||
// todo check if backup task has been disabled and backup histories have been deleted
|
||
|
||
d.InstanceApp.Delete(rc.MetaCtx, instanceId)
|
||
}
|
||
}
|
||
|
||
// 获取数据库实例的所有数据库名
|
||
func (d *Instance) GetDatabaseNames(rc *req.Ctx) {
|
||
instanceId := getInstanceId(rc.GinCtx)
|
||
instance, err := d.InstanceApp.GetById(new(entity.DbInstance), instanceId, "Password")
|
||
biz.ErrIsNil(err, "获取数据库实例错误")
|
||
instance.PwdDecrypt()
|
||
res, err := d.InstanceApp.GetDatabases(instance)
|
||
biz.ErrIsNil(err)
|
||
rc.ResData = res
|
||
}
|
||
|
||
// 获取数据库实例server信息
|
||
func (d *Instance) GetDbServer(rc *req.Ctx) {
|
||
instanceId := getInstanceId(rc.GinCtx)
|
||
conn, err := d.DbApp.GetDbConnByInstanceId(instanceId)
|
||
biz.ErrIsNil(err)
|
||
res, err := conn.GetDialect().GetDbServer()
|
||
biz.ErrIsNil(err)
|
||
rc.ResData = res
|
||
}
|
||
|
||
func getInstanceId(g *gin.Context) uint64 {
|
||
instanceId, _ := strconv.Atoi(g.Param("instanceId"))
|
||
biz.IsTrue(instanceId > 0, "instanceId 错误")
|
||
return uint64(instanceId)
|
||
}
|