feat: add new trace function to elton.Context

vicanso · Oct 9, 2021 · 14e8224 · 14e8224
1 parent cc95d38
commit 14e8224
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/context.go b/context.go
@@ -698,6 +698,13 @@ func (c *Context) GetTrace() *Trace {
 	return GetTrace(c.Context())
 }
 
+// NewTrace returns a new trace and set it to context value
+func (c *Context) NewTrace() *Trace {
+	trace := NewTrace()
+	c.WithContext(context.WithValue(c.Context(), ContextTraceKey, trace))
+	return trace
+}
+
 // ServerTiming converts trace info to http response server timing
 func (c *Context) ServerTiming(traceInfos TraceInfos, prefix string) {
 	value := traceInfos.ServerTiming(prefix)