mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-04 16:30:25 +08:00
120 lines
3.8 KiB
Go
120 lines
3.8 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/req"
|
|
"mayfly-go/pkg/utils/cryptox"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type Instance struct {
|
|
InstanceApp application.Instance `inject:"DbInstanceApp"`
|
|
DbApp application.Db `inject:""`
|
|
}
|
|
|
|
// Instances 获取数据库实例信息
|
|
// @router /api/instances [get]
|
|
func (d *Instance) Instances(rc *req.Ctx) {
|
|
queryCond, page := req.BindQueryAndPage[*entity.InstanceQuery](rc, 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 := req.BindJsonAndCopyTo[*entity.DbInstance](rc, 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 := req.BindJsonAndCopyTo[*entity.DbInstance](rc, 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)
|
|
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)
|
|
instanceEntity, err := d.InstanceApp.GetById(new(entity.DbInstance), instanceId, "Password")
|
|
biz.ErrIsNil(err, "获取数据库实例错误")
|
|
biz.ErrIsNil(instanceEntity.PwdDecrypt())
|
|
rc.ResData = instanceEntity.Password
|
|
}
|
|
|
|
// DeleteInstance 删除数据库实例信息
|
|
// @router /api/instances/:instance [DELETE]
|
|
func (d *Instance) DeleteInstance(rc *req.Ctx) {
|
|
idsStr := rc.F.PathParam("instanceId")
|
|
rc.ReqParam = idsStr
|
|
ids := strings.Split(idsStr, ",")
|
|
|
|
for _, v := range ids {
|
|
value, err := strconv.Atoi(v)
|
|
biz.ErrIsNilAppendErr(err, "删除数据库实例失败: %s")
|
|
instanceId := uint64(value)
|
|
err = d.InstanceApp.Delete(rc.MetaCtx, instanceId)
|
|
biz.ErrIsNilAppendErr(err, "删除数据库实例失败: %s")
|
|
}
|
|
}
|
|
|
|
// 获取数据库实例的所有数据库名
|
|
func (d *Instance) GetDatabaseNames(rc *req.Ctx) {
|
|
instanceId := getInstanceId(rc)
|
|
instance, err := d.InstanceApp.GetById(new(entity.DbInstance), instanceId, "Password")
|
|
biz.ErrIsNil(err, "获取数据库实例错误")
|
|
biz.ErrIsNil(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)
|
|
conn, err := d.DbApp.GetDbConnByInstanceId(instanceId)
|
|
biz.ErrIsNil(err)
|
|
res, err := conn.GetDialect().GetDbServer()
|
|
biz.ErrIsNil(err)
|
|
rc.ResData = res
|
|
}
|
|
|
|
func getInstanceId(rc *req.Ctx) uint64 {
|
|
instanceId := rc.F.PathParamInt("instanceId")
|
|
biz.IsTrue(instanceId > 0, "instanceId 错误")
|
|
return uint64(instanceId)
|
|
}
|