beet_net 0.0.3

A very flexible AI behavior library for games and robotics.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

use flume::Receiver;
use flume::TryRecvError;


#[extend::ext(name=FlumeReceiverExt)]
pub impl<T> Receiver<T> {
	fn try_recv_all(&mut self) -> anyhow::Result<Vec<T>> {
		let mut vec = Vec::new();
		loop {
			match self.try_recv() {
				Ok(message) => vec.push(message),
				Err(TryRecvError::Empty) => break Ok(vec),
				Err(other) => anyhow::bail!(other),
			}
		}
	}
}
#[extend::ext(name=FlumeReceiverExtFlat)]
pub impl<T> Receiver<Vec<T>> {
	fn try_recv_all_flat(&mut self) -> anyhow::Result<Vec<T>> {
		let mut vec = Vec::new();
		loop {
			match self.try_recv() {
				Ok(message) => vec.extend(message),
				Err(TryRecvError::Empty) => break Ok(vec),
				Err(other) => anyhow::bail!(other),
			}
		}
	}
}