自定义Header

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-Id及Trace-Id消息头的示例。

服务端

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

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

客户端

  1. 使用SetHeader方法

    1.  package main
    3.  import (
    4.      "fmt"
    5.      "github.com/gogf/gf/frame/g"
    6.  )
    8.  func main() {
    9.      c := g.Client()
    10.      c.SetHeader("Span-Id", "0.0.1")
    11.      c.SetHeader("Trace-Id", "NBC56410N97LJ016FQA")
    12.      if r, e := c.Get("http://127.0.0.1:8199/"); e != nil {
    13.          panic(e)
    14.      } else {
    15.          fmt.Println(r.ReadAllString())
    16.      }
    17.  }

    通过g.Client()创建一个自定义的HTTP请求客户端对象，并通过c.SetHeader设置自定义的Header信息。

  2. 使用SetHeaderRaw方法

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

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

  3. 执行结果

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

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