golang常用代码片段--定制化http-client组件

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

  • 封装标准库的http工具
import (
    "crypto/tls"
    "errors"
    "github.com/gin-gonic/gin"
    "io/ioutil"
    "net/http"
    "strconv"
    "strings"
    "time"
)

var httpClient = &http.Client{
    Timeout: 3 * time.Second,
    // 返回301、302重定向时,不会自动发起重定向访问
    CheckRedirect: func(req *http.Request, via []*http.Request) error {
        return http.ErrUseLastResponse
    },
    Transport: &http.Transport{
        TLSClientConfig: &tls.Config{
            // 不校验https证书
            InsecureSkipVerify: true,
        },
        MaxConnsPerHost:     300,
        MaxIdleConns:        150,
        MaxIdleConnsPerHost: 75,
        IdleConnTimeout:     10 * time.Second,
    },
}

func HttpPostWithRetry(url, body string, retryTime int) (result []byte, err error) {
    if retryTime < 0 {
        return nil, errors.New("HttpPostWithRetry retryTime is " + strconv.Itoa(retryTime))
    }
    for i := 0; i < retryTime; i++ {
        if result, err = HttpPost(url, body); err == nil {
            break
        }
    }
    return
}

func HttpGetWithRetry(url string, retryTime int) (result []byte, err error) {
    if retryTime < 0 {
        return nil, errors.New("HttpGetWithRetry retryTime is " + strconv.Itoa(retryTime))
    }
    for i := 0; i < retryTime; i++ {
        if result, err = HttpGet(url); err == nil {
            break
        }
    }
    return
}

func HttpPost(url, body string) ([]byte, error) {
    resp, err := httpClient.Post(url, gin.MIMEPOSTForm, strings.NewReader(body))
    // 重定向时,resp不为nil,如果CheckRedirect返回非ErrUseLastResponse错误,err也不为nil
    if resp != nil {
        defer resp.Body.Close()
    }
    if err != nil {
        return nil, err
    }
    return ioutil.ReadAll(resp.Body)
}

func HttpGet(url string) ([]byte, error) {
    resp, err := httpClient.Get(url)
    // 重定向时,resp不为nil,如果CheckRedirect返回非ErrUseLastResponse错误,err也不为nil
    if resp != nil {
        defer resp.Body.Close()
    }
    if err != nil {
        return nil, err
    }
    return ioutil.ReadAll(resp.Body)
}


有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:

查看原文:golang常用代码片段--定制化http-client组件

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

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