49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package zipkinHttp
|
||
|
||
import (
|
||
"encoding/json"
|
||
"strings"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
"github.com/zzfup/go-fetch"
|
||
)
|
||
|
||
/*c，需要在上层(application、domain、表现层等)传递*/
|
||
func SendByFetch(c *gin.Context, url, method string, body interface{}) (fetch.Resp, error) {
|
||
var headers = map[string]string{
|
||
"Accept": "application/json, text/plain, */*",
|
||
"Content-Type": "application/json",
|
||
}
|
||
|
||
traceSpanId, exists := c.Get("zipkin_span_id")
|
||
if exists {
|
||
value, exists := c.Get("zipkin_trace_span_id_" + traceSpanId.(string))
|
||
if exists && value != nil {
|
||
headers["Zipkin-Span-Context"] = value.(string)
|
||
}
|
||
}
|
||
|
||
options := fetch.Options{
|
||
Method: strings.ToUpper(method),
|
||
Header: headers,
|
||
}
|
||
|
||
if body != nil {
|
||
options.Body, _ = json.Marshal(body)
|
||
}
|
||
|
||
return fetch.Fetch(url, options)
|
||
}
|
||
|
||
func SendByFetchOptions(c *gin.Context, url string, options fetch.Options) (fetch.Resp, error) {
|
||
traceSpanId, exists := c.Get("zipkin_span_id")
|
||
if exists {
|
||
value, exists := c.Get("zipkin_trace_span_id_" + traceSpanId.(string))
|
||
if exists && value != nil {
|
||
options.Header["Zipkin-Span-Context"] = value.(string)
|
||
}
|
||
}
|
||
|
||
return fetch.Fetch(url, options)
|
||
}
|