邹方成
2a89a1ab9d
Some checks failed
Build docker and publish / linux (1.24.5) (push) Failing after 39s
refactor(service): 修改banner和guild删除逻辑为软删除 fix(service): 修复删除操作使用软删除而非物理删除 build: 添加SQLite测试仓库实现 docs: 新增奖励管理字段拆分和批量抽奖UI改造文档 ci: 更新CI忽略文件 style: 清理无用资源文件
123 lines
3.3 KiB
Go
123 lines
3.3 KiB
Go
package mysql
|
|
|
|
import (
|
|
"time"
|
|
|
|
"bindbox-game/internal/pkg/core"
|
|
"bindbox-game/internal/pkg/timeutil"
|
|
"bindbox-game/internal/pkg/trace"
|
|
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/utils"
|
|
)
|
|
|
|
const (
|
|
callBackBeforeName = "core:before"
|
|
callBackAfterName = "core:after"
|
|
startTime = "_start_time"
|
|
)
|
|
|
|
type TracePlugin struct{}
|
|
|
|
func (op *TracePlugin) Name() string {
|
|
return "tracePlugin"
|
|
}
|
|
|
|
func (op *TracePlugin) Initialize(db *gorm.DB) (err error) {
|
|
// 开始前
|
|
_ = db.Callback().Create().Before("gorm:before_create").Register(callBackBeforeName, before)
|
|
_ = db.Callback().Query().Before("gorm:query").Register(callBackBeforeName, before)
|
|
_ = db.Callback().Query().Before("gorm:query").Register("soft_delete:filter", softDeleteFilter)
|
|
_ = db.Callback().Delete().Before("gorm:before_delete").Register(callBackBeforeName, before)
|
|
_ = db.Callback().Update().Before("gorm:setup_reflect_value").Register(callBackBeforeName, before)
|
|
_ = db.Callback().Row().Before("gorm:row").Register(callBackBeforeName, before)
|
|
_ = db.Callback().Raw().Before("gorm:raw").Register(callBackBeforeName, before)
|
|
|
|
// 结束后
|
|
_ = db.Callback().Create().After("gorm:after_create").Register(callBackAfterName, after)
|
|
_ = db.Callback().Query().After("gorm:after_query").Register(callBackAfterName, after)
|
|
_ = db.Callback().Delete().After("gorm:after_delete").Register(callBackAfterName, after)
|
|
_ = db.Callback().Update().After("gorm:after_update").Register(callBackAfterName, after)
|
|
_ = db.Callback().Row().After("gorm:row").Register(callBackAfterName, after)
|
|
_ = db.Callback().Raw().After("gorm:raw").Register(callBackAfterName, after)
|
|
return
|
|
}
|
|
|
|
var _ gorm.Plugin = &TracePlugin{}
|
|
|
|
func before(db *gorm.DB) {
|
|
db.InstanceSet(startTime, time.Now())
|
|
return
|
|
}
|
|
|
|
func after(db *gorm.DB) {
|
|
_ctx := db.Statement.Context
|
|
ctx, ok := _ctx.(core.StdContext)
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
_ts, isExist := db.InstanceGet(startTime)
|
|
if !isExist {
|
|
return
|
|
}
|
|
|
|
ts, ok := _ts.(time.Time)
|
|
if !ok {
|
|
return
|
|
}
|
|
|
|
sql := db.Dialector.Explain(db.Statement.SQL.String(), db.Statement.Vars...)
|
|
|
|
sqlInfo := new(trace.SQL)
|
|
sqlInfo.Time = timeutil.CSTLayoutString()
|
|
sqlInfo.SQL = sql
|
|
sqlInfo.Stack = utils.FileWithLineNum()
|
|
sqlInfo.Rows = db.Statement.RowsAffected
|
|
sqlInfo.CostSeconds = time.Since(ts).Seconds()
|
|
|
|
if ctx.Trace != nil {
|
|
ctx.Trace.AppendSQL(sqlInfo)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
const softDeleteIgnoreKey = "soft_delete_ignore"
|
|
|
|
var softDeleteTables = map[string]struct{}{
|
|
"activities": {},
|
|
"activity_issues": {},
|
|
"activity_reward_settings": {},
|
|
"products": {},
|
|
"product_categories": {},
|
|
"banner": {},
|
|
"guild": {},
|
|
"system_titles": {},
|
|
"system_title_effects": {},
|
|
"system_item_cards": {},
|
|
"system_coupons": {},
|
|
"menus": {},
|
|
"menu_actions": {},
|
|
"roles": {},
|
|
"role_users": {},
|
|
}
|
|
|
|
func softDeleteFilter(db *gorm.DB) {
|
|
v, ok := db.InstanceGet(softDeleteIgnoreKey)
|
|
if ok {
|
|
b, _ := v.(bool)
|
|
if b {
|
|
return
|
|
}
|
|
}
|
|
tbl := db.Statement.Table
|
|
if tbl == "" {
|
|
return
|
|
}
|
|
if _, ok := softDeleteTables[tbl]; !ok {
|
|
return
|
|
}
|
|
db.Where("deleted_at IS NULL")
|
|
}
|