Golang 高度可配置的重试机制插件库 retry 使用指南

在分布式系统和网络编程中，重试机制是处理瞬时故障的重要策略。Go 语言中有多个优秀的重试库，其中 github.com/avast/retry-go 是一个功能强大且高度可配置的重试库。

安装 retry-go

go get github.com/avast/retry-go/v4

基础用法

package main

import (
	"fmt"
	"time"
	"errors"
	"github.com/avast/retry-go/v4"
)

func main() {
	err := retry.Do(
		func() error {
			fmt.Println("尝试执行操作...")
			return errors.New("模拟失败")
		},
		retry.Attempts(3),
		retry.Delay(100*time.Millisecond),
	)
	
	if err != nil {
		fmt.Printf("最终失败: %v\n", err)
	}
}

高级配置选项

retry-go 提供了丰富的配置选项：

err := retry.Do(
	func() error {
		// 你的业务逻辑
		return nil
	},
	retry.Attempts(5),                  // 最大尝试次数
	retry.Delay(100*time.Millisecond),  // 基础延迟
	retry.MaxDelay(1*time.Second),      // 最大延迟
	retry.DelayType(retry.BackOffDelay), // 退避策略
	retry.OnRetry(func(n uint, err error) {
		fmt.Printf("第 %d 次重试, 错误: %v\n", n, err)
	}),
	retry.RetryIf(func(err error) bool {
		// 只对特定错误重试
		return err.Error() == "可重试错误"
	}),
	retry.LastErrorOnly(true),           // 只返回最后错误
)

自定义重试条件

err := retry.Do(
	func() error {
		// 业务逻辑
		return errors.New("网络错误")
	},
	retry.RetryIf(func(err error) bool {
		// 只对网络错误重试
		return strings.Contains(err.Error(), "网络错误")
	}),
)

指数退避策略

err := retry.Do(
	func() error {
		// 业务逻辑
		return errors.New("服务不可用")
	},
	retry.Attempts(5),
	retry.DelayType(func(n uint, err error, config *retry.Config) time.Duration {
		// 自定义指数退避
		return time.Duration(math.Pow(2, float64(n))) * time.Second
	}),
)

上下文支持

ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()

err := retry.Do(
	func() error {
		// 业务逻辑
		return errors.New("超时重试")
	},
	retry.Attempts(10),
	retry.Context(ctx),
)

最佳实践

1. 合理设置重试次数：通常3-5次足够
2. 使用退避策略：避免雪崩效应
3. 明确重试条件：不是所有错误都适合重试
4. 添加日志：方便排查问题
5. 考虑幂等性：确保重试操作安全

完整示例

package main

import (
	"context"
	"errors"
	"fmt"
	"math/rand"
	"time"

	"github.com/avast/retry-go/v4"
)

func unreliableService() error {
	if rand.Intn(10) < 7 { // 70%概率失败
		return errors.New("服务暂时不可用")
	}
	return nil
}

func main() {
	// 配置重试策略
	retryOpts := []retry.Option{
		retry.Attempts(5),
		retry.Delay(100 * time.Millisecond),
		retry.MaxDelay(2 * time.Second),
		retry.DelayType(retry.BackOffDelay),
		retry.OnRetry(func(n uint, err error) {
			fmt.Printf("重试 #%d: %v\n", n, err)
		}),
		retry.LastErrorOnly(true),
	}

	ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
	defer cancel()

	err := retry.Do(
		func() error {
			fmt.Println("调用不可靠服务...")
			return unreliableService()
		},
		append(retryOpts, retry.Context(ctx))...,
	)

	if err != nil {
		fmt.Printf("最终失败: %v\n", err)
	} else {
		fmt.Println("操作成功!")
	}
}

retry-go 库提供了灵活的重试机制，通过合理配置可以满足大多数场景的需求。关键是根据业务特点选择合适的重试策略和参数。