http/httptrace

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

package main

import (
	"fmt"
	"log"
	"net/http"
	"net/http/httptrace"
)

func main() {
	req, _ := http.NewRequest("GET", "http://example.com", nil)
	trace := &httptrace.ClientTrace{
		GotConn: func(connInfo httptrace.GotConnInfo) {
			fmt.Printf("Got Conn: %+v\n", connInfo)
		},
		DNSDone: func(dnsInfo httptrace.DNSDoneInfo) {
			fmt.Printf("DNS Info: %+v\n", dnsInfo)
		},
	}
	req = req.WithContext(httptrace.WithClientTrace(req.Context(), trace))
	_, err := http.DefaultTransport.RoundTrip(req)
	if err != nil {
		log.Fatal(err)
	}
}

1

func WithClientTrace(ctx context.Context, trace *ClientTrace) context.Context

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

type ClientTrace struct {	
    // GetConn 在创建连接或从空闲连接池中获取连接之前调用。hostPort 是目标或代理的“host:port”。
	// 即使已经有一个可用的空闲缓存连接，也会调用 GetConn。
	GetConn func(hostPort string)

    // GotConn 在成功获取连接后调用。对于无法获取连接的情况，没有钩子；而是使用 Transport.RoundTrip 中的错误。
	GotConn func(GotConnInfo)

    // PutIdleConn 在连接返回到空闲连接池时调用。如果 err 为 nil，则连接成功返回到空闲连接池。
	// 如果 err 不为 nil，则表示未成功返回连接。
    // 如果通过 Transport.DisableKeepAlives 禁用连接重用，则不会调用 PutIdleConn。
	// 在调用方的 Response.Body.Close 调用返回之前，不会调用 PutIdleConn。
	// 对于 HTTP/2，此钩子当前不使用。
	PutIdleConn func(err error)

    // GotFirstResponseByte 在响应头的第一个字节可用时调用。
	GotFirstResponseByte func()

    // Got100Continue 在服务器回复“100 Continue”响应时调用。
	Got100Continue func()

    // Got1xxResponse 在最终的非 1xx 响应之前，对每个 1xx 信息响应头进行调用。
	// 对于返回的“100 Continue”响应，即使定义了 Got100Continue，也会调用 Got1xxResponse。
	// 如果返回错误，则客户端请求将中止并带有该错误值。
	Got1xxResponse func(code int, header textproto.MIMEHeader) error

    // DNSStart 在开始 DNS 查找时调用。
	DNSStart func(DNSStartInfo)

	// DNSDone is called when a DNS lookup ends.
    // DNSDone 在 DNS 查找结束时调用。
	DNSDone func(DNSDoneInfo)

    // ConnectStart 在开始新连接的拨号时调用。
    // 如果启用了 net.Dialer.DualStack（IPv6 “Happy Eyeballs”）支持，则可能会多次调用此函数。
	ConnectStart func(network, addr string)

    // ConnectDone 在新连接的拨号完成时调用。提供的 err 指示连接是否成功完成。
	// 如果启用了 net.Dialer.DualStack（“Happy Eyeballs”）支持，则可能会多次调用此函数。
	ConnectDone func(network, addr string, err error)

    // TLSHandshakeStart 在开始 TLS 握手时调用。
    // 当通过 HTTP 代理连接到 HTTPS 站点时，握手将在代理处理 CONNECT 请求后进行。
	TLSHandshakeStart func()

    // TLSHandshakeDone 在 TLS 握手完成后调用，
    // 其中包含成功握手的连接状态，或握手失败时的非 nil 错误。
	TLSHandshakeDone func(tls.ConnectionState, error)

    // WroteHeaderField 在 Transport 写入每个请求头之后调用。
    // 在此调用时，值可能已缓冲并尚未写入网络。
	WroteHeaderField func(key string, value []string)

    // WroteHeaders 在 Transport 写入所有请求头后调用。
	WroteHeaders func()

    // Wait100Continue 如果请求指定了“Expect: 100-continue”，
    // 且 Transport 已写入请求头但在写入请求主体之前等待服务器的“100 Continue”响应。
	Wait100Continue func()

    // WroteRequest 在写入请求和任何正文的结果上调用。
    // 在重试请求的情况下可能会被多次调用。
	WroteRequest func(WroteRequestInfo)
}

1

func ContextClientTrace(ctx context.Context) *ClientTrace

 1
 2
 3
 4
 5
 6
 7
 8
 9
10

type DNSDoneInfo struct {
    // Addrs 是 DNS 查找中找到的 IPv4 和/或 IPv6 地址。切片的内容不应被修改。
	Addrs []net.IPAddr

    // Err 是在 DNS 查找期间出现的任何错误。
	Err error

    // Coalesced 是 Addrs 是否与同时进行相同 DNS 查找的另一个调用者共享。
	Coalesced bool
}

1
2
3

type DNSStartInfo struct {
	Host string
}

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14

type GotConnInfo struct {
    // Conn 是获取的连接。它由 http.Transport 拥有，
    // 不应由 ClientTrace 的用户读取、写入或关闭。
	Conn net.Conn

    // Reused 是此连接是否已经用于另一个 HTTP 请求。
	Reused bool

    // WasIdle 是此连接是否是从空闲连接池获取的。
	WasIdle bool

    // IdleTime 在 WasIdle 为 true 时报告连接先前闲置的时间。
	IdleTime time.Duration
}

1
2
3
4

type WroteRequestInfo struct {
    // Err 是在写入请求时遇到的任何错误。
	Err error
}