12.10 HTTP连接超时

Search…
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() 请求花的时间比预期长，所以超时了！
12.9.1 Go web客户端进阶
12.10.1 SetDeadline 介绍
Last modified 2yr ago
Copy link