From df4f96a2d8111ef971643a836b5577e5837e1349 Mon Sep 17 00:00:00 2001 From: Vasiliy Tolstov Date: Mon, 23 Oct 2023 02:35:10 +0300 Subject: [PATCH 1/2] errors: add IsRetrayable func Signed-off-by: Vasiliy Tolstov --- errors/errors.go | 104 +++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 104 insertions(+) diff --git a/errors/errors.go b/errors/errors.go index 0a16c85b..da47598f 100644 --- a/errors/errors.go +++ b/errors/errors.go @@ -4,11 +4,17 @@ package errors // import "go.unistack.org/micro/v3/errors" import ( "bytes" + "context" "encoding/json" + "errors" "fmt" + "io" "net/http" "strconv" "strings" + + "google.golang.org/grpc/codes" + "google.golang.org/grpc/status" ) var ( @@ -340,3 +346,101 @@ func addslashes(str string) string { } return buf.String() } + +type retryableError struct { + err error +} + +// Retryable returns error that can be retried later +func Retryable(err error) error { + return &retryableError{err: err} +} + +// Unwrap provides error wrapping +func (e *retryableError) Unwrap() error { + return e.err +} + +// Error returns the error string +func (e *retryableError) Error() string { + if e.err == nil { + return "" + } + return e.err.Error() +} + +// IsRetryable checks error for ability to retry later +func IsRetryable(err error) bool { + switch verr := err.(type) { + case *Error: + switch verr.Code { + case 401, 403, 408, 500, 501, 502, 503, 504: + return true + default: + return false + } + case *retryableError: + return true + case interface{ SafeToRetry() bool }: + return verr.SafeToRetry() + case interface{ Timeout() bool }: + return verr.Timeout() + } + + switch { + case errors.Is(err, io.EOF), errors.Is(err, io.ErrUnexpectedEOF): + return true + case errors.Is(err, context.DeadlineExceeded): + return true + case errors.Is(err, io.ErrClosedPipe), errors.Is(err, io.ErrShortBuffer), errors.Is(err, io.ErrShortWrite): + return true + default: + st, ok := status.FromError(err) + if !ok { + errmsg := err.Error() + if strings.Contains(errmsg, `number of field descriptions must equal number of`) { + return true + } + if strings.Contains(errmsg, `not a pointer`) { + return true + } + if strings.Contains(errmsg, `values, but dst struct has only`) { + return true + } + if strings.Contains(errmsg, `struct doesn't have corresponding row field`) { + return true + } + if strings.Contains(errmsg, `cannot find field`) { + return true + } + if strings.Contains(errmsg, `cannot scan`) || strings.Contains(errmsg, `cannot convert`) { + return true + } + if strings.Contains(errmsg, `failed to connect to`) { + return true + } + + return false + } + switch st.Code() { + case codes.Unavailable, codes.ResourceExhausted: + return true + case codes.DeadlineExceeded: + return true + case codes.Internal: + if strings.Contains(st.Message(), `transport: received the unexpected content-type "text/html; charset=UTF-8"`) { + return true + } + if strings.Contains(st.Message(), io.ErrUnexpectedEOF.Error()) { + return true + } + if strings.Contains(st.Message(), `stream terminated by RST_STREAM with error code: INTERNAL_ERROR`) { + return true + } + default: + return false + } + } + + return false +} -- 2.45.2 From d167c8c67c9b98545f806d016d3731ca4729d737 Mon Sep 17 00:00:00 2001 From: Vasiliy Tolstov Date: Wed, 25 Oct 2023 02:36:52 +0300 Subject: [PATCH 2/2] cleanup Signed-off-by: Vasiliy Tolstov --- errors/errors.go | 172 +++++++++++++++++++++++++----------------- errors/errors_test.go | 7 ++ 2 files changed, 110 insertions(+), 69 deletions(-) diff --git a/errors/errors.go b/errors/errors.go index da47598f..e9f0e177 100644 --- a/errors/errors.go +++ b/errors/errors.go @@ -356,6 +356,106 @@ func Retryable(err error) error { return &retryableError{err: err} } +type IsRetryableFunc func(error) bool + +var ( + RetrayableOracleErrors = []IsRetryableFunc{ + func(err error) bool { + errmsg := err.Error() + switch { + case strings.Contains(errmsg, `ORA-`): + return true + case strings.Contains(errmsg, `can not assign`): + return true + case strings.Contains(errmsg, `can't assign`): + return true + } + return false + }, + } + RetrayablePostgresErrors = []IsRetryableFunc{ + func(err error) bool { + errmsg := err.Error() + switch { + case strings.Contains(errmsg, `number of field descriptions must equal number of`): + return true + case strings.Contains(errmsg, `not a pointer`): + return true + case strings.Contains(errmsg, `values, but dst struct has only`): + return true + case strings.Contains(errmsg, `struct doesn't have corresponding row field`): + return true + case strings.Contains(errmsg, `cannot find field`): + return true + case strings.Contains(errmsg, `cannot scan`) || strings.Contains(errmsg, `cannot convert`): + return true + case strings.Contains(errmsg, `failed to connect to`): + return true + } + return false + }, + } + RetryableMicroErrors = []IsRetryableFunc{ + func(err error) bool { + switch verr := err.(type) { + case *Error: + switch verr.Code { + case 401, 403, 408, 500, 501, 502, 503, 504: + return true + default: + return false + } + case *retryableError: + return true + } + return false + }, + } + RetryableGoErrors = []IsRetryableFunc{ + func(err error) bool { + switch verr := err.(type) { + case interface{ SafeToRetry() bool }: + return verr.SafeToRetry() + case interface{ Timeout() bool }: + return verr.Timeout() + } + switch { + case errors.Is(err, io.EOF), errors.Is(err, io.ErrUnexpectedEOF): + return true + case errors.Is(err, context.DeadlineExceeded): + return true + case errors.Is(err, io.ErrClosedPipe), errors.Is(err, io.ErrShortBuffer), errors.Is(err, io.ErrShortWrite): + return true + } + return false + }, + } + RetryableGrpcErrors = []IsRetryableFunc{ + func(err error) bool { + st, ok := status.FromError(err) + if !ok { + return false + } + switch st.Code() { + case codes.Unavailable, codes.ResourceExhausted: + return true + case codes.DeadlineExceeded: + return true + case codes.Internal: + switch { + case strings.Contains(st.Message(), `transport: received the unexpected content-type "text/html; charset=UTF-8"`): + return true + case strings.Contains(st.Message(), io.ErrUnexpectedEOF.Error()): + return true + case strings.Contains(st.Message(), `stream terminated by RST_STREAM with error code: INTERNAL_ERROR`): + return true + } + } + return false + }, + } +) + // Unwrap provides error wrapping func (e *retryableError) Unwrap() error { return e.err @@ -370,77 +470,11 @@ func (e *retryableError) Error() string { } // IsRetryable checks error for ability to retry later -func IsRetryable(err error) bool { - switch verr := err.(type) { - case *Error: - switch verr.Code { - case 401, 403, 408, 500, 501, 502, 503, 504: +func IsRetryable(err error, fns ...IsRetryableFunc) bool { + for _, fn := range fns { + if ok := fn(err); ok { return true - default: - return false - } - case *retryableError: - return true - case interface{ SafeToRetry() bool }: - return verr.SafeToRetry() - case interface{ Timeout() bool }: - return verr.Timeout() - } - - switch { - case errors.Is(err, io.EOF), errors.Is(err, io.ErrUnexpectedEOF): - return true - case errors.Is(err, context.DeadlineExceeded): - return true - case errors.Is(err, io.ErrClosedPipe), errors.Is(err, io.ErrShortBuffer), errors.Is(err, io.ErrShortWrite): - return true - default: - st, ok := status.FromError(err) - if !ok { - errmsg := err.Error() - if strings.Contains(errmsg, `number of field descriptions must equal number of`) { - return true - } - if strings.Contains(errmsg, `not a pointer`) { - return true - } - if strings.Contains(errmsg, `values, but dst struct has only`) { - return true - } - if strings.Contains(errmsg, `struct doesn't have corresponding row field`) { - return true - } - if strings.Contains(errmsg, `cannot find field`) { - return true - } - if strings.Contains(errmsg, `cannot scan`) || strings.Contains(errmsg, `cannot convert`) { - return true - } - if strings.Contains(errmsg, `failed to connect to`) { - return true - } - - return false - } - switch st.Code() { - case codes.Unavailable, codes.ResourceExhausted: - return true - case codes.DeadlineExceeded: - return true - case codes.Internal: - if strings.Contains(st.Message(), `transport: received the unexpected content-type "text/html; charset=UTF-8"`) { - return true - } - if strings.Contains(st.Message(), io.ErrUnexpectedEOF.Error()) { - return true - } - if strings.Contains(st.Message(), `stream terminated by RST_STREAM with error code: INTERNAL_ERROR`) { - return true - } - default: - return false } } - return false } diff --git a/errors/errors_test.go b/errors/errors_test.go index 62374c84..03c264c2 100644 --- a/errors/errors_test.go +++ b/errors/errors_test.go @@ -8,6 +8,13 @@ import ( "testing" ) +func TestIsRetrayable(t *testing.T) { + err := fmt.Errorf("ORA-") + if !IsRetryable(err, RetrayableOracleErrors...) { + t.Fatalf("IsRetrayable not works") + } +} + func TestMarshalJSON(t *testing.T) { e := InternalServerError("id", "err: %v", fmt.Errorf("err: %v", `xxx: "UNIX_TIMESTAMP": invalid identifier`)) _, err := json.Marshal(e) -- 2.45.2