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
|
use std::io::{BufRead, Write};
#[derive(Debug, Clone)]
pub struct Message {
pub event: lexpr::Value,
pub data: lexpr::Value,
}
pub struct Client {
reader: std::io::BufReader<std::net::TcpStream>,
}
impl Client {
pub fn new(addr: &str, subs: &[lexpr::Value]) -> Self {
let mut socket = std::net::TcpStream::connect(addr).expect("failed to connect to message bus");
socket.set_nonblocking(true).expect("failed to set message bus socket nonblocking");
for s in subs {
write!(socket, "(sub {})\n", s).expect("failed to send subscribe message to bus");
}
let reader = std::io::BufReader::new(socket);
Self { reader, }
}
pub fn pump(&mut self) -> Option<Message> {
let mut buf = String::new();
match self.reader.read_line(&mut buf) {
Ok(l) => match lexpr::from_str(&buf) {
Ok(v) => {
match v.as_cons() {
Some(cs) => {
Some(Message { event: cs.car().clone(), data: cs.cdr().clone() })
},
_ => { log::error!("malformed message bus input s-expression: {}", v); None },
}
},
Err(e) => { log::error!("malformed message bus input line: {}", e); None },
},
Err(e) if e.kind() == std::io::ErrorKind::WouldBlock => None,
Err(e) => panic!("IO error on message bus: {}", e),
}
}
}
|
