mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-04 08:20:25 +08:00
113 lines
2.4 KiB
Go
113 lines
2.4 KiB
Go
package ctx
|
||
|
||
import (
|
||
"mayfly-go/base/ginx"
|
||
"mayfly-go/base/model"
|
||
"net/http"
|
||
"time"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// 处理函数
|
||
type HandlerFunc func(*ReqCtx)
|
||
|
||
type ReqCtx struct {
|
||
Req *http.Request // http request
|
||
GinCtx *gin.Context // gin context
|
||
|
||
NeedToken bool // 是否需要token
|
||
LoginAccount *model.LoginAccount // 登录账号信息
|
||
|
||
LogInfo *LogInfo // 日志相关信息
|
||
ReqParam interface{} // 请求参数，主要用于记录日志
|
||
ResData interface{} // 响应结果
|
||
err interface{} // 请求错误
|
||
timed int64 // 执行时间
|
||
}
|
||
|
||
func (rc *ReqCtx) Handle(handler HandlerFunc) {
|
||
ginCtx := rc.GinCtx
|
||
defer func() {
|
||
if err := recover(); err != nil {
|
||
rc.err = err
|
||
ginx.ErrorRes(ginCtx, err)
|
||
}
|
||
// 应用所有请求后置处理器
|
||
ApplyAfterHandler(rc)
|
||
}()
|
||
if ginCtx == nil {
|
||
panic("ginContext == nil")
|
||
}
|
||
|
||
rc.Req = ginCtx.Request
|
||
// 默认为不记录请求参数，可在handler回调函数中覆盖赋值
|
||
rc.ReqParam = 0
|
||
// 默认响应结果为nil，可在handler中赋值
|
||
rc.ResData = nil
|
||
|
||
// 调用请求前所有处理器
|
||
err := ApplyBeforeHandler(rc)
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
|
||
begin := time.Now()
|
||
handler(rc)
|
||
rc.timed = time.Now().Sub(begin).Milliseconds()
|
||
ginx.SuccessRes(ginCtx, rc.ResData)
|
||
}
|
||
|
||
// 请求前置处理器，返回error则停止后续逻辑
|
||
type BeforeHandler interface {
|
||
BeforeHandle(rc *ReqCtx) error
|
||
}
|
||
|
||
// 请求后置处理器
|
||
type AfterHandler interface {
|
||
AfterHandle(rc *ReqCtx)
|
||
}
|
||
|
||
var (
|
||
BeforeHandlers []BeforeHandler
|
||
AfterHandlers []AfterHandler
|
||
)
|
||
|
||
// 应用所有请求前置处理器
|
||
func ApplyBeforeHandler(rc *ReqCtx) error {
|
||
for _, e := range BeforeHandlers {
|
||
if err := e.BeforeHandle(rc); err != nil {
|
||
return err
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// 应用所有后置处理器
|
||
func ApplyAfterHandler(rc *ReqCtx) {
|
||
for _, e := range AfterHandlers {
|
||
e.AfterHandle(rc)
|
||
}
|
||
}
|
||
|
||
// 新建请求上下文，默认需要校验token
|
||
func NewReqCtx() *ReqCtx {
|
||
return &ReqCtx{NeedToken: true}
|
||
}
|
||
|
||
func NewReqCtxWithGin(g *gin.Context) *ReqCtx {
|
||
return &ReqCtx{NeedToken: true, GinCtx: g}
|
||
}
|
||
|
||
// 调用该方法设置请求描述，则默认记录日志，并不记录响应结果
|
||
func (r *ReqCtx) WithLog(li *LogInfo) *ReqCtx {
|
||
r.LogInfo = li
|
||
return r
|
||
}
|
||
|
||
// 是否需要token
|
||
func (r *ReqCtx) WithNeedToken(needToken bool) *ReqCtx {
|
||
r.NeedToken = needToken
|
||
return r
|
||
}
|