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

方法列表:

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

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

服务端

  1. package main
  2. import (
  3. "github.com/gogf/gf/frame/g"
  4. "github.com/gogf/gf/net/ghttp"
  5. )
  6. func main() {
  7. s := g.Server()
  8. s.BindHandler("/", func(r *ghttp.Request) {
  9. r.Response.Writef(
  10. "Span-Id:%s,Trace-Id:%s",
  11. r.Header.Get("Span-Id"),
  12. r.Header.Get("Trace-Id"),
  13. )
  14. })
  15. s.SetPort(8199)
  16. s.Run()
  17. }

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

客户端

  1. 使用SetHeader方法

    ``` package main

  1. import (
  2. "fmt"
  3. "github.com/gogf/gf/frame/g"
  4. )
  5. func main() {
  6. c := g.Client()
  7. c.SetHeader("Span-Id", "0.0.1")
  8. c.SetHeader("Trace-Id", "NBC56410N97LJ016FQA")
  9. if r, e := c.Get("http://127.0.0.1:8199/"); e != nil {
  10. panic(e)
  11. } else {
  12. fmt.Println(r.ReadAllString())
  13. }
  14. }
  15. ```
  16. 通过`g.Client()`创建一个自定义的HTTP请求客户端对象,并通过`c.SetHeader`设置自定义的`Header`信息。
  1. 使用SetHeaderRaw方法

    这个方法更加简单,可以通过原始的Header字符串来设置客户端请求Header。

    ``` package main

  1. import (
  2. "fmt"
  3. "github.com/gogf/gf/frame/g"
  4. )
  5. func main() {
  6. c := g.Client()
  7. c.SetHeaderRaw(`
  8. Referer: https://localhost
  9. Span-Id: 0.0.1
  10. Trace-Id: NBC56410N97LJ016FQA
  11. User-Agent: MyTestClient
  12. `)
  13. if r, e := c.Get("http://127.0.0.1:8199/"); e != nil {
  14. panic(e)
  15. } else {
  16. fmt.Println(r.ReadAllString())
  17. }
  18. }
  19. ```
  1. 执行结果

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

    1. Span-Id:0.0.1,Trace-Id:NBC56410N97LJ016FQA

Content Menu