Question

这节将介绍一个处理网络连接超时的技巧，网络连接超时是花了太长的时间也没有连接上。记住您已经知道了一个技巧，是从第十章（Go 并发-进阶讨论），我们讨论 context 标准包时学到的。这个技巧展示在 useContext.go 源码文件中。

在这节介绍的方法实现起来非常简单。相关代码保存在 clientTimeOut.go 文件中，分为四个部分来介绍。这个程序接收俩个命令行参数，一个是 URL 另一个是超时秒数。注意第二个参数是可选的。

clientTimeOut.go 的第一部分如下：

package main

import (
    "fmt"
    "io"
    "net"
    "net/http"
    "os"
    "path/filepath"
    "strconv"
    "time"
)

var timeout = time.Duration(time.Second)

clientTimeOut.go 的第二段代码如下：

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 的第三部分代码如下：

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 的其余代码如下：

    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 俩次将产生如下输出：

$ 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() 请求花的时间比预期长，所以超时了！