mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-04 00:10:25 +08:00
feat: 新增系统操作日志&其他优化
This commit is contained in:
@@ -12,20 +12,38 @@ import (
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
type SaveLogFunc func(*ReqCtx)
|
||||
|
||||
var saveLog SaveLogFunc
|
||||
|
||||
// 设置保存日志处理函数
|
||||
func SetSaveLogFunc(sl SaveLogFunc) {
|
||||
saveLog = sl
|
||||
}
|
||||
|
||||
type LogInfo struct {
|
||||
LogResp bool // 是否记录返回结果
|
||||
Description string // 请求描述
|
||||
Save bool // 是否保存日志
|
||||
}
|
||||
|
||||
// 新建日志信息
|
||||
func NewLogInfo(description string) *LogInfo {
|
||||
return &LogInfo{Description: description, LogResp: false}
|
||||
}
|
||||
|
||||
// 是否记录返回结果
|
||||
func (i *LogInfo) WithLogResp(logResp bool) *LogInfo {
|
||||
i.LogResp = logResp
|
||||
return i
|
||||
}
|
||||
|
||||
// 是否保存日志
|
||||
func (i *LogInfo) WithSave(saveLog bool) *LogInfo {
|
||||
i.Save = saveLog
|
||||
return i
|
||||
}
|
||||
|
||||
func LogHandler(rc *ReqCtx) error {
|
||||
li := rc.LogInfo
|
||||
if li == nil {
|
||||
@@ -41,6 +59,10 @@ func LogHandler(rc *ReqCtx) error {
|
||||
req := rc.GinCtx.Request
|
||||
lfs[req.Method] = req.URL.Path
|
||||
|
||||
// 如果需要保存日志,并且保存日志处理函数存在则执行保存日志函数
|
||||
if li.Save && saveLog != nil {
|
||||
go saveLog(rc)
|
||||
}
|
||||
if err := rc.Err; err != nil {
|
||||
logger.Log.WithFields(lfs).Error(getErrMsg(rc, err))
|
||||
return nil
|
||||
|
||||
@@ -53,7 +53,7 @@ func (rc *ReqCtx) Handle(handler HandlerFunc) {
|
||||
|
||||
begin := time.Now()
|
||||
handler(rc)
|
||||
rc.timed = time.Now().Sub(begin).Milliseconds()
|
||||
rc.timed = time.Since(begin).Milliseconds()
|
||||
if !rc.NoRes {
|
||||
ginx.SuccessRes(ginCtx, rc.ResData)
|
||||
}
|
||||
|
||||
@@ -13,6 +13,9 @@ func RunWebServer() {
|
||||
ctx.UseBeforeHandlerInterceptor(ctx.PermissionHandler)
|
||||
// 日志处理器
|
||||
ctx.UseAfterHandlerInterceptor(ctx.LogHandler)
|
||||
// 设置日志保存函数
|
||||
ctx.SetSaveLogFunc(initialize.InitSaveLogFunc())
|
||||
|
||||
// 注册路由
|
||||
web := initialize.InitRouter()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user