From e396721ebd1b71162e424fdd4f7670c657667a60 Mon Sep 17 00:00:00 2001
From: Guilhem Lettron <guilhem@barpilot.io>
Date: Tue, 21 Sep 2021 12:21:30 +0200
Subject: [PATCH] feat: add afterFunc behavior

Useful to catch errors in a infinit worker pool
---
 workers.go | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/workers.go b/workers.go
index dc36433..63050a9 100644
--- a/workers.go
+++ b/workers.go
@@ -237,11 +237,10 @@ func (r *runner) startWork() {
 					<-r.limiter
 					workerWG.Done()
 				}()
-				if err := r.workFunc(input, r.outChan); err != nil {
+				if err := r.afterFunc(r.ctx, r.workFunc(input, r.outChan)); err != nil {
 					r.once.Do(func() {
 						r.errChan <- err
 						r.cancel()
-						return
 					})
 				}
 			}()