在Golang的网络编程场景中,HTTP请求受网络波动、服务端临时过载等因素影响,偶尔会出现失败情况。合理的重试机制可以有效降低这类临时故障带来的影响,提升程序的健壮性。我们可以通过自定义函数和灵活的回退策略来实现适配不同业务场景的重试逻辑。

重试机制的核心设计思路
一个完善的HTTP请求重试机制需要包含几个核心部分:重试次数控制、重试条件判断、回退等待逻辑。首先我们需要定义哪些错误或者状态码需要触发重试,比如网络超时、5xx服务端错误等,而4xx客户端错误通常不需要重试。其次需要设置最大重试次数,避免无限重试消耗资源。最后回退机制决定了每次重试之间的等待时间,常见的策略有固定间隔回退和指数退避回退。
1. 定义重试条件
我们可以通过自定义函数来判断当前请求是否需要重试,这样可以根据业务需求灵活调整判断逻辑。比如下面的函数会判断错误是否为网络错误,或者响应状态码是否为5xx:
package main
import (
"errors"
"net/http"
)
// 判断是否需要重试的函数
// 参数:err 请求错误,resp HTTP响应
// 返回:true表示需要重试,false表示不需要
func shouldRetry(err error, resp *http.Response) bool {
// 如果有请求错误,且不是主动取消的错误,需要重试
if err != nil {
var netErr interface{ Timeout() bool }
if errors.As(err, &netErr) && netErr.Timeout() {
return true
}
return true
}
// 响应状态码为5xx时重试
if resp != nil && resp.StatusCode >= 500 && resp.StatusCode < 600 {
return true
}
return false
}
2. 实现回退机制
回退机制分为固定间隔回退和指数退避回退两种常见类型。固定间隔回退每次重试等待相同的时间,实现简单但可能在服务端过载时加剧压力;指数退避回退每次等待时间是上一次的2倍,能更友好地应对服务端临时故障。
我们先实现固定间隔回退的辅助函数:
package main
import "time"
// 固定间隔回退函数
// 参数:baseDelay 基础等待时间
// 返回:每次重试需要等待的时间
func fixedBackoff(baseDelay time.Duration) func(int) time.Duration {
return func(attempt int) time.Duration {
return baseDelay
}
}
再实现指数退避回退的辅助函数,通常我们还会设置最大等待时间避免等待过久:
package main
import (
"math"
"time"
)
// 指数退避回退函数
// 参数:baseDelay 基础等待时间,maxDelay 最大等待时间
// 返回:每次重试需要等待的时间
func exponentialBackoff(baseDelay time.Duration, maxDelay time.Duration) func(int) time.Duration {
return func(attempt int) time.Duration {
// 计算当前等待时间:baseDelay * 2^attempt
delay := baseDelay * time.Duration(math.Pow(2, float64(attempt)))
if delay > maxDelay {
delay = maxDelay
}
return delay
}
}
实现完整的HTTP请求重试函数
接下来我们把重试条件判断、回退机制和HTTP请求逻辑组合起来,实现完整的重试函数。这个函数会接收请求参数、最大重试次数、重试判断函数、回退函数作为参数,支持灵活配置。
package main
import (
"fmt"
"io"
"log"
"net/http"
"time"
)
// HTTP请求重试函数
// 参数:
// url 请求地址
// maxRetries 最大重试次数
// retryChecker 重试判断函数
// backoff 回退函数
// 返回:最终的响应和错误
func httpRetry(url string, maxRetries int, retryChecker func(error, *http.Response) bool, backoff func(int) time.Duration) (*http.Response, error) {
var resp *http.Response
var err error
for attempt := 0; attempt <= maxRetries; attempt++ {
// 发起HTTP请求
resp, err = http.Get(url)
// 判断是否需要重试
if retryChecker(err, resp) && attempt < maxRetries {
// 计算等待时间
waitTime := backoff(attempt)
log.Printf("第%d次请求失败,%v后重试", attempt+1, waitTime)
time.Sleep(waitTime)
continue
}
// 不需要重试则直接返回结果
break
}
return resp, err
}
func main() {
// 测试:请求一个可能存在临时故障的地址,最多重试3次,使用指数退避回退
url := "http://192.168.0.1:8080/api/test"
maxRetries := 3
// 使用指数退避,基础等待100ms,最大等待2s
backoffFunc := exponentialBackoff(100*time.Millisecond, 2*time.Second)
resp, err := httpRetry(url, maxRetries, shouldRetry, backoffFunc)
if err != nil {
log.Fatalf("请求最终失败:%v", err)
}
defer resp.Body.Close()
// 读取响应内容
body, err := io.ReadAll(resp.Body)
if err != nil {
log.Fatalf("读取响应失败:%v", err)
}
fmt.Printf("请求成功,状态码:%d,响应内容:%s", resp.StatusCode, body)
}
使用注意事项
- 重试前需要确认请求是否是幂等的,比如POST请求如果非幂等,重试可能会导致重复操作,需要根据业务场景判断是否重试。
- 回退时间不要设置过短,避免频繁重试给服务端造成压力,指数退避机制是比较通用的选择。
- 重试次数也不要设置过多,一般3到5次即可,过多重试会延长请求的整体耗时。
- 如果请求需要携带自定义Header或者请求体,需要修改重试函数中的请求创建逻辑,确保每次重试的请求参数一致。
通过上面的自定义函数和回退机制,我们可以快速实现适配不同业务场景的HTTP请求重试逻辑,提升Golang程序中网络请求的稳定性和可靠性。