go get github.com/sethvargo/go-retry@latest

package main

import (
	"context"
	"fmt"
	"net/http"
	"time"

	retry "github.com/sethvargo/go-retry"
)

// 模擬不穩(wěn)定的 API 調(diào)用
func callUnstableAPI() error {
	resp, err := http.Get("https://api.example.com/unstable")
	if err != nil {
		fmt.Println("API 調(diào)用失敗:", err)
		return err // 網(wǎng)絡(luò)錯(cuò)誤，可重試
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		fmt.Printf("API 返回非 200 狀態(tài)碼: %d\n", resp.StatusCode)
		return fmt.Errorf("status code: %d", resp.StatusCode) // 非 200 可重試
	}

	fmt.Println("API 調(diào)用成功!")
	return nil
}

func main() {
	// 1. 創(chuàng)建上下文（支持超時(shí)/取消）
	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	defer cancel()

	// 2. 定義重試策略：固定間隔 2 秒，最多重試 5 次
	// ConstantBackoff(間隔) + LimitMaxRetries(最大次數(shù))
	strategy := retry.WithMaxRetries(5, retry.ConstantBackoff(2*time.Second))

	// 3. 執(zhí)行重試邏輯
	err := retry.Do(ctx, strategy, func(ctx context.Context) error {
		err := callUnstableAPI()
		if err != nil {
			// 標(biāo)記錯(cuò)誤為“可重試”，觸發(fā)下一次重試
			return retry.RetryableError(err)
		}
		return nil // 成功則終止重試
	})

	// 4. 處理最終結(jié)果
	if err != nil {
		fmt.Printf("所有重試失敗: %v\n", err)
		return
	}
	fmt.Println("重試流程正常結(jié)束")
}

// 每次重試間隔 1 秒，最多重試 3 次
strategy := retry.WithMaxRetries(3, retry.ConstantBackoff(1*time.Second))

// 初始間隔 1 秒，最大間隔 10 秒，最多重試 5 次
// 間隔：1s → 2s → 4s → 8s → 10s（后續(xù)保持 10s）
strategy := retry.WithMaxRetries(5, retry.ExponentialBackoff(1*time.Second, 10*time.Second))

// 指數(shù)退避 + 抖動(dòng)，初始 1s，最大 10s，最多 5 次
strategy := retry.WithMaxRetries(5, retry.Jitter(retry.ExponentialBackoff(1*time.Second, 10*time.Second), 0.5))
// 第二個(gè)參數(shù)是抖動(dòng)因子（0-1），因子越大，隨機(jī)波動(dòng)越明顯

// 初始間隔 1s，每次增加 1s，最大間隔 5s，最多重試 4 次
strategy := retry.WithMaxRetries(4, retry.LinearBackoff(1*time.Second, 1*time.Second, 5*time.Second))

func main() {
	ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
	defer cancel()

	// 僅重試“網(wǎng)絡(luò)錯(cuò)誤”和“5xx 狀態(tài)碼錯(cuò)誤”
	strategy := retry.WithMaxRetries(5, retry.ConstantBackoff(2*time.Second))

	err := retry.Do(ctx, strategy, func(ctx context.Context) error {
		err := callUnstableAPI()
		if err != nil {
			// 自定義過濾邏輯：判斷錯(cuò)誤類型是否可重試
			if isRetryableError(err) {
				return retry.RetryableError(err)
			}
			return err // 不可重試錯(cuò)誤，直接終止
		}
		return nil
	})

	if err != nil {
		fmt.Printf("最終失敗: %v\n", err)
	}
}

// 定義可重試錯(cuò)誤的判斷邏輯
func isRetryableError(err error) bool {
	// 網(wǎng)絡(luò)錯(cuò)誤（如連接超時(shí)、拒絕連接）
	if _, ok := err.(*http.Error); ok {
		return true
	}
	// 5xx 狀態(tài)碼錯(cuò)誤
	if err.Error()[:3] == "500" || err.Error()[:3] == "503" {
		return true
	}
	// 其他錯(cuò)誤（如 400、404）不可重試
	return false
}

func main() {
	// 1. 設(shè)置整體超時(shí) 15 秒
	ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
	defer cancel()

	// 2. 啟動(dòng)一個(gè) goroutine，模擬用戶手動(dòng)取消（5 秒后）
	go func() {
		time.Sleep(5 * time.Second)
		fmt.Println("用戶手動(dòng)取消重試")
		cancel()
	}()

	// 3. 重試策略：指數(shù)退避，最多 10 次（但會(huì)被超時(shí)/取消中斷）
	strategy := retry.WithMaxRetries(10, retry.ExponentialBackoff(1*time.Second, 5*time.Second))

	err := retry.Do(ctx, strategy, func(ctx context.Context) error {
		// 檢查上下文是否已取消/超時(shí)
		select {
		case <-ctx.Done():
			return ctx.Err() // 傳遞取消信號
		default:
		}

		return retry.RetryableError(callUnstableAPI())
	})

	if err != nil {
		fmt.Printf("重試終止: %v\n", err) // 可能是超時(shí)或取消錯(cuò)誤
	}
}

// 自定義策略：前 3 次間隔 1 秒，之后間隔 3 秒
type CustomStrategy struct {
	maxRetries int
}

func (s *CustomStrategy) NextRetry(ctx context.Context, attempt int) (time.Duration, error) {
	// attempt 是已重試次數(shù)（從 0 開始）
	if attempt >= s.maxRetries {
		return 0, retry.ErrMaxRetriesExceeded // 達(dá)到最大次數(shù)，終止
	}

	// 前 3 次間隔 1s，之后 3s
	if attempt < 3 {
		return 1 * time.Second, nil
	}
	return 3 * time.Second, nil
}

// 使用自定義策略
func main() {
	ctx := context.Background()
	strategy := &CustomStrategy{maxRetries: 5}

	err := retry.Do(ctx, strategy, func(ctx context.Context) error {
		return retry.RetryableError(callUnstableAPI())
	})
}

func main() {
	ctx := context.Background()

	// 添加回調(diào)函數(shù)，監(jiān)控每次重試
	strategy := retry.WithCallback(
		retry.WithMaxRetries(5, retry.ConstantBackoff(2*time.Second)),
		func(ctx context.Context, attempt int, err error, next time.Duration) {
			fmt.Printf("第 %d 次重試失敗: %v，下次重試間隔 %v\n", attempt+1, err, next)
			// 此處可添加指標(biāo)上報(bào)（如 Prometheus 計(jì)數(shù)器）
			// retryTotal.WithLabelValues("api").Inc()
		},
	)

	err := retry.Do(ctx, strategy, func(ctx context.Context) error {
		return retry.RetryableError(callUnstableAPI())
	})
}