GoFrame HTTPClient-自定義Header

?HTTP?客戶端發(fā)起請(qǐng)求時(shí)可以自定義發(fā)送給服務(wù)端的?Header?內(nèi)容，該特性使用?SetHeader*?相關(guān)方法實(shí)現(xiàn)。

方法列表：

func (c *Client) SetHeader(key, value string) *Client
func (c *Client) SetHeaderMap(m map[string]string) *Client
func (c *Client) SetHeaderRaw(headers string) *Client

我們來看一個(gè)客戶端通過?Header?來自定義發(fā)送自定義鏈路跟蹤信息?Span-Id?及?Trace-Id?消息頭的示例。

服務(wù)端

package main

import (
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/net/ghttp"
)

func main() {
	s := g.Server()
	s.BindHandler("/", func(r *ghttp.Request) {
		r.Response.Writef(
			"Span-Id:%s,Trace-Id:%s",
			r.Header.Get("Span-Id"),
			r.Header.Get("Trace-Id"),
		)
	})
	s.SetPort(8199)
	s.Run()
}

由于是作為示例，服務(wù)端的邏輯很簡(jiǎn)單，直接將接收到的?Span-Id?及?Trace-Id?參數(shù)返回給客戶端。

客戶端

• 使用?SetHeader?方法

package main

import (
	"fmt"

	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/gctx"
)

func main() {
	c := g.Client()
	c.SetHeader("Span-Id", "0.0.1")
	c.SetHeader("Trace-Id", "NBC56410N97LJ016FQA")
	if r, e := c.Get(gctx.New(), "http://127.0.0.1:8199/"); e != nil {
		panic(e)
	} else {
		fmt.Println(r.ReadAllString())
	}
}

通過?g.Client()?創(chuàng)建一個(gè)自定義的?HTTP?請(qǐng)求客戶端對(duì)象，并通過?c.SetHeader?設(shè)置自定義的?Header?信息。

• 使用?SetHeaderRaw?方法

這個(gè)方法更加簡(jiǎn)單，可以通過原始的?Header?字符串來設(shè)置客戶端請(qǐng)求?Header?。

package main

import (
	"fmt"

	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/gctx"
)

func main() {
	c := g.Client()
	c.SetHeaderRaw(`
        Referer: https://localhost
        Span-Id: 0.0.1
        Trace-Id: NBC56410N97LJ016FQA
        User-Agent: MyTestClient
    `)
	if r, e := c.Get(gctx.New(), "http://127.0.0.1:8199/"); e != nil {
		panic(e)
	} else {
		fmt.Println(r.ReadAllString())
	}
}

• 執(zhí)行結(jié)果

客戶端代碼執(zhí)行后，終端將會(huì)打印出服務(wù)端的返回結(jié)果，如下：

Span-Id:0.0.1,Trace-Id:NBC56410N97LJ016FQA