chore: fix clippy

This commit is contained in:
2021-05-17 23:01:11 +08:00
parent ae5169e3bd
commit b2ee806448
4 changed files with 295 additions and 274 deletions

View File

@@ -1,244 +1,44 @@
#[macro_use] extern crate lazy_static; #[macro_use] extern crate lazy_static;
#[macro_use] extern crate rust_util; #[macro_use] extern crate rust_util;
mod types;
mod msg;
mod slash_handles;
use std::env; use std::env;
use std::io::Error as IoError; use std::io::Error as IoError;
use std::collections::{HashMap, BTreeMap}; use std::collections::{HashMap, BTreeMap};
use std::net::SocketAddr; use std::net::SocketAddr;
use std::sync::{Arc, Mutex}; use std::sync::Mutex;
use futures_channel::mpsc::{unbounded, UnboundedSender}; use futures_channel::mpsc::unbounded;
use futures_util::{future, pin_mut, stream::TryStreamExt, StreamExt}; use futures_util::{future, pin_mut, stream::TryStreamExt, StreamExt};
use tokio::net::{TcpListener, TcpStream}; use tokio::net::{TcpListener, TcpStream};
use tungstenite::protocol::Message; use tungstenite::protocol::Message;
use rust_util::XResult; use rust_util::XResult;
use serde::{Deserialize, Serialize}; use crate::types::{PeerMap, RoomMap, Tx};
use crate::msg::{RoomMessageDown, RoomMessage, RoomMessageType, RoomMessageDownType};
type Tx = UnboundedSender<Message>; use crate::slash_handles::{HandleContext, HandleTextMessage};
type PeerMap = Arc<Mutex<HashMap<SocketAddr, Tx>>>; use crate::slash_handles::{HandleTextMessageExit, HandleTextMessageStatics};
type RoomMap = Arc<Mutex<BTreeMap<String, BTreeMap<String, SocketAddr>>>>; use crate::slash_handles::{HandleTextMessagePass, HandleTextMessageRoom};
use crate::slash_handles::{HandleTextMessageRooms, HandleTextMessageVersion};
const NAME: &str = env!("CARGO_PKG_NAME"); const NAME: &str = env!("CARGO_PKG_NAME");
const VERSION: &str = env!("CARGO_PKG_VERSION"); const VERSION: &str = env!("CARGO_PKG_VERSION");
#[derive(Clone, Copy, Debug, Serialize, Deserialize)]
enum RoomMessageType {
#[serde(rename = "Enter")]
CreateOrEnter,
Exit,
Destroy,
ListPeers,
Broadcast,
Peer,
}
#[derive(Clone, Copy, Debug, Serialize, Deserialize)]
enum RoomMessageDownType {
PeerEnter,
PeerExit,
PeerList,
PeerMessage,
BroadcastMessage,
ReplyMessage,
}
impl Default for RoomMessageDownType {
fn default() -> Self {
Self::ReplyMessage
}
}
#[derive(Clone, Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
struct RoomMessage {
r#type: RoomMessageType,
room_id: Option<String>,
client_id: Option<String>,
peer_id: Option<String>,
data: Option<String>,
}
#[derive(Clone, Debug, Default, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
struct RoomMessageDown {
r#type: RoomMessageDownType,
peer_id: Option<String>,
reply_code: Option<i32>,
reply_message: Option<String>,
data: Option<String>,
}
impl RoomMessageDown {
fn create_error_reply<S>(error_message: S) -> Self where S: Into<String> {
Self {
r#type: RoomMessageDownType::ReplyMessage,
reply_code: Some(500),
reply_message: Some(error_message.into()),
..Default::default()
}
}
fn create_success_reply<S>(success_message: S) -> Self where S: Into<String> {
Self {
r#type: RoomMessageDownType::ReplyMessage,
reply_code: Some(200),
reply_message: Some(success_message.into()),
..Default::default()
}
}
fn create_peer_enter<S>(peer_id: S) -> Self where S: Into<String> {
let peer_id = peer_id.into();
Self {
r#type: RoomMessageDownType::PeerEnter,
reply_code: Some(200),
reply_message: Some(format!("Peer {} entered", peer_id)),
peer_id: Some(peer_id),
..Default::default()
}
}
fn send(&self, tx: &Tx) {
if let Ok(mm) = serde_json::to_string(self) {
tx.unbounded_send(Message::Text(mm)).ok();
}
}
}
#[derive(Debug, Clone)]
struct HandleContext {
peer_map: PeerMap,
room_map: RoomMap,
admin_pass: Option<String>,
is_admin: bool,
room_id: Option<String>,
client_id: Option<String>,
}
lazy_static! { lazy_static! {
static ref TEXT_MESSAGE_HANDLES: Vec<Box<dyn HandleTextMessage>>= { static ref TEXT_MESSAGE_HANDLES: Vec<Box<dyn HandleTextMessage>>= {
let mut handles:Vec<Box<dyn HandleTextMessage>> = vec![]; let handles:Vec<Box<dyn HandleTextMessage>> = vec![
handles.push(Box::new(HandleTextMessageExit)); Box::new(HandleTextMessageExit),
handles.push(Box::new(HandleTextMessageStatics)); Box::new(HandleTextMessageStatics),
handles.push(Box::new(HandleTextMessagePass)); Box::new(HandleTextMessagePass),
handles.push(Box::new(HandleTextMessageRooms)); Box::new(HandleTextMessageRooms),
handles.push(Box::new(HandleTextMessageRoom)); Box::new(HandleTextMessageRoom),
handles.push(Box::new(HandleTextMessageVersion)); Box::new(HandleTextMessageVersion),
];
handles handles
}; };
} }
trait HandleTextMessage: Sync {
fn is_matches(&self, handle_context: &mut HandleContext, tx: &Tx, addr: SocketAddr, msg: &str) -> bool;
fn handle(&self, handle_context: &mut HandleContext, tx: &Tx, addr: SocketAddr, msg: &str);
}
struct HandleTextMessageExit;
impl HandleTextMessage for HandleTextMessageExit {
fn is_matches(&self, _handle_context: &mut HandleContext, _tx: &Tx, _addr: SocketAddr, msg: &str) -> bool {
msg == "/exit"
}
fn handle(&self, _handle_context: &mut HandleContext, tx: &Tx, _addr: SocketAddr, _msg: &str) {
tx.unbounded_send(Message::Close(None)).ok();
}
}
struct HandleTextMessageStatics;
impl HandleTextMessage for HandleTextMessageStatics {
fn is_matches(&self, _handle_context: &mut HandleContext, _tx: &Tx, _addr: SocketAddr, msg: &str) -> bool {
msg == "/statics"
}
fn handle(&self, handle_context: &mut HandleContext, tx: &Tx, _addr: SocketAddr, _msg: &str) {
tx.unbounded_send(Message::Text(
format!(
"room count: {}\npeer count: {}",
handle_context.room_map.lock().unwrap().len(),
handle_context.peer_map.lock().unwrap().len(),
)
)).ok();
}
}
struct HandleTextMessagePass;
impl HandleTextMessage for HandleTextMessagePass {
fn is_matches(&self, _handle_context: &mut HandleContext, _tx: &Tx, _addr: SocketAddr, msg: &str) -> bool {
msg.starts_with("/pass ")
}
fn handle(&self, handle_context: &mut HandleContext, tx: &Tx, _addr: SocketAddr, msg: &str) {
let pass = &msg[6..];
if let Some(admin_pass) = &handle_context.admin_pass {
if admin_pass == pass {
handle_context.is_admin = true;
tx.unbounded_send(Message::Text("Admin password success".into())).ok();
} else {
tx.unbounded_send(Message::Text("Admin password error".into())).ok();
}
}
}
}
struct HandleTextMessageRooms;
impl HandleTextMessage for HandleTextMessageRooms {
fn is_matches(&self, _handle_context: &mut HandleContext, _tx: &Tx, _addr: SocketAddr, msg: &str) -> bool {
msg == "/rooms"
}
fn handle(&self, handle_context: &mut HandleContext, tx: &Tx, _addr: SocketAddr, _msg: &str) {
if !handle_context.is_admin {
RoomMessageDown::create_error_reply("Not admin").send(tx);
return;
}
let rooms = handle_context.room_map.lock().unwrap().keys().map(
|k| k.clone()
).collect::<Vec<_>>().join("\n");
if rooms.is_empty() {
tx.unbounded_send(Message::Text("rooms: <empty>".into())).ok();
} else {
tx.unbounded_send(Message::Text(format!("rooms:\n{}", rooms))).ok();
}
}
}
struct HandleTextMessageRoom;
impl HandleTextMessage for HandleTextMessageRoom {
fn is_matches(&self, _handle_context: &mut HandleContext, _tx: &Tx, _addr: SocketAddr, msg: &str) -> bool {
msg.starts_with("/room ")
}
fn handle(&self, handle_context: &mut HandleContext, tx: &Tx, _addr: SocketAddr, msg: &str) {
if !handle_context.is_admin {
RoomMessageDown::create_error_reply("Not admin").send(tx);
return;
}
let room_id = &msg[6..];
let room_map = handle_context.room_map.lock().unwrap();
let mut client_ids = vec![];
if let Some(client_map) = room_map.get(room_id) {
for peer_client_id in client_map.keys() {
client_ids.push(peer_client_id.clone());
}
tx.unbounded_send(Message::Text(format!("clients in room {}:\n{}", room_id, client_ids.join("\n")))).ok();
} else {
tx.unbounded_send(Message::Text(format!("room not found: {}", room_id))).ok();
}
}
}
struct HandleTextMessageVersion;
impl HandleTextMessage for HandleTextMessageVersion {
fn is_matches(&self, _handle_context: &mut HandleContext, _tx: &Tx, _addr: SocketAddr, msg: &str) -> bool {
msg == "/version"
}
fn handle(&self, handle_context: &mut HandleContext, tx: &Tx, _addr: SocketAddr, _msg: &str) {
if !handle_context.is_admin {
RoomMessageDown::create_error_reply("Not admin").send(tx);
return;
}
tx.unbounded_send(Message::Text(format!("{} - v{}", NAME, VERSION))).ok();
}
}
fn handle_text_message(handle_context: &mut HandleContext, tx: &Tx, addr: SocketAddr, msg: String) { fn handle_text_message(handle_context: &mut HandleContext, tx: &Tx, addr: SocketAddr, msg: String) {
// process all registered handles // process all registered handles
if msg.starts_with('/') { if msg.starts_with('/') {
@@ -261,8 +61,9 @@ fn handle_text_message(handle_context: &mut HandleContext, tx: &Tx, addr: Socket
RoomMessageType::CreateOrEnter => { RoomMessageType::CreateOrEnter => {
if let (Some(room_id), Some(client_id)) = (&handle_context.room_id, &handle_context.client_id) { if let (Some(room_id), Some(client_id)) = (&handle_context.room_id, &handle_context.client_id) {
warning!("Client is already in room: {:?} - {:?}", room_id, client_id); warning!("Client is already in room: {:?} - {:?}", room_id, client_id);
RoomMessageDown::create_error_reply("Client is already in room").send(&tx); RoomMessageDown::create_error_reply("Client is already in room").send(tx);
} else { return;
}
if let (Some(msg_room_id), Some(msg_client_id)) = (room_message.room_id, room_message.client_id) { if let (Some(msg_room_id), Some(msg_client_id)) = (room_message.room_id, room_message.client_id) {
let mut room_map = handle_context.room_map.lock().unwrap(); let mut room_map = handle_context.room_map.lock().unwrap();
match room_map.get_mut(&msg_room_id) { match room_map.get_mut(&msg_room_id) {
@@ -271,7 +72,7 @@ fn handle_text_message(handle_context: &mut HandleContext, tx: &Tx, addr: Socket
Some(peer_addr) => { Some(peer_addr) => {
if peer_addr == &addr { if peer_addr == &addr {
information!("Duplicate enter to room: {:?} - {:?}", msg_room_id, msg_client_id); information!("Duplicate enter to room: {:?} - {:?}", msg_room_id, msg_client_id);
RoomMessageDown::create_success_reply("Duplicate enter to room").send(&tx); RoomMessageDown::create_success_reply("Duplicate enter to room").send(tx);
} else { } else {
information!( information!(
"Replace client: {:?} - {:?}, from {:?} -> {:?}", "Replace client: {:?} - {:?}, from {:?} -> {:?}",
@@ -282,7 +83,7 @@ fn handle_text_message(handle_context: &mut HandleContext, tx: &Tx, addr: Socket
tx.unbounded_send(Message::Close(None)).ok(); tx.unbounded_send(Message::Close(None)).ok();
} }
client_map.insert(msg_client_id.clone(), addr); client_map.insert(msg_client_id.clone(), addr);
RoomMessageDown::create_success_reply(client_replaced_message).send(&tx); RoomMessageDown::create_success_reply(client_replaced_message).send(tx);
} }
}, },
None => { None => {
@@ -299,7 +100,7 @@ fn handle_text_message(handle_context: &mut HandleContext, tx: &Tx, addr: Socket
} }
} }
} }
RoomMessageDown::create_success_reply(format!("Client entered: {:?}", addr)).send(&tx); RoomMessageDown::create_success_reply(format!("Client entered: {:?}", addr)).send(tx);
}, },
} }
}, },
@@ -308,14 +109,13 @@ fn handle_text_message(handle_context: &mut HandleContext, tx: &Tx, addr: Socket
let mut client_map = BTreeMap::new(); let mut client_map = BTreeMap::new();
client_map.insert(msg_client_id.clone(), addr); client_map.insert(msg_client_id.clone(), addr);
room_map.insert(msg_room_id.clone(), client_map); room_map.insert(msg_room_id.clone(), client_map);
RoomMessageDown::create_success_reply(format!("Client create room: {:?}", addr)).send(&tx); RoomMessageDown::create_success_reply(format!("Client create room: {:?}", addr)).send(tx);
}, },
} }
handle_context.room_id = Some(msg_room_id); handle_context.room_id = Some(msg_room_id);
handle_context.client_id = Some(msg_client_id); handle_context.client_id = Some(msg_client_id);
} else { } else {
RoomMessageDown::create_error_reply("Room id and client id must both assigned").send(&tx); RoomMessageDown::create_error_reply("Room id and client id must both assigned").send(tx);
}
} }
}, },
RoomMessageType::Exit => { RoomMessageType::Exit => {
@@ -355,7 +155,7 @@ fn handle_text_message(handle_context: &mut HandleContext, tx: &Tx, addr: Socket
let client_map = room_map.get(room_id); let client_map = room_map.get(room_id);
let mut client_ids = vec![]; let mut client_ids = vec![];
if let Some(client_map) = client_map { if let Some(client_map) = client_map {
for (client_id, _client_addr) in client_map { for client_id in client_map.keys() {
client_ids.push(client_id.clone()); client_ids.push(client_id.clone());
} }
} }
@@ -425,11 +225,11 @@ fn handle_text_message(handle_context: &mut HandleContext, tx: &Tx, addr: Socket
if let Some(client_map) = client_map { if let Some(client_map) = client_map {
if let Some(peer_client_addr) = client_map.get(peer_id) { if let Some(peer_client_addr) = client_map.get(peer_id) {
if let Some(peer_tx) = handle_context.peer_map.lock().unwrap().get(peer_client_addr) { if let Some(peer_tx) = handle_context.peer_map.lock().unwrap().get(peer_client_addr) {
peer_tx.unbounded_send(Message::Text(mm.clone())).ok(); peer_tx.unbounded_send(Message::Text(mm)).ok();
} }
RoomMessageDown::create_success_reply(format!("Message sent to: {}", peer_id)).send(&tx); RoomMessageDown::create_success_reply(format!("Message sent to: {}", peer_id)).send(tx);
} else { } else {
RoomMessageDown::create_error_reply(format!("Peer not found: {}", peer_id)).send(&tx); RoomMessageDown::create_error_reply(format!("Peer not found: {}", peer_id)).send(tx);
} }
} }
} }
@@ -472,7 +272,7 @@ async fn inner_handle_connection(
Message::Pong(_) => {}, Message::Pong(_) => {},
Message::Binary(_) => { Message::Binary(_) => {
warning!("Ignore binary message from: {:?} - {:?}", handle_context.room_id, handle_context.client_id); warning!("Ignore binary message from: {:?} - {:?}", handle_context.room_id, handle_context.client_id);
RoomMessageDown::create_error_reply("binary not supported").send(&tx); RoomMessageDown::create_error_reply("Binary message not supported").send(&tx);
}, },
Message::Text(msg) => { Message::Text(msg) => {
if !msg.is_empty() { if !msg.is_empty() {
@@ -534,7 +334,7 @@ async fn main() -> Result<(), IoError> {
// Create the event loop and TCP listener we'll accept connections on. // Create the event loop and TCP listener we'll accept connections on.
let try_socket = TcpListener::bind(&listen_addr).await; let try_socket = TcpListener::bind(&listen_addr).await;
let listener = try_socket.expect(&format!("Failed to bind ok: {}", listen_addr)); let listener = try_socket.unwrap_or_else(|_| panic!("Failed to bind ok: {}", listen_addr));
success!("Listening on: {}", listen_addr); success!("Listening on: {}", listen_addr);
// Let's spawn the handling of each connection in a separate task. // Let's spawn the handling of each connection in a separate task.

85
src/msg.rs Normal file
View File

@@ -0,0 +1,85 @@
use tungstenite::Message;
use serde::{Deserialize, Serialize};
use crate::types::Tx;
#[derive(Clone, Copy, Debug, Serialize, Deserialize)]
pub enum RoomMessageType {
#[serde(rename = "Enter")]
CreateOrEnter,
Exit,
Destroy,
ListPeers,
Broadcast,
Peer,
}
#[derive(Clone, Copy, Debug, Serialize, Deserialize)]
pub enum RoomMessageDownType {
PeerEnter,
PeerExit,
PeerList,
PeerMessage,
BroadcastMessage,
ReplyMessage,
}
impl Default for RoomMessageDownType {
fn default() -> Self {
Self::ReplyMessage
}
}
#[derive(Clone, Debug, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct RoomMessage {
pub r#type: RoomMessageType,
pub room_id: Option<String>,
pub client_id: Option<String>,
pub peer_id: Option<String>,
pub data: Option<String>,
}
#[derive(Clone, Debug, Default, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct RoomMessageDown {
pub r#type: RoomMessageDownType,
pub peer_id: Option<String>,
pub reply_code: Option<i32>,
pub reply_message: Option<String>,
pub data: Option<String>,
}
impl RoomMessageDown {
pub fn create_error_reply<S>(error_message: S) -> Self where S: Into<String> {
Self {
r#type: RoomMessageDownType::ReplyMessage,
reply_code: Some(500),
reply_message: Some(error_message.into()),
..Default::default()
}
}
pub fn create_success_reply<S>(success_message: S) -> Self where S: Into<String> {
Self {
r#type: RoomMessageDownType::ReplyMessage,
reply_code: Some(200),
reply_message: Some(success_message.into()),
..Default::default()
}
}
pub fn create_peer_enter<S>(peer_id: S) -> Self where S: Into<String> {
let peer_id = peer_id.into();
Self {
r#type: RoomMessageDownType::PeerEnter,
reply_code: Some(200),
reply_message: Some(format!("Peer {} entered", peer_id)),
peer_id: Some(peer_id),
..Default::default()
}
}
pub fn send(&self, tx: &Tx) {
if let Ok(mm) = serde_json::to_string(self) {
tx.unbounded_send(Message::Text(mm)).ok();
}
}
}

127
src/slash_handles.rs Normal file
View File

@@ -0,0 +1,127 @@
use crate::types::{PeerMap, RoomMap, Tx};
use std::net::SocketAddr;
use tungstenite::Message;
use crate::msg::RoomMessageDown;
use crate::{NAME, VERSION};
#[derive(Debug, Clone)]
pub struct HandleContext {
pub peer_map: PeerMap,
pub room_map: RoomMap,
pub admin_pass: Option<String>,
pub is_admin: bool,
pub room_id: Option<String>,
pub client_id: Option<String>,
}
pub trait HandleTextMessage: Sync {
fn is_matches(&self, handle_context: &mut HandleContext, tx: &Tx, addr: SocketAddr, msg: &str) -> bool;
fn handle(&self, handle_context: &mut HandleContext, tx: &Tx, addr: SocketAddr, msg: &str);
}
pub struct HandleTextMessageExit;
impl HandleTextMessage for HandleTextMessageExit {
fn is_matches(&self, _handle_context: &mut HandleContext, _tx: &Tx, _addr: SocketAddr, msg: &str) -> bool {
msg == "/exit"
}
fn handle(&self, _handle_context: &mut HandleContext, tx: &Tx, _addr: SocketAddr, _msg: &str) {
tx.unbounded_send(Message::Close(None)).ok();
}
}
pub struct HandleTextMessageStatics;
impl HandleTextMessage for HandleTextMessageStatics {
fn is_matches(&self, _handle_context: &mut HandleContext, _tx: &Tx, _addr: SocketAddr, msg: &str) -> bool {
msg == "/statics"
}
fn handle(&self, handle_context: &mut HandleContext, tx: &Tx, _addr: SocketAddr, _msg: &str) {
tx.unbounded_send(Message::Text(
format!(
"room count: {}\npeer count: {}",
handle_context.room_map.lock().unwrap().len(),
handle_context.peer_map.lock().unwrap().len(),
)
)).ok();
}
}
pub struct HandleTextMessagePass;
impl HandleTextMessage for HandleTextMessagePass {
fn is_matches(&self, _handle_context: &mut HandleContext, _tx: &Tx, _addr: SocketAddr, msg: &str) -> bool {
msg.starts_with("/pass ")
}
fn handle(&self, handle_context: &mut HandleContext, tx: &Tx, _addr: SocketAddr, msg: &str) {
let pass = &msg[6..];
if let Some(admin_pass) = &handle_context.admin_pass {
if admin_pass == pass {
handle_context.is_admin = true;
tx.unbounded_send(Message::Text("Admin password success".into())).ok();
} else {
tx.unbounded_send(Message::Text("Admin password error".into())).ok();
}
}
}
}
pub struct HandleTextMessageRooms;
impl HandleTextMessage for HandleTextMessageRooms {
fn is_matches(&self, _handle_context: &mut HandleContext, _tx: &Tx, _addr: SocketAddr, msg: &str) -> bool {
msg == "/rooms"
}
fn handle(&self, handle_context: &mut HandleContext, tx: &Tx, _addr: SocketAddr, _msg: &str) {
if !handle_context.is_admin {
RoomMessageDown::create_error_reply("Not admin").send(tx);
return;
}
let rooms = handle_context.room_map.lock().unwrap().keys().cloned().collect::<Vec<_>>().join("\n");
if rooms.is_empty() {
tx.unbounded_send(Message::Text("rooms: <empty>".into())).ok();
} else {
tx.unbounded_send(Message::Text(format!("rooms:\n{}", rooms))).ok();
}
}
}
pub struct HandleTextMessageRoom;
impl HandleTextMessage for HandleTextMessageRoom {
fn is_matches(&self, _handle_context: &mut HandleContext, _tx: &Tx, _addr: SocketAddr, msg: &str) -> bool {
msg.starts_with("/room ")
}
fn handle(&self, handle_context: &mut HandleContext, tx: &Tx, _addr: SocketAddr, msg: &str) {
if !handle_context.is_admin {
RoomMessageDown::create_error_reply("Not admin").send(tx);
return;
}
let room_id = &msg[6..];
let room_map = handle_context.room_map.lock().unwrap();
let mut client_ids = vec![];
if let Some(client_map) = room_map.get(room_id) {
for peer_client_id in client_map.keys() {
client_ids.push(peer_client_id.clone());
}
tx.unbounded_send(Message::Text(format!("clients in room {}:\n{}", room_id, client_ids.join("\n")))).ok();
} else {
tx.unbounded_send(Message::Text(format!("room not found: {}", room_id))).ok();
}
}
}
pub struct HandleTextMessageVersion;
impl HandleTextMessage for HandleTextMessageVersion {
fn is_matches(&self, _handle_context: &mut HandleContext, _tx: &Tx, _addr: SocketAddr, msg: &str) -> bool {
msg == "/version"
}
fn handle(&self, handle_context: &mut HandleContext, tx: &Tx, _addr: SocketAddr, _msg: &str) {
if !handle_context.is_admin {
RoomMessageDown::create_error_reply("Not admin").send(tx);
return;
}
tx.unbounded_send(Message::Text(format!("{} - v{}", NAME, VERSION))).ok();
}
}

9
src/types.rs Normal file
View File

@@ -0,0 +1,9 @@
use futures_channel::mpsc::UnboundedSender;
use tungstenite::Message;
use std::net::SocketAddr;
use std::collections::{HashMap, BTreeMap};
use std::sync::{Mutex, Arc};
pub type Tx = UnboundedSender<Message>;
pub type PeerMap = Arc<Mutex<HashMap<SocketAddr, Tx>>>;
pub type RoomMap = Arc<Mutex<BTreeMap<String, BTreeMap<String, SocketAddr>>>>;