feat: add tokio-cpu-affinity

2024-03-24 11:32:07 +08:00
parent a3c45895e1
commit 31ba44c45e
4 changed files with 530 additions and 0 deletions
--- a/__std/tokio-cpu-affinity/src/bin/bind-multiple.rs
+++ b/__std/tokio-cpu-affinity/src/bin/bind-multiple.rs
@@ -0,0 +1,35 @@
+pub fn main() {
+    let core_ids = core_affinity::get_core_ids().unwrap();
+    println!("core num {}", core_ids.len());
+
+    let rt = tokio::runtime::Builder::new_multi_thread()
+        .enable_all()
+        .build()
+        .unwrap();
+
+    let mut idx = 2;
+
+    rt.block_on(async {
+        for i in 0..8 {
+            println!("num {}", i);
+            let core_id = core_ids[idx];
+            if idx.eq(&(core_ids.len() - 1)) {
+                idx = 2;
+            } else {
+                idx += 1;
+            }
+
+            tokio::spawn(async move {
+                let res = core_affinity::set_for_current(core_id);
+                println!("{}", res);
+                loop {
+                    let mut sum: i32 = 0;
+                    for i in 0..100000000 {
+                        sum = sum.overflowing_add(i).0;
+                    }
+                    println!("sum {}", sum);
+                }
+            });
+        }
+    });
+}