Pin

Future에 대해 await를 호출하여 그 future가 준비되기를 기다릴 때, 모든 로컬 변수(일반적으로 스택 프레임에 저장됨)는 그 future객체 안에 저장됩니다. 만약 그 future에 스택에 있는 어떤 데이터로의 포인터가 있으면 이러한 포인터는 올바르지 않을 수 있습니다. 안전하지 않습니다.

따라서 future가 가리키는 주소가 변경되지 않도록 해야 합니다. 이것이 future를 pin(고정)해야 하는 이유입니다. select!에서 동일한 future를 반복적으로 사용하면 고정 값 문제가 발생하는 경우가 많습니다.

use tokio::sync::{mpsc, oneshot};
use tokio::task::spawn;
use tokio::time::{sleep, Duration};

// A work item. In this case, just sleep for the given time and respond
// with a message on the `respond_on` channel.
#[derive(Debug)]
struct Work {
    input: u32,
    respond_on: oneshot::Sender<u32>,
}

// A worker which listens for work on a queue and performs it.
async fn worker(mut work_queue: mpsc::Receiver<Work>) {
    let mut iterations = 0;
    loop {
        tokio::select! {
            Some(work) = work_queue.recv() => {
                sleep(Duration::from_millis(10)).await; // Pretend to work.
                work.respond_on
                    .send(work.input * 1000)
                    .expect("failed to send response");
                iterations += 1;
            }
            // TODO: report number of iterations every 100ms
        }
    }
}

// A requester which requests work and waits for it to complete.
async fn do_work(work_queue: &mpsc::Sender<Work>, input: u32) -> u32 {
    let (tx, rx) = oneshot::channel();
    work_queue
        .send(Work {
            input,
            respond_on: tx,
        })
        .await
        .expect("failed to send on work queue");
    rx.await.expect("failed waiting for response")
}

#[tokio::main]
async fn main() {
    let (tx, rx) = mpsc::channel(10);
    spawn(worker(rx));
    for i in 0..100 {
        let resp = do_work(&tx, i).await;
        println!("work result for iteration {i}: {resp}");
    }
}