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