Golang http 超时设置方法

nulijiabei · · 25153 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

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 超时设置方法

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

25153 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传