如何在没有不稳定功能或外部包装箱的情况下从 mpsc::channel 读取指定的时间？

Question

我正在尝试在指定的持续时间内连续读取 Receiver。我提出了以下解决方案

pub fn get<T>(
    rx: &Receiver<T>,
    get_duration: time::Duration,
) -> Result<(), Err> {
    let (dur_tx, dur_rx) = channel();
    let _ = thread::spawn(move || {
        // timer to kill receiving
        thread::sleep(get_duration);
        let _ = dur_tx.send("tick");
    });

    let mut time_to_break = false;
    while time_to_break == false {
        match rx.try_recv() {
            Ok(resp) => {
                //...
            }
            Err(_) => ()
        }
        thread::sleep(time::Duration::from_millis(1)); // avoid using cpu 100%
        let _ = dur_rx.try_recv().map(|_| time_to_break = true);
    }
    Ok(())
}

有没有更好的解决方法，没有不稳定或弃用的功能（例如 select）或外部板条箱？

Answer 1

你可以用 recv_timeout 来做到这一点，它也应该更整洁：

pub fn get<T>(rx: &Receiver<T>, get_duration: Duration) -> Result<(), Error> {
    let start_time = Instant::now();
    loop {
        let now = Instant::now();
        if start_time + get_duration > now {
            let duration = start_time + get_duration - now;
            match rx.recv_timeout(duration) {
                Ok(resp) => {
                    // handle message
                }
                Err(RecvTimeoutError::Timeout) => break,
                Err(RecvTimeoutError::Disconnected) => {
                    // handle disconnect
                }
            }
        } else {
            break;
        }
    }
    Ok(())
}

当 recv_deadline 稳定后，您可以使这更简单，因为您只需计算一次截止日期，然后继续通过它，而不是每次迭代都重新计算新的持续时间。

编辑： 当 recv_deadline 变得稳定时为什么要使用它的一个典型例子是我刚刚给你的代码中的一个错误。 Duration一直是正数，但是我原来的计算没有检查防止下溢，会造成间歇性的panic。我现在已经更新它以确保不会发生这种情况。

下面是使用截止日期（在 Nightly Rust 中）的样子：

#![feature(deadline_api)]

pub fn get<T>(rx: &Receiver<T>, get_duration: Duration) -> Result<(), Error> {
    let deadline = Instant::now() + get_duration;
    loop {
        match rx.recv_deadline(deadline) {
            Ok(resp) => {
                // handle message
            }
            Err(RecvTimeoutError::Timeout) => break,
            Err(RecvTimeoutError::Disconnected) => {
                // handle disconnect
            }
        }
    }
    Ok(())
}

如何在没有不稳定功能或外部包装箱的情况下从 mpsc::channel 读取指定的时间？

How do I read from a mpsc::channel for a specified amount of time without unstable features or external crates?

timeout

channel

rust