fix method of getting cpu nums

- related PRs: tikv#8425, tikv#8427 Signed-off-by: baurine <2008.hbl@gmail.com>
ti-srebot · Sep 21, 2020 · 6efdb72 · 6efdb72
1 parent 11d0f85
commit 6efdb72
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/components/tikv_util/src/sys/mod.rs b/components/tikv_util/src/sys/mod.rs
@@ -22,7 +22,7 @@ pub mod sys_quota {
         }
 
         pub fn cpu_cores_quota(&self) -> f64 {
-            let cpu_num = sysinfo::get_logical_cores() as f64;
+            let cpu_num = num_cpus::get() as f64;
             match self.cgroup.cpu_cores_quota() {
                 Some(cgroup_quota) if cgroup_quota > 0.0 && cgroup_quota < cpu_num => cgroup_quota,
                 _ => cpu_num,
@@ -64,7 +64,7 @@ pub mod sys_quota {
         }
 
         pub fn cpu_cores_quota(&self) -> f64 {
-            sysinfo::get_logical_cores() as f64
+            num_cpus::get() as f64
         }
 
         pub fn memory_limit_in_bytes(&self) -> u64 {