实现ws的链路追踪

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

Hide file tree

Showing 2 changed files with 6 additions and 2 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, contentType string, header map[string]string, requestIp string) trace.ITraceContext {
+func (*traceManager) EntryWebSocket(domain string, path 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"))
@@ -83,7 +83,7 @@ func (*traceManager) EntryWebSocket(domain string, path string, contentType stri
 			WebDomain:      domain,
 			WebPath:        path,
 			WebMethod:      "WEBSOCKET",
-			WebContentType: contentType,
+			WebContentType: "",
 			WebHeaders:     headerDictionary.ToDictionary(),
 			WebRequestBody: "Conn",
 			WebRequestIp:   requestIp,

diff --git a/trackContext.go b/trackContext.go
@@ -87,6 +87,10 @@ func (receiver *TraceContext) SetBody(requestBody string, statusCode int, respon
 	receiver.WebContext.WebResponseBody = responseBody
 }
 
+func (receiver *TraceContext) SetResponseBody(responseBody string) {
+	receiver.WebContext.WebResponseBody = responseBody
+}
+
 func (receiver *TraceContext) GetTraceId() string {
 	return receiver.TraceId
 }