实现ws的链路追踪

farseer-go · Sep 16, 2024 · 9c983b9 · 9c983b9
1 parent f2dc8c4
commit 9c983b9
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/trace/emptyManager.go b/trace/emptyManager.go
@@ -6,7 +6,7 @@ type EmptyManager struct {
 func (*EmptyManager) EntryWebApi(domain string, path string, method string, contentType string, header map[string]string, requestIp string) ITraceContext {
  return &emptyTraceContext{}
 }
-func (*EmptyManager) EntryWebSocket(domain string, path string, contentType string, header map[string]string, requestIp string) ITraceContext {
+func (*EmptyManager) EntryWebSocket(domain string, path string, header map[string]string, requestIp string) ITraceContext {
  return &emptyTraceContext{}
 }
 
@@ -62,6 +62,7 @@ type emptyTraceContext struct{}
 
 func (*emptyTraceContext) Error(err error) {}
 func (*emptyTraceContext) SetBody(requestBody string, statusCode int, responseBody string) {}
+func (*emptyTraceContext) SetResponseBody(responseBody string) {}
 func (*emptyTraceContext) GetTraceId() string { return "" }
 func (*emptyTraceContext) GetTraceLevel() int { return 0 }
 func (*emptyTraceContext) GetStartTs() int64 { return 0 }

diff --git a/trace/iManager.go b/trace/iManager.go
@@ -8,7 +8,7 @@ type IManager interface {
  // EntryWebApi 创建webapi的链路追踪入口
  EntryWebApi(domain string, path string, method string, contentType string, headerDictionary map[string]string, requestIp string) ITraceContext
  // EntryWebSocket 创建WebSocket的链路追踪入口
- EntryWebSocket(domain string, path string, contentType string, headerDictionary map[string]string, requestIp string) ITraceContext
+ EntryWebSocket(domain string, path string, headerDictionary map[string]string, requestIp string) ITraceContext
  // EntryMqConsumer 创建MQ消费入口
  EntryMqConsumer(parentTraceId, parentAppName, server string, queueName string, routingKey string) ITraceContext
  // EntryQueueConsumer 创建Queue消费入口

diff --git a/trace/iTraceContext.go b/trace/iTraceContext.go
@@ -5,6 +5,8 @@ type ITraceContext interface {
  End(err error)
  // SetBody 设置webapi的响应报文
  SetBody(requestBody string, statusCode int, responseBody string)
+ // SetResponseBody 设置webapi的响应报文
+ SetResponseBody(responseBody string)
  // GetTraceId 获取traceId
  GetTraceId() string
  // GetStartTs 获取链路开启时间