2024年最全使用go-retryablehttp包实现http“链接池”效果，【金九银十】-CSDN博客

本文链接：https://blog.csdn.net/2401_84925977/article/details/138886123

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上Go语言开发知识点，真正体系化！

由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新

// 带证书认证的结构
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)
res, err := httpClient.Do(req)

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