调整：链路追踪End接受error参数，

farseer-go · Sep 16, 2024 · 500f8e6 · 500f8e6
1 parent 7387226
commit 500f8e6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/traceManager.go b/traceManager.go
@@ -61,7 +61,7 @@ func (*traceManager) EntryWebApi(domain string, path string, method string, cont
 }
 
 // EntryWebSocket WebSocket入口
-func (*traceManager) EntryWebSocket(domain string, path string, method string, contentType string, header map[string]string, requestIp string) trace.ITraceContext {
+func (*traceManager) EntryWebSocket(domain string, path string, contentType string, header map[string]string, requestIp string) trace.ITraceContext {
 	headerDictionary := collections.NewDictionaryFromMap(header)
 	traceId := parse.ToString(headerDictionary.GetValue("Trace-Id"))
 	traceLevel := parse.ToInt(headerDictionary.GetValue("Trace-Level"))
@@ -82,7 +82,7 @@ func (*traceManager) EntryWebSocket(domain string, path string, method string, c
 		WebContext: WebContext{
 			WebDomain:      domain,
 			WebPath:        path,
-			WebMethod:      method,
+			WebMethod:      "WEBSOCKET",
 			WebContentType: contentType,
 			WebHeaders:     headerDictionary.ToDictionary(),
 			WebRequestBody: "Conn",

diff --git a/trackContext.go b/trackContext.go
@@ -96,7 +96,7 @@ func (receiver *TraceContext) GetStartTs() int64 {
 }
 
 // End 结束当前链路
-func (receiver *TraceContext) End() {
+func (receiver *TraceContext) End(err error) {
 	// 清空当前上下文
 	trace.CurTraceContext.Remove()