Add error catcher

swaggest · Apr 14, 2021 · 5c4080c · 5c4080c
1 parent 36494be
commit 5c4080c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/wrap.go b/wrap.go
@@ -11,14 +11,14 @@ type Middleware interface {
 }
 
 // ErrorCatcher is a use case middleware that collects non empty errors.
-type ErrorCatcher func(err error)
+type ErrorCatcher func(ctx context.Context, input interface{}, err error)
 
 // Wrap implements Middleware.
 func (e ErrorCatcher) Wrap(interactor Interactor) Interactor {
 	return Interact(func(ctx context.Context, input, output interface{}) error {
 		err := interactor.Interact(ctx, input, output)
 		if err != nil {
-			e(err)
+			e(ctx, input, err)
 		}
 
 		return err

diff --git a/wrap_test.go b/wrap_test.go
@@ -131,7 +131,7 @@ func TestErrorCatcher_Wrap(t *testing.T) {
 	})
 
 	called := false
-	uw := usecase.Wrap(u, usecase.ErrorCatcher(func(err error) {
+	uw := usecase.Wrap(u, usecase.ErrorCatcher(func(ctx context.Context, input interface{}, err error) {
 		called = true
 		assert.EqualError(t, err, "failed")
 	}))