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