使用go-retryablehttp包实现http“链接池”效果，2024Golang开发面试题及答案

2401_84266249

已于 2024-04-15 23:00:24 修改

阅读量568

点赞数 23

分类专栏： 2024年程序员学习文章标签： golang http 开发语言

于 2024-04-15 23:00:23 首次发布

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

版权

2024年程序员学习专栏收录该内容

43 篇文章 0 订阅

订阅专栏

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

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

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

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

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

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

正文

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

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

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

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