From d47b19071770f251a87689c8501bc1327a7eb388 Mon Sep 17 00:00:00 2001 From: kevin Date: Fri, 20 Oct 2023 14:25:50 +0800 Subject: [PATCH] chore: refactor errors --- core/breaker/breaker.go | 6 +++--- core/breaker/googlebreaker_test.go | 4 ++-- core/logx/writer.go | 1 - 3 files changed, 5 insertions(+), 6 deletions(-) diff --git a/core/breaker/breaker.go b/core/breaker/breaker.go index 3d5102ef89fd..9fc4a48321df 100644 --- a/core/breaker/breaker.go +++ b/core/breaker/breaker.go @@ -46,7 +46,7 @@ type ( // DoWithAcceptable returns an error instantly if the Breaker rejects the request. // If a panic occurs in the request, the Breaker handles it as an error // and causes the same panic again. - // acceptable checks if it's a successful call, even if the err is not nil. + // acceptable checks if it's a successful call, even if the error is not nil. DoWithAcceptable(req func() error, acceptable Acceptable) error // DoWithFallback runs the given request if the Breaker accepts it. @@ -59,7 +59,7 @@ type ( // DoWithFallbackAcceptable runs the fallback if the Breaker rejects the request. // If a panic occurs in the request, the Breaker handles it as an error // and causes the same panic again. - // acceptable checks if it's a successful call, even if the err is not nil. + // acceptable checks if it's a successful call, even if the error is not nil. DoWithFallbackAcceptable(req func() error, fallback func(err error) error, acceptable Acceptable) error } @@ -179,7 +179,7 @@ func (lt loggedThrottle) doReq(req func() error, fallback func(err error) error, } func (lt loggedThrottle) logError(err error) error { - if err == ErrServiceUnavailable { + if errors.Is(err, ErrServiceUnavailable) { // if circuit open, not possible to have empty error window stat.Report(fmt.Sprintf( "proc(%s/%d), callee: %s, breaker is open and requests dropped\nlast errors:\n%s", diff --git a/core/breaker/googlebreaker_test.go b/core/breaker/googlebreaker_test.go index e525de5bb726..7a8c65b8ce61 100644 --- a/core/breaker/googlebreaker_test.go +++ b/core/breaker/googlebreaker_test.go @@ -95,7 +95,7 @@ func TestGoogleBreakerAcceptable(t *testing.T) { assert.Equal(t, errAcceptable, b.doReq(func() error { return errAcceptable }, nil, func(err error) bool { - return err == errAcceptable + return errors.Is(err, errAcceptable) })) } @@ -105,7 +105,7 @@ func TestGoogleBreakerNotAcceptable(t *testing.T) { assert.Equal(t, errAcceptable, b.doReq(func() error { return errAcceptable }, nil, func(err error) bool { - return err != errAcceptable + return !errors.Is(err, errAcceptable) })) } diff --git a/core/logx/writer.go b/core/logx/writer.go index d38fb2a6091f..1d60a5bd77f2 100644 --- a/core/logx/writer.go +++ b/core/logx/writer.go @@ -13,7 +13,6 @@ import ( "sync/atomic" fatihcolor "github.com/fatih/color" - "github.com/zeromicro/go-zero/core/color" )