add data to error handler

yuyang0 · May 10, 2024 · 7892587 · 7892587
1 parent 22bd2a1
commit 7892587
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/flow/executor.go b/flow/executor.go
@@ -66,13 +66,13 @@ func (e *Executor) Execute(ctx context.Context) error {
 				}
 			}()
 			if node.execOpts.failureHandler != nil {
-				if err1 := node.execOpts.failureHandler(err); err1 != nil {
+				if err1 := node.execOpts.failureHandler(e.Body, err); err1 != nil {
 					return errors.CombineErrors(err, err1)
 				}
 			}
 			if retried >= maxRetry {
 				if node.execOpts.finalFailureHandler != nil {
-					if err2 := node.execOpts.finalFailureHandler(err); err2 != nil {
+					if err2 := node.execOpts.finalFailureHandler(e.Body, err); err2 != nil {
 						return errors.CombineErrors(err, err2)
 					}
 				}

diff --git a/flow/option.go b/flow/option.go
@@ -14,7 +14,7 @@ type Forwarder func([]byte) []byte
 // ForEach definition for the foreach function
 type ForEach func([]byte) map[string][]byte
 
-type FuncErrorHandler func(error) error
+type FuncErrorHandler func([]byte, error) error
 
 type ExecutionOptions struct {
 	aggregator Aggregator