fix: fix task_pool AddTaskAlways

eeb1680e · watermelon · 0cdef164 · eeb1680e
Commit eeb1680e authored Nov 22, 2020 by watermelon
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

task_pool.go sync/task_pool.go +6 -0

No files found.
--- a/sync/task_pool.go
+++ b/sync/task_pool.go
@@ -267,6 +267,12 @@ func (p *taskPoolSimple) AddTask(t task) bool {

 func (p *taskPoolSimple) AddTaskAlways(t task) {
 	select {
+	case <-p.done:
+		return
+	default:
+	}
+
+	select {
 	case p.work <- t:
 		return
 	default: