mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-03 16:00:25 +08:00
91 lines
2.3 KiB
Go
91 lines
2.3 KiB
Go
package api
|
||
|
||
import (
|
||
"mayfly-go/internal/db/api/form"
|
||
"mayfly-go/internal/db/application"
|
||
"mayfly-go/internal/db/domain/entity"
|
||
"mayfly-go/pkg/biz"
|
||
"mayfly-go/pkg/model"
|
||
"mayfly-go/pkg/req"
|
||
)
|
||
|
||
type DbSql struct {
|
||
dbSqlApp application.DbSql `inject:"T"`
|
||
}
|
||
|
||
func (d *DbSql) ReqConfs() *req.Confs {
|
||
reqs := [...]*req.Conf{
|
||
// 用户sql相关
|
||
req.NewPost(":dbId/sql", d.SaveSql),
|
||
|
||
req.NewGet(":dbId/sql", d.GetSql),
|
||
|
||
req.NewDelete(":dbId/sql", d.DeleteSql),
|
||
|
||
req.NewGet(":dbId/sql-names", d.GetSqlNames),
|
||
}
|
||
|
||
return req.NewConfs("/dbs", reqs[:]...)
|
||
}
|
||
|
||
// @router /api/db/:dbId/sql [post]
|
||
func (d *DbSql) SaveSql(rc *req.Ctx) {
|
||
dbSqlForm := &form.DbSqlSaveForm{}
|
||
req.BindJsonAndValid(rc, dbSqlForm)
|
||
rc.ReqParam = dbSqlForm
|
||
|
||
dbId := getDbId(rc)
|
||
|
||
account := rc.GetLoginAccount()
|
||
// 获取用于是否有该dbsql的保存记录，有则更改，否则新增
|
||
dbSql := &entity.DbSql{Type: dbSqlForm.Type, DbId: dbId, Name: dbSqlForm.Name, Db: dbSqlForm.Db}
|
||
dbSql.CreatorId = account.Id
|
||
e := d.dbSqlApp.GetByCond(dbSql)
|
||
|
||
// 更新sql信息
|
||
dbSql.Sql = dbSqlForm.Sql
|
||
if e == nil {
|
||
d.dbSqlApp.UpdateById(rc.MetaCtx, dbSql)
|
||
} else {
|
||
d.dbSqlApp.Insert(rc.MetaCtx, dbSql)
|
||
}
|
||
}
|
||
|
||
// 获取所有保存的sql names
|
||
func (d *DbSql) GetSqlNames(rc *req.Ctx) {
|
||
dbId := getDbId(rc)
|
||
dbName := getDbName(rc)
|
||
// 获取用于是否有该dbsql的保存记录，有则更改，否则新增
|
||
dbSql := &entity.DbSql{Type: 1, DbId: dbId, Db: dbName}
|
||
dbSql.CreatorId = rc.GetLoginAccount().Id
|
||
sqls, _ := d.dbSqlApp.ListByCond(model.NewModelCond(dbSql).Columns("id", "name"))
|
||
|
||
rc.ResData = sqls
|
||
}
|
||
|
||
// 删除保存的sql
|
||
func (d *DbSql) DeleteSql(rc *req.Ctx) {
|
||
dbSql := &entity.DbSql{Type: 1, DbId: getDbId(rc)}
|
||
dbSql.CreatorId = rc.GetLoginAccount().Id
|
||
dbSql.Name = rc.Query("name")
|
||
dbSql.Db = rc.Query("db")
|
||
|
||
biz.ErrIsNil(d.dbSqlApp.DeleteByCond(rc.MetaCtx, dbSql))
|
||
}
|
||
|
||
// @router /api/db/:dbId/sql [get]
|
||
func (d *DbSql) GetSql(rc *req.Ctx) {
|
||
dbId := getDbId(rc)
|
||
dbName := getDbName(rc)
|
||
// 根据创建者id， 数据库id，以及sql模板名称查询保存的sql信息
|
||
dbSql := &entity.DbSql{Type: 1, DbId: dbId, Db: dbName}
|
||
dbSql.CreatorId = rc.GetLoginAccount().Id
|
||
dbSql.Name = rc.Query("name")
|
||
|
||
e := d.dbSqlApp.GetByCond(dbSql)
|
||
if e != nil {
|
||
return
|
||
}
|
||
rc.ResData = dbSql
|
||
}
|