TRKJ/mayfly-go
2
0
Fork 0
mirror of https://gitee.com/dromara/mayfly-go synced 2025-11-25 10:30:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: redis支持list查看&其他小优化

Browse Source
This commit is contained in:
meilin.huang
2022-09-22 11:56:21 +08:00
parent be00b90c1d
commit 22c401f9d8
82 changed files with 604 additions and 242 deletions
Download Patch File Download Diff File Expand all files Collapse all files

122
server/internal/db/application/db_sql_exec.go
View File

@@ -13,13 +13,23 @@ import (
"github.com/xwb1989/sqlparser"
)
type DbSqlExecReq struct {
DbId uint64
Db string
Sql string
Remark string
LoginAccount *model.LoginAccount
DbInstance *DbInstance
}
type DbSqlExecRes struct {
ColNames []string
Res []map[string]interface{}
}
type DbSqlExec interface {
// 生成sql执行记录
GenExecLog(loginAccount *model.LoginAccount, dbId uint64, db string, sql string, dbInstance *DbInstance) *entity.DbSqlExec
// 保存sql执行记录
Save(*entity.DbSqlExec)
// 执行sql
Exec(execSqlReq *DbSqlExecReq) (*DbSqlExecRes, error)
// 根据条件删除sql执行记录
DeleteBy(condition *entity.DbSqlExec)
@@ -38,37 +48,58 @@ type dbSqlExecAppImpl struct {
dbSqlExecRepo repository.DbSqlExec
}
func (d *dbSqlExecAppImpl) GenExecLog(loginAccount *model.LoginAccount, dbId uint64, db string, sql string, dbInstance *DbInstance) *entity.DbSqlExec {
defer func() {
if err := recover(); err != nil {
global.Log.Error("生成sql执行记录失败", err)
}
}()
func (d *dbSqlExecAppImpl) Exec(execSqlReq *DbSqlExecReq) (*DbSqlExecRes, error) {
sql := execSqlReq.Sql
loginAccount := execSqlReq.LoginAccount
// 如果是pgsql并且为查询语句，则直接返回，sqlparser不支持解析pgsql select语句
if execSqlReq.DbInstance.Type == entity.DbTypePostgres && (strings.HasPrefix(sql, "SELECT") || strings.HasPrefix(sql, "select")) {
return doSelect(nil, execSqlReq, nil)
}
stmt, err := sqlparser.Parse(sql)
if err != nil {
global.Log.Error("记录数据库sql执行记录失败", err)
global.Log.Error("sql解析失败: ", err)
return nil, err
}
dbSqlExecRecord := new(entity.DbSqlExec)
dbSqlExecRecord.DbId = dbId
dbSqlExecRecord.Db = db
dbSqlExecRecord.DbId = execSqlReq.DbId
dbSqlExecRecord.Db = execSqlReq.Db
dbSqlExecRecord.Sql = sql
dbSqlExecRecord.Remark = execSqlReq.Remark
dbSqlExecRecord.SetBaseInfo(loginAccount)
var execRes *DbSqlExecRes
isSelect := false
switch stmt := stmt.(type) {
case *sqlparser.Select:
isSelect = true
execRes, err = doSelect(stmt, execSqlReq, dbSqlExecRecord)
case *sqlparser.Show:
isSelect = true
execRes, err = doSelect(nil, execSqlReq, dbSqlExecRecord)
case *sqlparser.OtherRead:
isSelect = true
execRes, err = doSelect(nil, execSqlReq, dbSqlExecRecord)
case *sqlparser.Update:
doUpdate(stmt, dbInstance, dbSqlExecRecord)
execRes, err = doUpdate(stmt, execSqlReq, dbSqlExecRecord)
case *sqlparser.Delete:
doDelete(stmt, dbInstance, dbSqlExecRecord)
execRes, err = doDelete(stmt, execSqlReq, dbSqlExecRecord)
case *sqlparser.Insert:
doInsert(stmt, dbSqlExecRecord)
execRes, err = doInsert(stmt, execSqlReq, dbSqlExecRecord)
default:
execRes, err = doExec(execSqlReq.Sql, execSqlReq.DbInstance)
}
if err != nil {
return nil, err
}
return dbSqlExecRecord
}
func (d *dbSqlExecAppImpl) Save(dse *entity.DbSqlExec) {
d.dbSqlExecRepo.Insert(dse)
if !isSelect {
// 保存执行记录
d.dbSqlExecRepo.Insert(dbSqlExecRecord)
}
return execRes, nil
}
func (d *dbSqlExecAppImpl) DeleteBy(condition *entity.DbSqlExec) {
@@ -79,7 +110,22 @@ func (d *dbSqlExecAppImpl) GetPageList(condition *entity.DbSqlExec, pageParam *m
return d.dbSqlExecRepo.GetPageList(condition, pageParam, toEntity, orderBy...)
}
func doUpdate(update *sqlparser.Update, dbInstance *DbInstance, dbSqlExec *entity.DbSqlExec) {
func doSelect(selectStmt *sqlparser.Select, execSqlReq *DbSqlExecReq, dbSqlExec *entity.DbSqlExec) (*DbSqlExecRes, error) {
dbInstance := execSqlReq.DbInstance
sql := execSqlReq.Sql
colNames, res, err := dbInstance.SelectData(sql)
if err != nil {
return nil, err
}
return &DbSqlExecRes{
ColNames: colNames,
Res: res,
}, nil
}
func doUpdate(update *sqlparser.Update, execSqlReq *DbSqlExecReq, dbSqlExec *entity.DbSqlExec) (*DbSqlExecRes, error) {
dbInstance := execSqlReq.DbInstance
tableStr := sqlparser.String(update.TableExprs)
// 可能使用别名，故空格切割
tableName := strings.Split(tableStr, " ")[0]
@@ -103,9 +149,13 @@ func doUpdate(update *sqlparser.Update, dbInstance *DbInstance, dbSqlExec *entit
dbSqlExec.OldValue = string(bytes)
dbSqlExec.Table = tableName
dbSqlExec.Type = entity.DbSqlExecTypeUpdate
return doExec(execSqlReq.Sql, dbInstance)
}
func doDelete(delete *sqlparser.Delete, dbInstance *DbInstance, dbSqlExec *entity.DbSqlExec) {
func doDelete(delete *sqlparser.Delete, execSqlReq *DbSqlExecReq, dbSqlExec *entity.DbSqlExec) (*DbSqlExecRes, error) {
dbInstance := execSqlReq.DbInstance
tableStr := sqlparser.String(delete.TableExprs)
// 可能使用别名，故空格切割
table := strings.Split(tableStr, " ")[0]
@@ -119,12 +169,32 @@ func doDelete(delete *sqlparser.Delete, dbInstance *DbInstance, dbSqlExec *entit
dbSqlExec.OldValue = string(bytes)
dbSqlExec.Table = table
dbSqlExec.Type = entity.DbSqlExecTypeDelete
return doExec(execSqlReq.Sql, dbInstance)
}
func doInsert(insert *sqlparser.Insert, dbSqlExec *entity.DbSqlExec) {
func doInsert(insert *sqlparser.Insert, execSqlReq *DbSqlExecReq, dbSqlExec *entity.DbSqlExec) (*DbSqlExecRes, error) {
tableStr := sqlparser.String(insert.Table)
// 可能使用别名，故空格切割
table := strings.Split(tableStr, " ")[0]
dbSqlExec.Table = table
dbSqlExec.Type = entity.DbSqlExecTypeInsert
return doExec(execSqlReq.Sql, execSqlReq.DbInstance)
}
func doExec(sql string, dbInstance *DbInstance) (*DbSqlExecRes, error) {
rowsAffected, err := dbInstance.Exec(sql)
if err != nil {
return nil, err
}
res := make([]map[string]interface{}, 0)
resData := make(map[string]interface{})
resData["影响条数"] = rowsAffected
res = append(res, resData)
return &DbSqlExecRes{
ColNames: []string{"影响条数"},
Res: res,
}, nil
}