码迷,mamicode.com
首页 > 其他好文 > 详细

rust 实现协程池

时间:2021-07-19 16:57:38      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:highlight   oss   handle   async   wait   receiver   ann   使用   协程   

use crossbeam_channel::{Receiver, bounded};
use tokio::time::{Duration, delay_for};

#[tokio::main]
async fn main() {
    let (s, r) = bounded(10);

    for i in 0..100 {
        s.send(i).unwrap();
        let rx = r.clone();
        tokio::task::spawn(async move {
            handle(i, rx).await;
        });
    }

    delay_for(Duration::from_secs(10)).await;
}

async fn handle(i: i32, rx: Receiver<i32>) {
    let _d = defer::defer(|| {
        println!("{} been defer", i);
        if let Err(err) = rx.recv() {
            println!("err : {:?}", err);
        }
    });

    delay_for(Duration::from_secs(1)).await;
    println!("{} been excute", i);
}

 

使用 channel 做数量控制, defer 来保证 所有 消息被释放

 

rust 实现协程池

标签:highlight   oss   handle   async   wait   receiver   ann   使用   协程   

原文地址:https://www.cnblogs.com/mafeng/p/15029791.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有
迷上了代码!