[feat] Add WithFallbackCtxFunc hystrix.Option (#129)

Author: psrajat

hystrix/hystrix_client.go

@@ -17,6 +17,7 @@ import (
 )
 
 type fallbackFunc func(error) error
+type fallbackCtxFunc func(context.Context, error) error
 
 // Client is the hystrix client implementation
 type Client struct {

hystrix/hystrix_client_test.go

@@ -508,7 +508,7 @@ func TestFallBackFunctionIsCalledWithHystrixHTTPClient(t *testing.T) {
 		WithErrorPercentThreshold(10),
 		WithSleepWindow(100),
 		WithRequestVolumeThreshold(10),
-		WithFallbackFunc(func(err error) error {
+		WithFallbackCtxFunc(func(ctx context.Context, err error) error {
 			called = true
 			return err
 		}),
@@ -529,7 +529,7 @@ func TestHystrixHTTPClientReturnsFallbackFailureWithAFallBackFunctionWhichReturn
 		WithErrorPercentThreshold(10),
 		WithSleepWindow(100),
 		WithRequestVolumeThreshold(10),
-		WithFallbackFunc(func(err error) error {
+		WithFallbackCtxFunc(func(ctx context.Context, err error) error {
 			// do something in the fallback function
 			return nil
 		}),

hystrix/options.go

@@ -63,13 +63,18 @@ func WithErrorPercentThreshold(errorPercentThreshold int) Option {
 
 // WithFallbackFunc sets the fallback function
 func WithFallbackFunc(fn fallbackFunc) Option {
+	if fn == nil {
+		return func(_ *Client) {}
+	}
+	return WithFallbackCtxFunc(func(_ context.Context, err error) error {
+		return fn(err)
+	})
+}
+
+// WithFallbackCtxFunc sets the fallback function with context support
+func WithFallbackCtxFunc(fn fallbackCtxFunc) Option {
 	return func(c *Client) {
-		if fn == nil {
-			return
-		}
-		c.fallbackFunc = func(_ context.Context, err error) error {
-			return fn(err)
-		}
+		c.fallbackFunc = fn
 	}
 }