Golang 重试机制实现与 retry 库使用

在分布式系统和网络编程中，重试机制是处理瞬时故障的常见策略。下面我将介绍几种在 Go 中实现重试机制的方法，并重点介绍一个简单易用的重试库 retry 的使用。

基础重试实现

1. 简单 for 循环重试

func Retry(attempts int, sleep time.Duration, f func() error) error {
    var err error
    for i := 0; i < attempts; i++ {
        if i > 0 {
            time.Sleep(sleep)
            sleep *= 2 // 指数退避
        }
        err = f()
        if err == nil {
            return nil
        }
    }
    return fmt.Errorf("after %d attempts, last error: %s", attempts, err)
}

// 使用示例
err := Retry(3, time.Second, func() error {
    return SomeOperation()
})

2. 带上下文的重试

func RetryWithContext(ctx context.Context, maxRetries int, delay time.Duration, f func() error) error {
    var err error
    for i := 0; i < maxRetries; i++ {
        select {
        case <-ctx.Done():
            return ctx.Err()
        default:
        }
        
        err = f()
        if err == nil {
            return nil
        }
        
        if i < maxRetries-1 {
            select {
            case <-time.After(delay):
            case <-ctx.Done():
                return ctx.Err()
            }
            delay *= 2
        }
    }
    return fmt.Errorf("after %d attempts: %v", maxRetries, err)
}

使用 retry 库

github.com/avast/retry-go 是一个简单易用的重试库，提供了丰富的配置选项。

安装

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

基本用法

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

err := retry.Do(
    func() error {
        return SomeOperation()
    },
    retry.Attempts(3),
    retry.Delay(time.Second),
    retry.OnRetry(func(n uint, err error) {
        log.Printf("Retry #%d: %v\n", n, err)
    }),
)

高级配置选项

err := retry.Do(
    func() error {
        return SomeOperation()
    },
    retry.Attempts(5),                   // 最大重试次数
    retry.Delay(100*time.Millisecond),    // 初始延迟
    retry.MaxDelay(5*time.Second),        // 最大延迟
    retry.DelayType(retry.BackOffDelay),  // 指数退避
    retry.LastErrorOnly(true),           // 只返回最后一次错误
    retry.RetryIf(func(err error) bool { // 自定义重试条件
        return shouldRetry(err)
    }),
    retry.Context(ctx),                  // 上下文控制
)

自定义重试条件

err := retry.Do(
    func() error {
        return SomeOperation()
    },
    retry.RetryIf(func(err error) bool {
        // 只对特定错误重试
        if netErr, ok := err.(net.Error); ok && netErr.Timeout() {
            return true
        }
        return false
    }),
)

其他重试库推荐

1. cenkalti/backoff: 提供了多种退避算法

   b := backoff.NewExponentialBackOff()
   b.MaxElapsedTime = time.Minute
   err := backoff.Retry(operation, b)
   

2. sethvargo/go-retry: 更灵活的重试策略组合

   ctx := context.Background()
   err := retry.Do(ctx, retry.WithMaxRetries(3, retry.NewFixed(100*time.Millisecond)), func(ctx context.Context) error {
       return SomeOperation()
   })
   

最佳实践

1. 设置合理的重试次数和退避策略 - 避免无限重试和重试风暴
2. 使用上下文 - 允许取消长时间运行的重试操作
3. 记录重试事件 - 便于监控和调试
4. 区分可重试错误 - 不是所有错误都值得重试
5. 考虑幂等性 - 确保重试操作不会产生副作用

重试机制是提高系统健壮性的重要手段，但需要谨慎使用以避免引入其他问题。retry-go 库提供了简单而强大的接口，适合大多数 Go 项目的重试需求。