author | alfadur |
Thu, 16 Aug 2018 19:33:03 +0300 | |
changeset 13671 | 09f4a30e50cc |
parent 13529 | 5359ff75da3a |
child 13803 | 4664da990556 |
permissions | -rw-r--r-- |
12132 | 1 |
use slab; |
13671 | 2 |
use crate::utils; |
13421 | 3 |
use super::{ |
13483 | 4 |
client::HWClient, room::HWRoom, actions, handlers, |
5 |
coretypes::{ClientId, RoomId}, |
|
13424 | 6 |
actions::{Destination, PendingMessage} |
13421 | 7 |
}; |
13671 | 8 |
use crate::protocol::messages::*; |
12131 | 9 |
|
12857 | 10 |
type Slab<T> = slab::Slab<T>; |
12132 | 11 |
|
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
12 |
|
12131 | 13 |
pub struct HWServer { |
12148 | 14 |
pub clients: Slab<HWClient>, |
15 |
pub rooms: Slab<HWRoom>, |
|
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
16 |
pub lobby_id: RoomId, |
13424 | 17 |
pub output: Vec<(Vec<ClientId>, HWServerMessage)>, |
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
18 |
pub removed_clients: Vec<ClientId>, |
12131 | 19 |
} |
20 |
||
21 |
impl HWServer { |
|
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
22 |
pub fn new(clients_limit: usize, rooms_limit: usize) -> HWServer { |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
23 |
let rooms = Slab::with_capacity(rooms_limit); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
24 |
let clients = Slab::with_capacity(clients_limit); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
25 |
let mut server = HWServer { |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
26 |
clients, rooms, |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
27 |
lobby_id: 0, |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
28 |
output: vec![], |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
29 |
removed_clients: vec![] |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
30 |
}; |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
31 |
server.lobby_id = server.add_room(); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
32 |
server |
12131 | 33 |
} |
34 |
||
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
35 |
pub fn add_client(&mut self) -> ClientId { |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
36 |
let key: ClientId; |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
37 |
{ |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
38 |
let entry = self.clients.vacant_entry(); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
39 |
key = entry.key(); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
40 |
let client = HWClient::new(entry.key()); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
41 |
entry.insert(client); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
42 |
} |
13529 | 43 |
self.send(key, &Destination::ToSelf, HWServerMessage::Connected(utils::PROTOCOL_VERSION)); |
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
44 |
key |
12131 | 45 |
} |
12132 | 46 |
|
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
47 |
pub fn client_lost(&mut self, client_id: ClientId) { |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
48 |
actions::run_action(self, client_id, |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
49 |
actions::Action::ByeClient("Connection reset".to_string())); |
12132 | 50 |
} |
51 |
||
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
52 |
pub fn add_room(&mut self) -> RoomId { |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
53 |
let entry = self.rooms.vacant_entry(); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
54 |
let key = entry.key(); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
55 |
let room = HWRoom::new(entry.key()); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
56 |
entry.insert(room); |
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
57 |
key |
12132 | 58 |
} |
12142
193dfdcb0620
- Use logging facilities instead of plain println!
unc0rr
parents:
12141
diff
changeset
|
59 |
|
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
60 |
pub fn handle_msg(&mut self, client_id: ClientId, msg: HWProtocolMessage) { |
13421 | 61 |
debug!("Handling message {:?} for client {}", msg, client_id); |
13447
c6a3784ff2c1
Check for client's existence before handling messages
alfadur
parents:
13431
diff
changeset
|
62 |
if self.clients.contains(client_id) { |
c6a3784ff2c1
Check for client's existence before handling messages
alfadur
parents:
13431
diff
changeset
|
63 |
handlers::handle(self, client_id, msg); |
c6a3784ff2c1
Check for client's existence before handling messages
alfadur
parents:
13431
diff
changeset
|
64 |
} |
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
65 |
} |
12144 | 66 |
|
13529 | 67 |
fn get_recipients(&self, client_id: ClientId, destination: &Destination) -> Vec<ClientId> { |
68 |
let mut ids = match *destination { |
|
13424 | 69 |
Destination::ToSelf => vec![client_id], |
13431 | 70 |
Destination::ToId(id) => vec![id], |
13424 | 71 |
Destination::ToAll {room_id: Some(id), ..} => |
72 |
self.room_clients(id), |
|
73 |
Destination::ToAll {protocol: Some(proto), ..} => |
|
74 |
self.protocol_clients(proto), |
|
75 |
Destination::ToAll {..} => |
|
13427 | 76 |
self.clients.iter().map(|(id, _)| id).collect::<Vec<_>>() |
13424 | 77 |
}; |
78 |
if let Destination::ToAll {skip_self: true, ..} = destination { |
|
79 |
if let Some(index) = ids.iter().position(|id| *id == client_id) { |
|
80 |
ids.remove(index); |
|
81 |
} |
|
82 |
} |
|
83 |
ids |
|
13421 | 84 |
} |
85 |
||
13529 | 86 |
pub fn send(&mut self, client_id: ClientId, destination: &Destination, message: HWServerMessage) { |
87 |
let ids = self.get_recipients(client_id, &destination); |
|
13424 | 88 |
self.output.push((ids, message)); |
13421 | 89 |
} |
90 |
||
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
91 |
pub fn react(&mut self, client_id: ClientId, actions: Vec<actions::Action>) { |
12144 | 92 |
for action in actions { |
13124
1e39b8749072
separated the server logic from all the async io mess.
alfadur
parents:
12857
diff
changeset
|
93 |
actions::run_action(self, client_id, action); |
12143 | 94 |
} |
95 |
} |
|
13421 | 96 |
|
13526 | 97 |
pub fn lobby(&self) -> &HWRoom { &self.rooms[self.lobby_id] } |
98 |
||
13421 | 99 |
pub fn has_room(&self, name: &str) -> bool { |
100 |
self.rooms.iter().any(|(_, r)| r.name == name) |
|
101 |
} |
|
102 |
||
103 |
pub fn find_room(&self, name: &str) -> Option<&HWRoom> { |
|
104 |
self.rooms.iter().find(|(_, r)| r.name == name).map(|(_, r)| r) |
|
105 |
} |
|
106 |
||
107 |
pub fn find_room_mut(&mut self, name: &str) -> Option<&mut HWRoom> { |
|
108 |
self.rooms.iter_mut().find(|(_, r)| r.name == name).map(|(_, r)| r) |
|
109 |
} |
|
110 |
||
13483 | 111 |
pub fn find_client(&self, nick: &str) -> Option<&HWClient> { |
112 |
self.clients.iter().find(|(_, c)| c.nick == nick).map(|(_, c)| c) |
|
113 |
} |
|
114 |
||
115 |
pub fn find_client_mut(&mut self, nick: &str) -> Option<&mut HWClient> { |
|
116 |
self.clients.iter_mut().find(|(_, c)| c.nick == nick).map(|(_, c)| c) |
|
117 |
} |
|
118 |
||
13421 | 119 |
pub fn select_clients<F>(&self, f: F) -> Vec<ClientId> |
120 |
where F: Fn(&(usize, &HWClient)) -> bool { |
|
121 |
self.clients.iter().filter(f) |
|
122 |
.map(|(_, c)| c.id).collect() |
|
123 |
} |
|
124 |
||
125 |
pub fn room_clients(&self, room_id: RoomId) -> Vec<ClientId> { |
|
126 |
self.select_clients(|(_, c)| c.room_id == Some(room_id)) |
|
127 |
} |
|
128 |
||
13525 | 129 |
pub fn protocol_clients(&self, protocol: u16) -> Vec<ClientId> { |
13421 | 130 |
self.select_clients(|(_, c)| c.protocol_number == protocol) |
131 |
} |
|
132 |
||
133 |
pub fn other_clients_in_room(&self, self_id: ClientId) -> Vec<ClientId> { |
|
134 |
let room_id = self.clients[self_id].room_id; |
|
135 |
self.select_clients(|(id, c)| *id != self_id && c.room_id == room_id ) |
|
136 |
} |
|
137 |
||
138 |
pub fn client_and_room(&mut self, client_id: ClientId) -> (&mut HWClient, Option<&mut HWRoom>) { |
|
139 |
let c = &mut self.clients[client_id]; |
|
140 |
if let Some(room_id) = c.room_id { |
|
141 |
(c, Some(&mut self.rooms[room_id])) |
|
142 |
} else { |
|
143 |
(c, None) |
|
144 |
} |
|
145 |
} |
|
13450
d3c86ade3d4d
Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents:
13447
diff
changeset
|
146 |
|
d3c86ade3d4d
Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents:
13447
diff
changeset
|
147 |
pub fn room(&mut self, client_id: ClientId) -> Option<&mut HWRoom> { |
d3c86ade3d4d
Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents:
13447
diff
changeset
|
148 |
self.client_and_room(client_id).1 |
d3c86ade3d4d
Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents:
13447
diff
changeset
|
149 |
} |
d3c86ade3d4d
Send the rnd reply to the room only.
Marcin Mielniczuk <marmistrz.dev@zoho.eu>
parents:
13447
diff
changeset
|
150 |
} |