fix: polling cpu load

xxjwxc · Jun 12, 2023 · e4b4bfd · e4b4bfd
1 parent 4fbf750
commit e4b4bfd
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/workpool/workpool.go b/workpool/workpool.go
@@ -104,13 +104,23 @@ func (p *WorkPool) startQueue() {
 }
 
 func (p *WorkPool) waitTask() {
+	var c int
+
 	for {
-		runtime.Gosched() // 出让时间片
+		c++
+
 		if p.IsDone() {
 			if atomic.LoadInt32(&p.isQueTask) == 0 {
 				break
 			}
 		}
+
+		if c <= 100 {
+			runtime.Gosched() // 出让时间片
+			continue
+		}
+
+		time.Sleep(20 * time.Microsecond) // 20us, cpu cost < 1%
 	}
 }