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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
use crate::protocol::{ClientboundPacket, ServerboundPacket};
use futures_util::{SinkExt, StreamExt, TryFutureExt};
use log::error;
use std::collections::HashMap;
use tokio::sync::{mpsc, RwLock};
use warp::ws::{Message, WebSocket};
#[derive(Debug)]
pub struct Client {
pub name: String,
pub out: mpsc::UnboundedSender<ClientboundPacket>,
}
#[derive(Debug, Default)]
pub struct Room {
pub clients: RwLock<HashMap<usize, Client>>,
}
impl Room {
pub async fn client_connect(&self, ws: WebSocket) {
let (mut user_ws_tx, mut user_ws_rx) = ws.split();
let (tx, mut rx) = mpsc::unbounded_channel();
tokio::task::spawn(async move {
while let Some(packet) = rx.recv().await {
user_ws_tx
.send(Message::text(serde_json::to_string(&packet).unwrap()))
.unwrap_or_else(|e| {
eprintln!("websocket send error: {}", e);
})
.await;
}
});
let mut g = self.clients.write().await;
let id = g.len();
let name = format!("user no. {id}");
g.insert(
id,
Client {
out: tx,
name: name.clone(),
},
);
drop(g);
self.broadcast(id, ClientboundPacket::ClientJoin { id, name })
.await;
while let Some(result) = user_ws_rx.next().await {
let msg = match result {
Ok(msg) => msg,
Err(e) => {
error!("websocket error(id={id}): {e}");
break;
}
};
if let Ok(s) = msg.to_str() {
let p = match serde_json::from_str::<ServerboundPacket>(s) {
Ok(p) => p,
Err(e) => {
error!("client(id={id}) sent invalid packet: {e:?}");
break;
}
};
self.client_message(id, p).await;
};
}
self.clients.write().await.remove(&id);
}
pub async fn broadcast(&self, sender: usize, packet: ClientboundPacket) {
for (&id, tx) in self.clients.read().await.iter() {
if sender != id {
let _ = tx.out.send(packet.clone());
}
}
}
pub async fn send_to_client(&self, recipient: usize, packet: ClientboundPacket) {
if let Some(c) = self.clients.read().await.get(&recipient) {
let _ = c.out.send(packet);
}
}
pub async fn client_message(&self, sender: usize, packet: ServerboundPacket) {
match packet {
ServerboundPacket::Relay { recipient, message } => {
if let Some(recipient) = recipient {
self.send_to_client(recipient, ClientboundPacket::Message { sender, message })
.await;
}
}
}
}
pub async fn should_remove(&self) -> bool {
self.clients.read().await.len() == 0
}
}
|
