net.Conn 具有相关函数 Set[Read|Write]Deadline(time.Time), 参数是绝对时间,当时间超时后会将操作的 I/O 失败,并返回 timeout 的error。
Deadline并不是timeout超时,如果使用 SetDeadline 当做超时使用的话,需要在每一次的 Read和Write前调用来设置本次IO操作的到期时间。
我们可能不想使用 SetDeadline来设置超时,而是采用更高层次的超时函数通过 net/http 来随设置,但是在底层实现上仍然是通过底层调用 SetDeadline来实现的,读取或者写入数据不会重新设置超时的值。
对于 Http Server 来讲设置客户端连接的超时非常有必要,以用来规避大量连接的静默客户端或者异常退出客户端消耗掉服务端大量的文件句柄,甚至导致Accept报错。
http: Accept error: accept tcp [::]:80: accept4: too many open files; retrying in 5ms
HttpServer 暴露了两个超时设置 ReadTimeout 和 WriteTimeout, server.go
type Server struct {
// ReadTimeout is the maximum duration for reading the entire
// request, including the body.
//
// Because ReadTimeout does not let Handlers make per-request
// decisions on each request body's acceptable deadline or
// upload rate, most users will prefer to use
// ReadHeaderTimeout. It is valid to use them both.
ReadTimeout time.Duration
// ReadHeaderTimeout is the amount of time allowed to read
// request headers. The connection's read deadline is reset
// after reading the headers and the Handler can decide what
// is considered too slow for the body.
ReadHeaderTimeout time.Duration
// WriteTimeout is the maximum duration before timing out
// writes of the response. It is reset whenever a new
// request's header is read. Like ReadTimeout, it does not
// let Handlers make decisions on a per-request basis.
WriteTimeout time.Duration
// IdleTimeout is the maximum amount of time to wait for the
// next request when keep-alives are enabled. If IdleTimeout
// is zero, the value of ReadTimeout is used. If both are
// zero, ReadHeaderTimeout is used.
IdleTimeout time.Duration
// ...
}
srv := &http.Server{
ReadTimeout: 5 * time.Second,
WriteTimeout: 10 * time.Second,
}
log.Println(srv.ListenAndServe())ReadTimeout 超时时间涵盖了 Accept 到 Read Header和Body的时间,具体参见上图;其实现方式是 net/http 在 Accept 后马上调用 SetReadDeadline实现的;
WriteTimeout 一般包括 Header 读取后到写入 Response 结束,参见上图所示;
源码如下:
func (c *conn) readRequest(ctx context.Context) (w *response, err error) {
if c.hijacked() {
return nil, ErrHijacked
}
if d := c.server.ReadTimeout; d != 0 {
c.rwc.SetReadDeadline(time.Now().Add(d))
}
if d := c.server.WriteTimeout; d != 0 {
defer func() {
c.rwc.SetWriteDeadline(time.Now().Add(d))
}()
}
// ...
}但是涉及到 HTTPS,由于中间多出了 TLS 握手过程,因此情况更加复杂,SetWriteDeadline会在 Accept后立即设置以包括在 TLS 握手过程中涉及的 写入操作,在这种情况下会导致 WriteTimeout 的时间从连接开始到 HEAD 被读取,参见上图中的虚线部分,源码参见:
// Serve a new connection.
func (c *conn) serve(ctx context.Context) {
c.remoteAddr = c.rwc.RemoteAddr().String()
// ...
if tlsConn, ok := c.rwc.(*tls.Conn); ok {
if d := c.server.ReadTimeout; d != 0 {
c.rwc.SetReadDeadline(time.Now().Add(d))
}
if d := c.server.WriteTimeout; d != 0 {
c.rwc.SetWriteDeadline(time.Now().Add(d))
}
// ...
}
}通过以上的分析可以得知, http.Server 中的 http.ListenAndServe/http.ListenAndServeTLS/http.Serve都默认没有提供设置超时的设置,这些函数默认都关闭了超时机制,而且也没有提供明确的函数启动超时机制,这样会检查导致服务端出现文件句柄泄露,最终拖垮服务程序,因此需要通过设置 http.Server实例中的 ReadTimeout 和 WriteTimeout 属性值来设置。
ServeHTTP 从不能获取到底层的 net.Conn 因此在涉及到需要流式发送 Response 而必须清除掉 WriteTimeout的时候就没有办法做到,则可能也是为什么默认值是0的原因。因为不能够获取到底层的 net.Conn 对象,因此没有办法在每次 Write之前调用 SetWriteDeadline来获取到特定的超时设置。
同样,也没有办法取消一个阻塞的 ResponseWriter.Write,因为 ResponseWriter.Close 也不能简单唤醒一个阻塞的写操作,因此也没有办法手工设置定时器来解决这个问题。
Client 侧的超时可能更加简单或者复杂,取决于我们的具体使用场景。
最简单的方式是采用以下方式:
c := &http.Client{
Timeout: 15 * time.Second,
}
resp, err := c.Get("https://xxx.yyyy/")同服务端的情况以下,如果使用 http.Get而不设置超时时间,可能会导致程序卡在某个地方不能自拔。http.Get 提供了更加细粒度的控制:
net.Dialer.Timeoutlimits the time spent establishing a TCP connection (if a new one is needed).http.Transport.TLSHandshakeTimeoutlimits the time spent performing the TLS handshake.http.Transport.ResponseHeaderTimeoutlimits the time spent reading the headers of the response.http.Transport.ExpectContinueTimeoutlimits the time the client will wait between sending the request headers when including an Expect: 100-continueand receiving the go-ahead to send the body. Note that setting this in 1.6 will disable HTTP/2 (DefaultTransportis special-cased from 1.6.2).
使用如下:
c := &http.Client{
Transport: &http.Transport{
Dial: (&net.Dialer{
Timeout: 30 * time.Second,
KeepAlive: 30 * time.Second,
}).Dial,
TLSHandshakeTimeout: 10 * time.Second,
ResponseHeaderTimeout: 10 * time.Second,
ExpectContinueTimeout: 1 * time.Second,
}
}没有特别具体的时间来确定发送请求的发送所花费的时间。花费在读取影响的主题数据可以通过 time.Timer来进行控制,因为这个行为发生在Client的方法返回以后。 最后,在go 1.7 中,定义了一个 http.Transport.IdleConnTimeout,这个不是控制 client 请求的阻塞阶段,而是表明连接在连接池中空闲的时间。注意 Client 默认会追溯重定向的地址,http.Client.Timeout这个时间也包括重定向获取的消耗,而 http.Transport 是更底层的控制,不会涉及到重定向的相关时间消耗。
net/http提供了两种方式来取消客户端的请求:Request.Cancel 和 Context (1.7),go推荐使用 Context, Request.Cancel会逐渐被废弃。
Request.Cancel
package main
import (
"io"
"io/ioutil"
"log"
"net/http"
"time"
)
func main() {
c := make(chan struct{})
timer := time.AfterFunc(5*time.Second, func() {
close(c)
})
// Serve 256 bytes every second.
req, err := http.NewRequest("GET", "http://httpbin.org/range/2048?duration=8&chunk_size=256", nil)
if err != nil {
log.Fatal(err)
}
req.Cancel = c
log.Println("Sending request...")
resp, err := http.DefaultClient.Do(req)
if err != nil {
log.Fatal(err)
}
defer resp.Body.Close()
log.Println("Reading body...")
for {
timer.Reset(2 * time.Second)
// Try instead: timer.Reset(50 * time.Millisecond)
_, err = io.CopyN(ioutil.Discard, resp.Body, 256)
if err == io.EOF {
break
} else if err != nil {
log.Fatal(err)
}
}
}context.WithCancel
ctx, cancel := context.WithCancel(context.TODO())
timer := time.AfterFunc(5*time.Second, func() {
cancel()
})
req, err := http.NewRequest("GET", "http://httpbin.org/range/2048?duration=8&chunk_size=256", nil)
if err != nil {
log.Fatal(err)
}
req = req.WithContext(ctx)net/http/httptrace Package httptrace provides mechanisms to trace the events within HTTP client requests.
Go 1.7 中新增的特性,具体介绍参见 Introducing HTTP Tracing,通过设置回调函数来跟踪 http 客户端发起过程中的各类事件:
- Connection 创建
- Connection 复用
- DNS 查找
- 写入 Request 请求
- 读取 Response 消息
Hook的回调函数,通过 context.Context 来进行传递,由于 trace 的相关事件回调函数是基于 Request-Scope 的,因此通过 context.Context 来进行传递是合适的,具体样例参见一下代码:
package main
import (
"fmt"
"log"
"net/http"
"net/http/httptrace"
)
func main() {
req, _ := http.NewRequest("GET", "http://www.baidu.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)
}
}ClientTrace的完整定义:
// ClientTrace is a set of hooks to run at various stages of an outgoing
// HTTP request. Any particular hook may be nil. Functions may be
// called concurrently from different goroutines and some may be called
// after the request has completed or failed.
//
// ClientTrace currently traces a single HTTP request & response
// during a single round trip and has no hooks that span a series
// of redirected requests.
//
// See https://blog.golang.org/http-tracing for more.
type ClientTrace struct {
// GetConn is called before a connection is created or
// retrieved from an idle pool. The hostPort is the
// "host:port" of the target or proxy. GetConn is called even
// if there's already an idle cached connection available.
GetConn func(hostPort string)
// GotConn is called after a successful connection is
// obtained. There is no hook for failure to obtain a
// connection; instead, use the error from
// Transport.RoundTrip.
GotConn func(GotConnInfo)
// PutIdleConn is called when the connection is returned to
// the idle pool. If err is nil, the connection was
// successfully returned to the idle pool. If err is non-nil,
// it describes why not. PutIdleConn is not called if
// connection reuse is disabled via Transport.DisableKeepAlives.
// PutIdleConn is called before the caller's Response.Body.Close
// call returns.
// For HTTP/2, this hook is not currently used.
PutIdleConn func(err error)
// GotFirstResponseByte is called when the first byte of the response
// headers is available.
GotFirstResponseByte func()
// Got100Continue is called if the server replies with a "100
// Continue" response.
Got100Continue func()
// DNSStart is called when a DNS lookup begins.
DNSStart func(DNSStartInfo)
// DNSDone is called when a DNS lookup ends.
DNSDone func(DNSDoneInfo)
// ConnectStart is called when a new connection's Dial begins.
// If net.Dialer.DualStack (IPv6 "Happy Eyeballs") support is
// enabled, this may be called multiple times.
ConnectStart func(network, addr string)
// ConnectDone is called when a new connection's Dial
// completes. The provided err indicates whether the
// connection completedly successfully.
// If net.Dialer.DualStack ("Happy Eyeballs") support is
// enabled, this may be called multiple times.
ConnectDone func(network, addr string, err error)
// TLSHandshakeStart is called when the TLS handshake is started. When
// connecting to a HTTPS site via a HTTP proxy, the handshake happens after
// the CONNECT request is processed by the proxy.
TLSHandshakeStart func()
// TLSHandshakeDone is called after the TLS handshake with either the
// successful handshake's connection state, or a non-nil error on handshake
// failure.
TLSHandshakeDone func(tls.ConnectionState, error)
// WroteHeaders is called after the Transport has written
// the request headers.
WroteHeaders func()
// Wait100Continue is called if the Request specified
// "Expected: 100-continue" and the Transport has written the
// request headers but is waiting for "100 Continue" from the
// server before writing the request body.
Wait100Continue func()
// WroteRequest is called with the result of writing the
// request and any body. It may be called multiple times
// in the case of retried requests.
WroteRequest func(WroteRequestInfo)
}
