HTTP连接超时

这节将介绍一个处理网络连接超时的技巧,网络连接超时是花了太长的时间也没有连接上。记住您已经知道了一个技巧,是从第十章(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() 请求花的时间比预期长,所以超时了!

最后编辑: kuteng  文档更新时间: 2021-03-27 20:14   作者:kuteng