rust学习(tokio协程分析一)

代码:

rust 复制代码
async fn doAsyncPrint(v:u32) {
    println!("start doAsyncPrint,v is {},tid is {:?}",v,system::myTid());
    //thread::sleep(Duration::from_secs(1));
    time::sleep(Duration::from_secs(10)).await;
    println!("end,v is {},tid is {:?}",v,system::myTid());
}

fn testCoroutine3() {
    println!("testCoroutine3 start");
    let array = Arc::new(BlockingQueue::<u32>::new());

    let arrayclosure1 = array.clone();
    thread::spawn(move|| {
        let mut count:u32 = 0;
        loop {
            //println!("thread start,count is {}",count);
            arrayclosure1.putLast(count);
            count += 1;
            thread::sleep(Duration::from_secs(1));
        }
    });

    let arrayclosure2 = array.clone();
    thread::spawn(move ||{
        let rt = tokio::runtime::Builder::new_current_thread().enable_all().build().unwrap();
        let guard1 = rt.enter();
        println!("coroutine start,tid is {:?}", system::myTid());
        loop {
            let v = arrayclosure2.takeFirst();
            //println!("coroutine trace1,v is {}",v);
            tokio::spawn(doAsyncPrint(v));
        }
        
    }).join();
}

代码运行时会直接卡住,这个和协程的原理有关:

1.let rt = tokio::runtime::Builder::new_current_thread().enable_all().build().unwrap();

这里创建一个协程的runtime,注意:这里协程是跑在当前的线程中

2.let v = arrayclosure2.takeFirst();

从blocking队列中获取数据,注意这个blocking队列使用了mutex和condvar,所以当condvar卡住的时候,实际上是整个线程直接卡住,由于协程只是在线程中来回切换调用栈,所以协程中所有的等待处理实际上都是切换调用栈,但是这个condvar不是切换调用栈,所以就会卡住。

3.tokio::spawn(doAsyncPrint(v));

doAsyncPrint一直没有被调用

方案:

将new_current_thread改成new_multi_thread

代码如下:

rust 复制代码
async fn doAsyncPrint(v:u32) {
    println!("start doAsyncPrint,v is {},tid is {:?}",v,system::myTid());
    //thread::sleep(Duration::from_secs(1));
    time::sleep(Duration::from_secs(10)).await;
    println!("end,v is {},tid is {:?}",v,system::myTid());
}

thread::spawn(move ||{
        let rt = tokio::runtime::Builder::new_multi_thread().worker_threads(1).enable_all().build().unwrap();
        let guard1 = rt.enter();
        println!("coroutine start,tid is {:?}", system::myTid());
        loop {
            let v = arrayclosure2.takeFirst();
            //println!("coroutine trace1,v is {}",v);
            tokio::spawn(doAsyncPrint(v));
        }
        
    }).join();

这样实际上协程的runtime会创建线程,让协程跑在这些线程上,如上述代码,我们通过worker_threads创建了一个线程,所有的协程都会跑在这个线程上。

我们看一下运行结果:

你会发现,所有的协程都跑在了thread4上,我们看一下代码:

rust 复制代码
async fn doAsyncPrint(v:u32) {
    println!("start doAsyncPrint,v is {},tid is {:?}",v,system::myTid());
    //thread::sleep(Duration::from_secs(1));
    time::sleep(Duration::from_secs(10)).await;
    println!("end,v is {},tid is {:?}",v,system::myTid());
}

发现没有,一开始的时候连续打印了多个"start doAsyncPrint",但是没有打印"end,v is ...."这个是为啥咧?原因就在time::sleep这个函数(tokio::time::sleep),实际上,这个sleep没有真让线程sleep,而且直接切换调用栈,切换到下一个函数上,所以一开始我们可以看到连续的"start doAsyncPrint"。

协程的使用还是比较要当心的,所有操作io操作啥估计都要使用tokio的接口,否者就会导致卡住。例如上面的代码,如果改成thread::sleep的话,就会直接卡住线程,输出结果就如下:

相关推荐
大数据编程之光18 分钟前
Flink Standalone集群模式安装部署全攻略
java·大数据·开发语言·面试·flink
爪哇学长31 分钟前
双指针算法详解:原理、应用场景及代码示例
java·数据结构·算法
风尚云网35 分钟前
风尚云网前端学习:一个简易前端新手友好的HTML5页面布局与样式设计
前端·css·学习·html·html5·风尚云网
ExiFengs35 分钟前
实际项目Java1.8流处理, Optional常见用法
java·开发语言·spring
paj12345678937 分钟前
JDK1.8新增特性
java·开发语言
繁依Fanyi1 小时前
简易安卓句分器实现
java·服务器·开发语言·算法·eclipse
慧都小妮子1 小时前
Spire.PDF for .NET【页面设置】演示:打开 PDF 时自动显示书签或缩略图
java·pdf·.net
m51271 小时前
LinuxC语言
java·服务器·前端
IU宝1 小时前
C/C++内存管理
java·c语言·c++
瓜牛_gn1 小时前
依赖注入注解
java·后端·spring