一文带你了解Golang中强大的重试机制

package main

import (
	"fmt"
	"math/rand"
	"time"
)

// 模拟一个可能失败的操作
func unreliableOperation() error {
	// 模拟随机失败的情况
	if rand.Float32() < 0.7 {
		return fmt.Errorf("transient error")
	}
	return nil
}

// 重试逻辑
func retryOperation(retries int, delay time.Duration) error {
	var err error
	for i := 0; i < retries; i++ {
		err = unreliableOperation()
		if err == nil {
			return nil // 操作成功
		}

		// 打印错误并等待一段时间
		fmt.Printf("Retry %d/%d: %v\n", i+1, retries, err)
		time.Sleep(delay)
	}
	return fmt.Errorf("failed after %d retries: %w", retries, err)
}

func main() {
	rand.Seed(time.Now().UnixNano())

	// 尝试最多 5 次，每次重试间隔 1 秒
	err := retryOperation(5, time.Second)
	if err != nil {
		fmt.Println("Operation failed:", err)
	} else {
		fmt.Println("Operation succeeded")
	}
}

go get github.com/cenkalti/backoff/v4

package main

import (
	"fmt"
	"github.com/cenkalti/backoff/v4"
	"math/rand"
	"time"
)

// 模拟一个可能失败的操作
func unreliableOperation() error {
	// 模拟随机失败的情况
	if rand.Float32() < 0.7 {
		return fmt.Errorf("transient error")
	}
	return nil
}

// 使用 backoff 重试操作
func retryOperationWithBackoff() error {
	// 设置指数退避策略，最大重试间隔为 10 秒
	bo := backoff.NewExponentialBackOff()
	bo.MaxElapsedTime = 30 * time.Second // 最大重试时间限制
	bo.MaxInterval = 10 * time.Second   // 最大间隔时间

	// 定义重试逻辑
	return backoff.Retry(func() error {
		err := unreliableOperation()
		if err != nil {
			return err // 如果操作失败，返回错误并重试
		}
		return nil // 操作成功
	}, bo)
}

func main() {
	rand.Seed(time.Now().UnixNano())

	err := retryOperationWithBackoff()
	if err != nil {
		fmt.Println("Operation failed:", err)
	} else {
		fmt.Println("Operation succeeded")
	}
}

go get github.com/avast/retry-go

package main

import (
	"fmt"
	"github.com/avast/retry-go"
	"math/rand"
	"time"
)

// 模拟一个可能失败的操作
func unreliableOperation() error {
	// 模拟随机失败的情况
	if rand.Float32() < 0.7 {
		return fmt.Errorf("transient error")
	}
	return nil
}

// 使用 retry-go 重试操作
func retryOperationWithRetryGo() error {
	// 使用 retry-go 实现重试，最多重试 5 次，每次重试间隔 1 秒
	err := retry.Do(func() error {
		return unreliableOperation()
	}, retry.Attempts(5), retry.Delay(time.Second))

	return err
}

func main() {
	rand.Seed(time.Now().UnixNano())

	err := retryOperationWithRetryGo()
	if err != nil {
		fmt.Println("Operation failed:", err)
	} else {
		fmt.Println("Operation succeeded")
	}
}