Fix Issue #3

letsfire · Nov 18, 2019 · 5b15285 · 5b15285
1 parent f1e722c
commit 5b15285
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/issue/0001_test.go b/issue/0001_test.go
@@ -48,6 +48,6 @@ func TestNewMaster(t *testing.T) {
 	fmt.Println(master.Running()) // 正在运行的协程工人数量
 
 	fmt.Println(runtime.NumGoroutine()) // 通常会大于 2
-	time.Sleep(time.Nanosecond)         // 极短暂休眠
+	time.Sleep(time.Millisecond * 5)    // 极短暂休眠
 	fmt.Println(runtime.NumGoroutine()) // 必须会等于 2
 }
diff --git a/worker.go b/worker.go
@@ -3,6 +3,7 @@ package factory
 import (
 	"fmt"
 	"sync/atomic"
+	"time"
 )
 
 type exitSignal struct{}
@@ -60,7 +61,7 @@ func newWorker() (w *worker) {
 		select {
 		case params := <-w.params:
 			w.action(params)
-		default:
+		case <-time.Tick(time.Millisecond):
 		}
 		// 关闭任务通道
 		close(w.params)