wait_task busy loop before waiting on PL

apache · Oct 13, 2023 · 0ab0b69 · 0ab0b69
1 parent e5f06dc
commit 0ab0b69
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/bthread/task_group.cpp b/src/bthread/task_group.cpp
@@ -116,17 +116,23 @@ bool TaskGroup::is_stopped(bthread_t tid) {
 }
 
 bool TaskGroup::wait_task(bthread_t* tid) {
+    int64_t poll_start_ms = butil::cpuwide_time_ms();
     do {
 #ifndef BTHREAD_DONT_SAVE_PARKING_STATE
         if (_last_pl_state.stopped()) {
             return false;
         }
 
-        if (pop_resume_task(tid)) {
+        if (pop_resume_task(tid) || steal_task(tid)) {
             return true;
         }
 
-        _pl->wait(_last_pl_state);
+        // keep polling for some time before waiting on parking lot
+        if (butil::cpuwide_time_ms() - poll_start_ms > 100) {
+            _pl->wait(_last_pl_state);
+            poll_start_ms = butil::cpuwide_time_ms();
+        }
+
         if (steal_task(tid)) {
             return true;
         }