Adding the ability to specify a function to check if micro should retry a failed rpc call

client/client.go

@@ -68,6 +68,8 @@ var (
 	DefaultClient Client = newRpcClient()
 	// DefaultBackoff is the default backoff function for retries
 	DefaultBackoff = exponentialBackoff
+	// DefaultCheckIfRetriable is the default check-for-retry function for retries
+	DefaultCheckIfRetriable = AlwaysRetry
 	// DefaultRetries is the default number of times a request is tried
 	DefaultRetries = 1
 	// DefaultRequestTimeout is the default request timeout

client/is_retriable.go

@@ -0,0 +1,22 @@
+package client
+
+import (
+	"github.com/micro/go-micro/errors"
+)
+
+type IsRetriableFunc func(err error) bool
+
+// always retry on error
+func AlwaysRetry(err error) bool {
+	return true
+}
+
+func Only500Errors(err error) bool {
+	errorData := errors.Parse(err.Error())
+
+	if(errorData.Code >= 500) {
+		return true
+	}
+
+	return false
+}

client/options.go

@@ -43,6 +43,8 @@ type CallOptions struct {
 
 	// Backoff func
 	Backoff BackoffFunc
+	// Check if retriable func
+	CheckIfRetriable IsRetriableFunc
 	// Transport Dial Timeout
 	DialTimeout time.Duration
 	// Number of Call attempts
@@ -73,10 +75,11 @@ func newOptions(options ...Option) Options {
 	opts := Options{
 		Codecs: make(map[string]codec.NewCodec),
 		CallOptions: CallOptions{
-			Backoff:        DefaultBackoff,
-			Retries:        DefaultRetries,
-			RequestTimeout: DefaultRequestTimeout,
-			DialTimeout:    transport.DefaultDialTimeout,
+			Backoff:          DefaultBackoff,
+			CheckIfRetriable: DefaultCheckIfRetriable,
+			Retries:          DefaultRetries,
+			RequestTimeout:   DefaultRequestTimeout,
+			DialTimeout:      transport.DefaultDialTimeout,
 		},
 		PoolSize: DefaultPoolSize,
 		PoolTTL:  DefaultPoolTTL,
@@ -221,6 +224,14 @@ func WithBackoff(fn BackoffFunc) CallOption {
 	}
 }
 
+// WithCheckIfRetriable is a CallOption which overrides that which
+// set in Options.CallOptions
+func WithCheckIfRetriable(fn IsRetriableFunc) CallOption {
+	return func(o *CallOptions) {
+		o.CheckIfRetriable = fn
+	}
+}
+
 // WithRetries is a CallOption which overrides that which
 // set in Options.CallOptions
 func WithRetries(i int) CallOption {

client/rpc_client.go

@@ -299,6 +299,11 @@ func (r *rpcClient) Call(ctx context.Context, request Request, response interfac
 			if err == nil {
 				return nil
 			}
+
+			if !callOpts.CheckIfRetriable(err) {
+				return err
+			}
+
 			gerr = err
 		}
 	}