这节将介绍一个处理网络连接超时的技巧,网络连接超时是花了太长的时间也没有连接上。记住您已经知道了一个技巧,是从第十章 (Go 并发-进阶讨论),我们讨论 context
标准包时学到的。这个技巧展示在 useContext.go
源码文件中。
在这节介绍的方法实现起来非常简单。相关代码保存在 clientTimeOut.go
文件中,分为四个部分来介绍。这个程序接收俩个命令行参数,一个是 URL 另一个是超时秒数。注意第二个参数是可选的。
clientTimeOut.go
的第一部分如下:
Copy package main
import (
"fmt"
"io"
"net"
"net/http"
"os"
"path/filepath"
"strconv"
"time"
)
var timeout = time. Duration (time.Second)
clientTimeOut.go
的第二段代码如下:
Copy func Timeout (network, host string ) ( net . Conn , error ) {
conn, err := net. DialTimeout (network, host,timeout)
if err != nil {
return nil , err
}
conn. SetDeadline (time. Now (). Add (timeout))
return conn, nil
}
在下节,您将学习更多关于 SetDeadline()
的功能。Timeout()
函数用在 http.Transport
变量的 Dial
字段。
clientTimeOut.go
的第三部分代码如下:
Copy func main () {
if len (os.Args) == 1 {
fmt. Printf ( "Usage: %s URL TIMEOUT\n" , filepath. Base (os.Args[ 0 ]))
return
}
if len (os.Args) == 3 {
temp, err := strconv. Atoi (os.Args[ 2 ])
if err != nil {
fmt. Println ( "Using Default Timeout!" )
} else {
timeout = time. Duration (time. Duration (temp) * time.Second)
}
}
URL := os.Args[ 1 ]
t := http . Transport {
Dial: Timeout,
}
clientTimeOut.go
的其余代码如下:
Copy client := http.Client {
Transport: & t,
}
data, err := client. Get (URL)
if err != nil {
fmt. Println (err)
return
} else {
defer data.Body. Close ()
_, err := io. Copy (os.Stdout, data.Body)
if err != nil {
fmt. Println (err)
return
}
}
}
使用在第十章 (Go 并发-进阶讨论)开发的 slowWWW.go
web 服务器测试clientTimeOut.go
web 客户端。
执行 clientTimeOut.go
俩次将产生如下输出:
Copy $ go run clientTimeOut.go http://localhost:8001
Serving: /
Delay: 0
$ go run clientTimeOut.go http://localhost:8001
Get http://localhost:8001: read tcp[::1]:57397->[::1]:8001: i/o timeout
从上面的输出您能看出,第一个请求连接所期望的 web 服务器没有问题。然而,第二个 http.Get()
请求花的时间比预期长,所以超时了!