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

问题描述 投票:1回答:1

我试图连续读取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)或外部板条箱,是否有更好的解决方法?

rust timeout channel
1个回答
2
投票

你可以用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总是积极的,但我原来的计算没有检查以防止下溢,这将导致间歇性的恐慌。我现在更新它以确保不会发生。

这是使用截止日期(在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(())
}
© www.soinside.com 2019 - 2024. All rights reserved.