35 lines
1.4 KiB
Rust
35 lines
1.4 KiB
Rust
use simple_websockets::{Event, Responder};
|
|
use std::collections::HashMap;
|
|
|
|
fn main() {
|
|
let event_hub = simple_websockets::launch(8080).expect("failed to listen on port 8080");
|
|
let mut clients: HashMap<u64, Responder> = HashMap::new();
|
|
println!("Listening for websocket connection...");
|
|
loop {
|
|
match event_hub.poll_event() {
|
|
Event::Connect(client_id, responder) => {
|
|
println!("A client connected with id #{}", client_id);
|
|
// add their Responder to our `clients` map:
|
|
clients.insert(client_id, responder);
|
|
}
|
|
Event::Disconnect(client_id) => {
|
|
println!("Client #{} disconnected.", client_id);
|
|
// remove the disconnected client from the clients map:
|
|
clients.remove(&client_id);
|
|
}
|
|
Event::Message(client_id, message) => {
|
|
println!("\nReceived a message from client #{}", client_id);
|
|
|
|
let all_clients = clients.keys().collect::<Vec<_>>();
|
|
for client in all_clients {
|
|
if *client != client_id {
|
|
println!("Sending message to client #{}", client);
|
|
let responder = clients.get(client).unwrap();
|
|
responder.send(message.clone());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|