traceDetailGrpc.go

package linkTrace

import (
	"fmt"
	"github.com/farseer-go/collections"
	"github.com/farseer-go/fs/flog"
	"github.com/farseer-go/fs/parse"
	"github.com/farseer-go/fs/trace"
)

type TraceDetailGrpc struct {
	trace.BaseTraceDetail
	Method          string                                 // post/get/put/delete
	Url             string                                 // url
	Headers         collections.Dictionary[string, string] // 头部
	RequestBody     string                                 // 入参
	ResponseBody    string                                 // 出参
	ResponseHeaders collections.Dictionary[string, string] // 响应头部
	StatusCode      int                                    // 状态码
}

func (receiver *TraceDetailGrpc) GetTraceDetail() *trace.BaseTraceDetail {
	return &receiver.BaseTraceDetail
}

func (receiver *TraceDetailGrpc) ToString() string {
	return fmt.Sprintf("[%s]耗时：%s，%s [%s]%s", flog.Yellow(receiver.CallType.ToString()), flog.Red(receiver.UseTs.String()), receiver.MethodName, receiver.Method, receiver.Url)
}

func (receiver *TraceDetailGrpc) SetHttpRequest(url string, reqHead map[string]any, rspHead map[string]string, requestBody string, responseBody string, statusCode int) {
	receiver.Url = url
	receiver.Headers = collections.NewDictionary[string, string]()
	receiver.RequestBody = requestBody
	receiver.ResponseBody = responseBody
	receiver.StatusCode = statusCode
	for k, v := range reqHead {
		receiver.Headers.Add(k, parse.ToString(v))
	}
	if rspHead != nil {
		receiver.ResponseHeaders = collections.NewDictionaryFromMap(rspHead)
	}
}