error.go

package retry

import (
	"errors"
	"fmt"
)

// ShouldRetryFunc 根据错误判断是否应该停止重试
// 参数说明:
//   - error: 需要处理的错误
//
// 返回值说明:
//   - ok: 是否应该重试,true表示继续重试,false表示停止重试
type ShouldRetryFunc func(error) bool

type RetryError struct {
	Code  string
	Cause error
}

func (e *RetryError) Error() string {
	if e.Code != "" {
		return fmt.Sprintf("Code: %s, Cause: %s", e.Code, e.Cause.Error())
	}
	return fmt.Sprintf("Retry Cause: %s", e.Error())
}

// 实现errors.Is链式比较
func (e *RetryError) Unwrap() error {
	return e.Cause
}

// 实现errors.Is比较
func (e *RetryError) Is(target error) bool {
	if re, ok := target.(*RetryError); ok {
		return re.Code == e.Code
	}
	return false
}

func NewRetryError(code string, cause error) *RetryError {
	return &RetryError{
		Code:  code,
		Cause: cause,
	}
}

func WrapError(code string, cause error) error {
	if cause == nil {
		return nil
	}

	// 原错误是否已经是RetryError
	var re *RetryError
	if errors.As(cause, &re) {
		return re
	}

	return &RetryError{
		Code:  code,
		Cause: cause,
	}
}

func WrapErrorWithShould(code string, cause error, should ShouldRetryFunc) error {
	if should(cause) {
		return WrapError(code, cause)
	}
	return cause
}