任务

Rust 有一个任务系统,这是一种轻量级线程处理形式。

每个任务只有一个顶级 Future,执行器会对此进行轮询来推进任务进度。该 Future 可能包含一个或多个嵌套的 Future,可以通过其 poll 方法对它们进行轮询,类似于调用堆栈。可以通过轮询多个子 Future(例如争用定时器和 I/O 操作)在任务内部实现并发操作。

  1. use tokio::io::{self, AsyncReadExt, AsyncWriteExt};
  2. use tokio::net::TcpListener;
  3. #[tokio::main]
  4. async fn main() -> io::Result<()> {
  5. let listener = TcpListener::bind("127.0.0.1:0").await?;
  6. println!("listening on port {}", listener.local_addr()?.port());
  7. loop {
  8. let (mut socket, addr) = listener.accept().await?;
  9. println!("connection from {addr:?}");
  10. tokio::spawn(async move {
  11. socket.write_all(b"Who are you?\n").await.expect("socket error");
  12. let mut buf = vec![0; 1024];
  13. let name_size = socket.read(&mut buf).await.expect("socket error");
  14. let name = std::str::from_utf8(&buf[..name_size]).unwrap().trim();
  15. let reply = format!("Thanks for dialing in, {name}!\n");
  16. socket.write_all(reply.as_bytes()).await.expect("socket error");
  17. });
  18. }
  19. }

将此示例复制到准备好的 src/main.rs 文件中,并从该文件运行它。

请尝试使用像 nctelnet 这样的 TCP 连接工具进行连接。

  • 让学生想象一下,当连接多个客户端时,示例服务器会达到怎样的状态。存在哪些任务?具有哪些 Future?

  • This is the first time we’ve seen an async block. This is similar to a closure, but does not take any arguments. Its return value is a Future, similar to an async fn.

  • Refactor the async block into a function, and improve the error handling using ?.