如何用国外网站做头条,wordpress页面调用分类文章,廊坊seo网站管理,重庆建站免费模板tokio 是 rust 生态中流行的异步运行时框架。在实际生产中我们如果希望 tokio 应用程序与特定的 cpu core 绑定该怎么处理呢#xff1f;这次我们来聊聊这个话题。
首先我们先写一段简单的多任务程序。
use tokio::runtime;
pub fn main() {let rt runtime::Builder::new_mu…
tokio 是 rust 生态中流行的异步运行时框架。在实际生产中我们如果希望 tokio 应用程序与特定的 cpu core 绑定该怎么处理呢这次我们来聊聊这个话题。
首先我们先写一段简单的多任务程序。
use tokio::runtime;
pub fn main() {let rt runtime::Builder::new_multi_thread().enable_all().build().unwrap();rt.block_on(async {for i in 0..8 {println!(num {}, i);tokio::spawn(async move {loop {let mut sum: i32 0;for i in 0..100000000 {sum sum.overflowing_add(i).0;}println!(sum {}, sum);}});}});
}
程序非常简单首先构造一个 tokio runtime 环境然后派生多个 tokio 并发每个并发执行一个无限循环做 overflowing_add。overflowing_add 函数返回一个加法的元组以及一个表示是否会发生算术溢出的布尔值。如果会发生溢出那么将返回包装好的值。然后取元祖的第一个元素打印。
这个程序运行在 Ubuntu 20 OS4 core cpu。通过 nmon 的监控如下 可以看到每个 core 都有负载。
要想把负载绑定在某一 core 上需要使用 core_affinity_rs。core_affinity_rs 是一个用于管理 CPU 亲和力的 Rust crate。目前支持 Linux、Mac OSX 和 Windows。官方宣称支持多平台本人只做了 linux 操作系统的测试。
我们把代码修改一下
use tokio::runtime;pub fn main() {let core_ids core_affinity::get_core_ids().unwrap();println!(core num {}, core_ids.len());let core_id core_ids[1];let rt runtime::Builder::new_multi_thread().on_thread_start(move || {core_affinity::set_for_current(core_id.clone());}).enable_all().build().unwrap();rt.block_on(async {for i in 0..8 {println!(num {}, i);tokio::spawn(async move { loop {let mut sum: i32 0;for i in 0..100000000 {sum sum.overflowing_add(i).0;}println!(sum {}, sum); }});}});
}
在构建多线程 runtime 时在 on_thread_start 设置 cpu 亲和。可以看到负载被绑定到了指定的 core 上。 上面的代码只是把负载绑定到了一个 core 上那么要绑定多个核怎么办呢 我们看看下面的代码
pub fn main() {let core_ids core_affinity::get_core_ids().unwrap();println!(core num {}, core_ids.len());let rt 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);}});}});
}
代码需要把所有负载绑在 core3 和 core4 上。原理是在派生任务中加入 core_affinity 设置。通过调整 idx将派生并发平均绑定在指定的 core 上。代码运行的监控如下图。
