Golang http 超时设置方法

c := http.Client{
    Transport: &http.Transport{
        Dial: func(netw, addr string) (net.Conn, error) {
        deadline := time.Now().Add(25 * time.Second)
        c, err := net.DialTimeout(netw, addr, time.Second*20)
        if err != nil {
            return nil, err
        }
        c.SetDeadline(deadline)
        return c, nil
        },
    },
    }

c.Get("http://www.qq.com")

//////////////////////////////////////////////

// 上载请求
func NetUploadJson(addr string, buf interface{}) (*[]byte, *int, error) {
    // 将需要上传的JSON转为Byte
    v, _ := json.Marshal(buf)
    // 上传JSON数据
    req, e := http.NewRequest("POST", addr, bytes.NewReader(v))
    if e != nil {
    // 提交异常,返回错误
    return nil, nil, e
    }
    // Body Type
    req.Header.Set("Content-Type", "application/json")
    // 完成后断开连接
    req.Header.Set("Connection", "close")
    // -------------------------------------------
    // 设置 TimeOut
    DefaultClient := http.Client{
    Transport: &http.Transport{
        Dial: func(netw, addr string) (net.Conn, error) {
        deadline := time.Now().Add(30 * time.Second)
        c, err := net.DialTimeout(netw, addr, time.Second*30)
        if err != nil {
            return nil, err
        }
        c.SetDeadline(deadline)
        return c, nil
        },
    },
    }
    // -------------------------------------------
    // 执行
    resp, ee := DefaultClient.Do(req)
    if ee != nil {
    // 提交异常,返回错误
    return nil, nil, ee
    }
    // 保证I/O正常关闭
    defer resp.Body.Close()
    // 判断返回状态
    if resp.StatusCode == http.StatusOK {
    // 读取返回的数据
    data, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        // 读取异常,返回错误
        return nil, nil, err
    }
    // 将收到的数据与状态返回
    return &data, &resp.StatusCode, nil
    } else if resp.StatusCode != http.StatusOK {
    // 返回异常状态
    return nil, &resp.StatusCode, nil
    }
    // 不会到这里
    return nil, nil, nil
}

// 下载文件
func NetDownloadFile(addr string) (*[]byte, *int, *http.Header, error) {
    // 上传JSON数据
    req, e := http.NewRequest("GET", addr, nil)
    if e != nil {
    // 返回异常
    return nil, nil, nil, e
    }
    // 完成后断开连接
    req.Header.Set("Connection", "close")
    // -------------------------------------------
    // 设置 TimeOut
    DefaultClient := http.Client{
    Transport: &http.Transport{
        Dial: func(netw, addr string) (net.Conn, error) {
        deadline := time.Now().Add(30 * time.Second)
        c, err := net.DialTimeout(netw, addr, time.Second*30)
        if err != nil {
            return nil, err
        }
        c.SetDeadline(deadline)
        return c, nil
        },
    },
    }
    // -------------------------------------------
    // 执行
    resp, ee := DefaultClient.Do(req)
    if ee != nil {
    // 返回异常
    return nil, nil, nil, ee
    }
    // 保证I/O正常关闭
    defer resp.Body.Close()
    // 判断请求状态
    if resp.StatusCode == 200 {
    data, err := ioutil.ReadAll(resp.Body)
    if err != nil {
        // 读取错误,返回异常
        return nil, nil, nil, err
    }
    // 成功,返回数据及状态
    return &data, &resp.StatusCode, &resp.Header, nil
    } else {
    // 失败,返回状态
    return nil, &resp.StatusCode, nil, nil
    }
    // 不会到这里
    return nil, nil, nil, nil
}

本文来自:努力加贝

感谢作者:nulijiabei

查看原文:Golang http 超时设置方法

郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。