diff --git a/request/wrapper.go b/request/wrapper.go index 3c51751..c5acda8 100644 --- a/request/wrapper.go +++ b/request/wrapper.go @@ -11,6 +11,8 @@ import ( "strings" "sync" + "git.zhangdeman.cn/zhangdeman/trace" + "git.zhangdeman.cn/zhangdeman/gin/define" "git.zhangdeman.cn/zhangdeman/gin/request/parse_body" "git.zhangdeman.cn/zhangdeman/wrapper" @@ -298,3 +300,8 @@ func (wh *wrapperHandle) GetLogicAfterResponse(ctx *gin.Context) *define.LogicAf func (wh *wrapperHandle) GetCustomContext(ctx *gin.Context) *define.Context { return define.NewContext(ctx) } + +// GetTraceInstance 获取trace实例 +func (wh *wrapperHandle) GetTraceInstance(ctx *gin.Context) *trace.Runtime { + return define.NewContext(ctx).Trace +}