fix bug : panic: send on closed channel

workpool/def.go

@@ -13,6 +13,7 @@ type TaskHandler func() error
 // WorkPool serves incoming connections via a pool of workers
 type WorkPool struct {
 	closed       int32
+	isQueTask    bool          // Mark whether queue retrieval is task.标记是否队列取出任务
 	errChan      chan error    // error chan
 	timeout      time.Duration // max timeout
 	wg           sync.WaitGroup

workpool/workpool.go

@@ -2,6 +2,7 @@ package workpool
 
 import (
 	"context"
+	"runtime"
 	"sync/atomic"
 	"time"
 
@@ -55,6 +56,7 @@ func (p *WorkPool) DoWait(task TaskHandler) { // 添加到工作池，并等待
 // Wait Waiting for the worker thread to finish executing
 func (p *WorkPool) Wait() error { // 等待工作线程执行结束
 	p.waitingQueue.Wait() // 等待队列结束
+	p.waitQueTask()       // wait que down
 	close(p.task)
 	p.wg.Wait() // 等待结束
 	select {
@@ -85,14 +87,26 @@ func (p *WorkPool) IsClosed() bool { // 是否已经关闭
 func (p *WorkPool) startQueue() {
 	for {
 		fn := p.waitingQueue.Pop().(TaskHandler)
+		p.isQueTask = true
 		if p.IsClosed() { // closed
 			p.waitingQueue.Close()
+			p.isQueTask = false
 			break
 		}
 
 		if fn != nil {
 			p.task <- fn
 		}
+		p.isQueTask = false
+	}
+}
+
+func (p *WorkPool) waitQueTask() {
+	for {
+		runtime.Gosched() // 出让时间片
+		if !p.isQueTask {
+			break
+		}
 	}
 }