Return if channel is closed

vardius · May 10, 2019 · 4b5279a · 4b5279a
1 parent 0417bd5
commit 4b5279a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pool.go b/pool.go
@@ -47,9 +47,11 @@ func (p *pool) Start(maxWorkers int, fn interface{}) error {
 			for {
 				select {
 				case args, ok := <-p.queue:
-					if ok {
-						h.Call(args)
+					if !ok {
+						return
 					}
+
+					h.Call(args)
 				case <-p.ctx.Done():
 					return
 				}