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 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
extern crate log; use log::{info}; use std::sync::mpsc::{channel, Sender, Receiver}; use std::collections::HashMap; use std::time::Duration; #[derive(Clone)] pub struct Envelope { pub to: u64, pub msg: String } impl Envelope { pub fn new(to: u64, msg: String) -> Envelope { Envelope { to, msg } } } pub struct MessageChannel { pub addr: u64, pub tx: Sender<Envelope>, pub rx: Receiver<Envelope> } impl MessageChannel { pub fn new(addr: u64) -> MessageChannel { let (tx, rx) = channel(); MessageChannel { addr, tx, rx } } } pub struct Bus { channels: HashMap<u64,MessageChannel> } impl Bus { pub fn new() -> Bus { Bus { channels: HashMap::new() } } pub fn register(&mut self, addr: u64) { self.channels.insert(addr, MessageChannel::new(addr)); } pub fn register_all(&mut self, addr: [u64;2]) { info!("register {:?}",addr); for i in 0..1 { self.channels.insert(addr[i], MessageChannel::new(addr[i])); } } pub fn send(&mut self, env: Envelope) -> bool { match self.channels.get(&env.to) { Some(ch) => { match ch.tx.send(env) { Ok(()) => true, Err(_) => false } }, None => false } } pub fn poll(&mut self, addr: u64) -> Option<Envelope> { match self.channels.get(&addr) { Some(ch) => { match ch.rx.recv_timeout(Duration::from_millis(100)) { Ok(env) => Option::Some(env), Err(_) => Option::None } }, None => Option::None } } }