added error to MaxRetriesExceeded func

Wr4thon · May 6, 2021 · fff9876 · fff9876
1 parent 67626e1
commit fff9876
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/middleware/errorcounter.go b/middleware/errorcounter.go
@@ -15,7 +15,7 @@ const errorCounterKey = "@errorCounter"
 // TODO validate
 type ErrorCounterConfig struct {
 	MaxRetries         int32
-	MaxRetriesExceeded func(gorabbitmq.Context) error
+	MaxRetriesExceeded func(gorabbitmq.Context, error) error
 }
 
 // ErrorCounterWithConfig is a middleware that counts errors when they occur.
@@ -41,7 +41,7 @@ func ErrorCounterWithConfig(config ErrorCounterConfig) gorabbitmq.MiddlewareFunc
 
 				if errorCounter, ok := c.Value(ErrorCounter{}).(int32); ok {
 					if errorCounter >= config.MaxRetries {
-						retryError := config.MaxRetriesExceeded(c)
+						retryError := config.MaxRetriesExceeded(c, err)
 						if retryError != nil {
 							return errors.Wrapf(retryError, "failed to execute maxRetriesExceeded handler, after %v", err)
 						}

diff --git a/middleware/errorcounter_test.go b/middleware/errorcounter_test.go
@@ -59,7 +59,7 @@ func Test_Middleware(t *testing.T) {
 		middleware.ErrorCounterWithConfig(
 			middleware.ErrorCounterConfig{
 				MaxRetries: 5,
-				MaxRetriesExceeded: func(gorabbitmq.Context) error {
+				MaxRetriesExceeded: func(gorabbitmq.Context, error) error {
 					log.Print("Exceeded!")
 					return nil
 				},