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