支持多链路

go-admin-team · Oct 30, 2020 · 5078403 · 5078403
1 parent 5e3e160
commit 5078403
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/common/middleware/trace.go b/common/middleware/trace.go
@@ -20,7 +20,8 @@ func Trace() gin.HandlerFunc {
 		} else {
 			sp = opentracing.StartSpan(opName, opentracing.ChildOf(wireContext))
 		}
-		defer sp.Finish()
+		ctx.Set("traceSpan", sp)
 		ctx.Next()
+		sp.Finish()
 	}
 }