实现ws的链路追踪

farseer-go · Sep 16, 2024 · 2a15543 · 2a15543
1 parent f2e270d
commit 2a15543
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/websocket/baseContext.go b/websocket/baseContext.go
@@ -91,7 +91,7 @@ func (receiver *BaseContext) Send(msg any) error {
 	var message string
 	// 基础类型不需要进行序列化
 	if fastReflect.PointerOf(msg).Type == fastReflect.GoBasicType {
-		message = parse.ToString(message)
+		message = parse.ToString(msg)
 	} else {
 		// 其余类型，一律使用json
 		marshal, _ := json.Marshal(msg)

diff --git a/websocket/socketHandler.go b/websocket/socketHandler.go
@@ -17,7 +17,7 @@ func SocketHandler(route *context.HttpRoute) websocket.Handler {
 		httpContext.SetWebsocket(conn)
 
 		// 创建链路追踪上下文
-		trackContext := container.Resolve[trace.IManager]().EntryWebSocket(httpContext.URI.Host, httpContext.URI.Url, httpContext.ContentType, httpContext.Header.ToMap(), httpContext.URI.GetRealIp())
+		trackContext := container.Resolve[trace.IManager]().EntryWebSocket(httpContext.URI.Host, httpContext.URI.Url, httpContext.Header.ToMap(), httpContext.URI.GetRealIp())
 		trackContext.SetBody(httpContext.Request.BodyString, httpContext.Response.GetHttpCode(), string(httpContext.Response.BodyBytes))
 		trackContext.End(nil)
 		//httpContext.Data.Set("Trace", trackContext)