使用go-retryablehttp包实现http“链接池”效果(1)，2024年最新阿里后台开发-CSDN博客

本文链接：https://blog.csdn.net/2401_84266249/article/details/137799100

先自我介绍一下，小编浙江大学毕业，去过华为、字节跳动等大厂，目前阿里P7

深知大多数程序员，想要提升技能，往往是自己摸索成长，但自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年最新Golang全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友。

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

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

如果你需要这些资料，可以添加V获取：vip1024b （备注go）

正文

在go中使用http的方式获取数据时每次通常都会创建一个http的Client对象处理请求，但是如果一次任务中请求的非常频繁，每一次请求都要创建一个Client对象的话势必会造成链接资源的浪费。

在实际中我们知道有一种“链接池”的概念，就是说提前在链接池中创建好链接，每一次请求前都从这个“链接池”中获取链接，请求处理完毕后不释放链接而是将这个链接重新放入链接池中，以便下一次请求使用，这样便十分有效的利用了链接资源，同时也有效的降低了服务器的负载。

Go中有一个第三方包go-retryablehttp能够实现上述的效果。

实现链接池

package utils

import (
“crypto/tls”
“github.com/hashicorp/go-retryablehttp”
“net”
“net/http”
“sync”
“time”
)

const (
ConnectTimeout = 10 * time.Second
RequestTimeout = 30 * time.Second
)

// 加一个锁防止多线程同时写入字典的情况
var muClient sync.Mutex

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”
)

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

需要这份系统化的资料的朋友，可以添加V获取：vip1024b （备注Go）