タスク

Rust にはタスクシステムがあり、これは軽量スレッディングの一形態です。

タスクは単一のトップレベル Future を持ち、executor はそれをポーリングして処理を進めます。その Future は、poll メソッドがポーリングする 1 つ以上のネストされた Future を持つ場合があり、これは大まかにはコールスタックに対応します。タスク内での並行性は、タイマーと I/O 操作を競合させるように複数の子 Future をポーリングすることで実現できます。

// 著作権 2024 Google LLC
// SPDX-License-Identifier: Apache-2.0

use tokio::io::{self, AsyncReadExt, AsyncWriteExt};
use tokio::net::TcpListener;

#[tokio::main]
async fn main() -> io::Result<()> {
    let listener = TcpListener::bind("127.0.0.1:0").await?;
    println!("listening on port {}", listener.local_addr()?.port());

    loop {
        let (mut socket, addr) = listener.accept().await?;

        println!("connection from {addr:?}");

        tokio::spawn(async move {
            socket.write_all(b"Who are you?\n").await.expect("socket error");

            let mut buf = vec![0; 1024];
            let name_size = socket.read(&mut buf).await.expect("socket error");
            let name = std::str::from_utf8(&buf[..name_size]).unwrap().trim();
            let reply = format!("Thanks for dialing in, {name}!\n");
            socket.write_all(reply.as_bytes()).await.expect("socket error");
        });
    }
}

この例を用意した src/main.rs にコピーし、そこから実行してください。

nc や telnet のような TCP 接続ツールを使って接続してみてください。

数人のクライアントが接続しているとき、このサーバーの状態がどうなっているかを受講者にイメージしてもらってください。どのようなタスクが存在しますか？それらの Future は何ですか？
ここで初めて async ブロックを見ました。これはクロージャに似ていますが、引数は取りません。その戻り値は Future で、async fn と似ています。
async ブロックを関数にリファクタリングし、? を使ってエラーハンドリングを改善してください。

Comprehensive Rust 🦀

タスク