diff --git a/common/retry/retry.go b/common/retry/retry.go index a9ba75d5a..d647956b1 100644 --- a/common/retry/retry.go +++ b/common/retry/retry.go @@ -6,7 +6,7 @@ import ( ) var ( - errorRetryFailed = errors.New("All retry attempts failed.") + ErrorRetryFailed = errors.New("All retry attempts failed.") ) // Strategy is a way to retry on a specific function. @@ -29,7 +29,7 @@ func (r *retryer) On(method func() error) error { } delay := r.NextDelay(attempt) if delay < 0 { - return errorRetryFailed + return ErrorRetryFailed } <-time.After(time.Duration(delay) * time.Millisecond) attempt++ diff --git a/common/retry/retry_test.go b/common/retry/retry_test.go index 33bb47b8f..0b245cdfb 100644 --- a/common/retry/retry_test.go +++ b/common/retry/retry_test.go @@ -1,10 +1,11 @@ -package retry +package retry_test import ( "errors" "testing" "time" + . "github.com/v2ray/v2ray-core/common/retry" v2testing "github.com/v2ray/v2ray-core/testing" "github.com/v2ray/v2ray-core/testing/assert" ) @@ -76,6 +77,6 @@ func TestRetryExhausted(t *testing.T) { }) duration := time.Since(startTime) - assert.Error(err).Equals(errorRetryFailed) + assert.Error(err).Equals(ErrorRetryFailed) assert.Int64(int64(duration / time.Millisecond)).AtLeast(1900) }