Fix scheduler (#1359)

yangzq50 · web-flow · commit 4c33402e4b64 · 2024-06-20T18:06:37.000+08:00
### What problem does this PR solve?

Support cpu_limit option in infinity_conf.toml

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
diff --git a/src/scheduler/task_scheduler.cpp b/src/scheduler/task_scheduler.cpp
@@ -58,14 +58,15 @@ void TaskScheduler::Init(Config *config_ptr) {
         cpu_select_step = 1;
     }
 
-    for (u64 cpu_id = 0; cpu_id < worker_count_; cpu_id += cpu_select_step) {
+    for (u64 cpu_id = 0, worker_id = 0; worker_id < worker_count_; ++worker_id) {
         UniquePtr<FragmentTaskBlockQueue> worker_queue = MakeUnique<FragmentTaskBlockQueue>();
-        UniquePtr<Thread> worker_thread = MakeUnique<Thread>(&TaskScheduler::WorkerLoop, this, worker_queue.get(), cpu_id);
+        UniquePtr<Thread> worker_thread = MakeUnique<Thread>(&TaskScheduler::WorkerLoop, this, worker_queue.get(), worker_id);
         // Pin the thread to specific cpu
         ThreadUtil::pin(*worker_thread, cpu_id % cpu_count);
 
         worker_array_.emplace_back(cpu_id, std::move(worker_queue), std::move(worker_thread));
-        worker_workloads_[cpu_id] = 0;
+        worker_workloads_[worker_id] = 0;
+        cpu_id += cpu_select_step;
     }
 
     if (worker_array_.empty()) {
