perf: gopool enable PreemptOff

joway · Sep 20, 2023 · b52892f · b52892f
1 parent a612b33
commit b52892f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/util/gopool/worker.go b/util/gopool/worker.go
@@ -44,8 +44,6 @@ func newWorker() interface{} {
 func (w *worker) run() {
 	go func() {
 		g := runtimex.GetG()
-		m := g.M()
-		p := m.P()
 		for {
 			var t *task
 			now := runtimex.Nanotime()
@@ -63,6 +61,8 @@ func (w *worker) run() {
 				return
 			}
 
+			m := g.M()
+			p := m.P()
 			// update sysmon tick
 			if w.updateSched {
 				st := *p.Sysmontick()
@@ -81,7 +81,9 @@ func (w *worker) run() {
 							logger.CtxErrorf(t.ctx, msg)
 						}
 					}
+					*(g.M().PreemptOff()) = ""
 				}()
+				*m.PreemptOff() = "holding"
 				t.f()
 			}()
 			t.Recycle()