Return if channel is closed

vardius · vardius · commit 4b5279aa0c52 · 2019-05-10T17:32:54.000+10:00
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
 				}

Original file line number	Diff line number	Diff line change
`@@ -47,9 +47,11 @@ func (p *pool) Start(maxWorkers int, fn interface{}) error {`
`47`	`47`	`for {`
`48`	`48`	`select {`
`49`	`49`	`case args, ok := <-p.queue:`
`50`		`- if ok {`
`51`		`- h.Call(args)`
	`50`	`+ if !ok {`
	`51`	`+ return`
`52`	`52`	`}`
	`53`	`+`
	`54`	`+ h.Call(args)`
`53`	`55`	`case <-p.ctx.Done():`
`54`	`56`	`return`
`55`	`57`	`}`