mayfly-go/server/internal/db/api/db_sql.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
}