GoFrame入门教程 GoFrame HTTPClient-自定义Header

2024-02-25 开发教程 GoFrame入门教程 匿名 3

HTTP​客户端发起请求时可以自定义发送给服务端的​Header​内容,该特性使用​SetHeader*​相关方法实现。

方法列表:

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

我们来看一个客户端通过​Header​来自定义发送自定义链路跟踪信息​Span-Id​及​Trace-Id​消息头的示例。

服务端

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()
}

由于是作为示例,服务端的逻辑很简单,直接将接收到的​Span-Id​及​Trace-Id​参数返回给客户端。

客户端

  • 使用​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()​创建一个自定义的​HTTP​请求客户端对象,并通过​c.SetHeader​设置自定义的​Header​信息。

  • 使用​SetHeaderRaw​方法

这个方法更加简单,可以通过原始的​Header​字符串来设置客户端请求​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())
}
}
  • 执行结果

客户端代码执行后,终端将会打印出服务端的返回结果,如下:

Span-Id:0.0.1,Trace-Id:NBC56410N97LJ016FQA