rust学习(tokio协程分析二)

例子:

我们如果使用new_current_thread来创建tokio的协程运行runtime时,

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

发现只有调用rt.block_on(...)才能触发。这里我们分析一下为何在new_current_thread的runtime下无法运行的原因。

代码1

rust 复制代码
fn testCoroutine4() {
    thread::spawn(||{
        let rt = tokio::runtime::Builder::new_current_thread().enable_all().build().unwrap();
        let guard1 = rt.enter();
        println!("trace1", );
        rt.spawn(doSayHi());
        println!("trace2", );
        rt.spawn(doSayHi());
        println!("trace3", );
    }).join();
}

这个无法运行的情况是由于执行完最后代码println!("trace3", );之后,线程直接就退出了,没有机会做调用栈切换。

根据代码1的情况,我们考虑是否可以保持线程不退出(如代码2),这样是否可以执行到协程(doSayHi)呢?

代码2:

rust 复制代码
fn testCoroutine5() {
    thread::spawn(||{
        let rt = tokio::runtime::Builder::new_current_thread().enable_all().build().unwrap();
        let guard1 = rt.enter();
        println!("trace1", );
        rt.spawn(doSayHi());
        println!("trace2", );
        rt.spawn(doSayHi());
        println!("trace3", );

        thread::sleep(Duration::from_secs(10));
    }).join();
}

和遗憾,还是没有运行doSayHi,这是为什么呢?因为thread::sleep直接把线程sleep了,协程的调用栈也没有被切换。

我们来看一下可以运行的代码:

代码3:

rust 复制代码
fn testCoroutine6() {
    thread::spawn(||{
        let rt = tokio::runtime::Builder::new_current_thread().enable_all().build().unwrap();
        let guard1 = rt.enter();
        println!("trace1", );
        let v1 = rt.spawn(doSayHi());
        println!("trace2", );
        let v2 = rt.spawn(doSayHi());
        println!("trace3", );

        rt.block_on(v1);
        rt.block_on(v2);
    }).join();
}

输出:

我们可以看到可以正常打印了。所以我们来想一下协程需要依赖哪些情况才能运行:

1.没有退出的线程

2.有切换调用栈的入口

按照上面的思路,我们是否不用block_on(v1)也能保证协程运行能?其实是可以的,参看

rust 复制代码
fn testCoroutine4() {
    thread::spawn(||{
        let rt = tokio::runtime::Builder::new_current_thread().enable_all().build().unwrap();
        let guard1 = rt.enter();
        println!("trace1", );
        rt.spawn(doSayHi());
        println!("trace2", );
        rt.spawn(doSayHi());
        println!("trace3", );

        loop {
            rt.block_on(async {
                println!("sleep start", );
                time::sleep(Duration::from_secs(10)).await;
                println!("sleep end", );
            });
        }
    }).join();
}

哈哈,可以看到,线程最后启动了一个和之前毫无关系的协程代码,里面只是一个sleep(注意!!这个是tokio的sleep),这个sleep的作用就是满足线程有一个协程切换的入口点。我们运行一下看看结果:

正常运行。不过这个代码只是用来验证我们对协程的认识,不建议正式代码中使用。

总结:

之前用过C语言的libco,它的原理是通过注册了libc的大部分posix的io操作函数,然后通过epoll来实现异步io来实现。例如读取socket的数据,如果没有数据,由于是异步io,所以直接切换到下一个调用栈。这样就实现了简单的协程。rust这块貌似对整个协程做了统一的接口管理?在协程中的sleep等操作一定要调用tokio的接口,否则就会直接造成卡死。哈哈。不过如果是multi_thread的话,实际上也没啥关系,就是直接将协程变成了线程操作,程序应该不会出现卡死。

相关推荐
用户967151139167221 小时前
Rust 如何轻松实现 RTMP 流媒体推送?深入解析直播推流场景与解决方案
rust·ffmpeg
云上艺旅21 小时前
K8S学习之基础七十四:部署在线书店bookinfo
学习·云原生·容器·kubernetes
无名之逆1 天前
Rust 开发提效神器:lombok-macros 宏库
服务器·开发语言·前端·数据库·后端·python·rust
s9123601011 天前
rust 同时处理多个异步任务
java·数据库·rust
你觉得2051 天前
哈尔滨工业大学DeepSeek公开课:探索大模型原理、技术与应用从GPT到DeepSeek|附视频与讲义下载方法
大数据·人工智能·python·gpt·学习·机器学习·aigc
A旧城以西1 天前
数据结构(JAVA)单向,双向链表
java·开发语言·数据结构·学习·链表·intellij-idea·idea
无所谓จุ๊บ1 天前
VTK知识学习(50)- 交互与Widget(一)
学习·vtk
FAREWELL000751 天前
C#核心学习(七)面向对象--封装(6)C#中的拓展方法与运算符重载: 让代码更“聪明”的魔法
学习·c#·面向对象·运算符重载·oop·拓展方法
吴梓穆1 天前
UE5学习笔记 FPS游戏制作38 继承标准UI
笔记·学习·ue5
Three~stone1 天前
MySQL学习集--DDL
数据库·sql·学习