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
|
use karlcommon::{ServerboundPacket, ProtoError};
use log::{debug, error, info};
use std::net::{TcpListener, TcpStream};
use std::thread;
use tungstenite::{accept, Message};
use crate::{handle_packet, CLIENT_ID_COUNTER};
pub fn run() {
info!("binding websocket server");
let server = TcpListener::bind("127.0.0.1:9001").unwrap();
info!("listening");
for stream in server.incoming() {
thread::spawn(move || {
if let Err(e) = handle_connection(stream) {
error!("{e}");
}
});
}
}
fn handle_connection(stream: Result<TcpStream, std::io::Error>) -> anyhow::Result<()> {
let id = CLIENT_ID_COUNTER.fetch_add(1, std::sync::atomic::Ordering::Relaxed);
let stream = stream?;
stream.set_nonblocking(true)?;
let mut websocket = accept(stream)?;
let (responder, responses) = crossbeam_channel::unbounded();
loop {
match websocket.read_message() {
Ok(Message::Text(t)) => {
debug!("<- {t:?}");
match serde_json::from_str::<ServerboundPacket>(&t) {
Ok(packet) => {
handle_packet(id, packet, responder.clone())
},
Err(e) => {
responder.send(karlcommon::ClientboundPacket::Error(ProtoError::FormatError(format!("{e}"))))?
},
}
}
Ok(_) => (),
Err(tungstenite::Error::ConnectionClosed) => {
break Ok(());
}
Err(tungstenite::Error::Io(e))
if let std::io::ErrorKind::WouldBlock = e.kind() => {
// its fine^
}
Err(e) => Err(e)?,
}
for r in responses.try_iter() {
websocket.write_message(Message::Text( serde_json::to_string(&r)?))?;
}
thread::sleep(std::time::Duration::from_millis(50)); // how would you do this properly??
}
}
|
