2024年Go最新使用go-retryablehttp包实现http“链接池”效果(1)，Golang开发者必须收藏的8个开源库

本文链接：https://blog.csdn.net/2401_84911002/article/details/138834894

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

func GetHttpClient(tag string, config *tls.Config) *http.Client {
muClient.Lock()
// 连接池字典
clientMap := make(map[string]*retryablehttp.Client)

// 带证书认证的结构
transportMap := make(map[string]\*http.Transport)

defer muClient.Unlock()
// 如果是一个带证书的请求，在这里处理
if config != nil {
    if \_, ok := transportMap[tag]; !ok {
        transportMap[tag] = NewTransport()
        transportMap[tag].TLSClientConfig = config
    }
    return &http.Client{
        Transport: transportMap[tag],
        Timeout:   RequestTimeout,
    }
}
if \_, ok := clientMap[tag]; !ok {
    clientMap[tag] = NewRetryHttpClient()
}
return clientMap[tag].StandardClient()

}

// 使用http连接池
func NewTransport() *http.Transport {
dialContext := (&net.Dialer{
Timeout: ConnectTimeout,
KeepAlive: 30 * time.Second,
}).DialContext

return &http.Transport{
    Proxy:                 http.ProxyFromEnvironment,
    DialContext:           dialContext,
    MaxIdleConns:          100,
    MaxConnsPerHost:       25,
    IdleConnTimeout:       30 \* time.Second,
    TLSHandshakeTimeout:   ConnectTimeout,
    ExpectContinueTimeout: 1 \* time.Second,
}

}

func NewRetryHttpClient() *retryablehttp.Client {
retryClient := retryablehttp.NewClient()
retryClient.RetryMax = 10
retryClient.Logger = nil
return retryClient
}


**使用链接池并发处理http请求**

package main

import (
“fmt”
“http_pool/utils”
“io/ioutil”
“log”
“net/http”
“sync”
)

func main() {

// 使用 waitGroup开goroutine
wait := sync.WaitGroup{}
for i := 0; i < 10; i++ {
    wait.Add(1)
    go getBaidu(i, &wait)
}
// 等子goroutine走完了再走主的
wait.Wait()
fmt.Println("------ 所有goroutine均请求完成 ------")

}

// 测试请求百度链接的代码 ———— TODO 里面使用 “http链接池” 做优化
func getBaidu(i int, wait *sync.WaitGroup) {
// 在这里写 wait.Done()
defer wait.Done()

url := "http://www.baidu.com"
req, err := http.NewRequest("GET", url, nil)

if err != nil {
    log.Panic("err1: ", err)
}

req.Header.Add("cache-control", "no-cache")
req.Header.Add("Postman-Token", "f4a59d50-9672-4710-a8c0-abb8a453b199")

// TODO：注意！不使用"http链接池"会在程序运行期间产生大量无效的链接资源，给服务器增加不必要的负担
// res, err := http.DefaultClient.Do(req)

// 使用 "http链接池" 可以优化http链接资源
httpClient := utils.GetHttpClient("get\_baidu", nil)