1
0
mirror of https://github.com/actix/examples synced 2024-11-27 16:02:57 +01:00

Merge pull request #862 from c-git/minor-changes

refactor: replace raw types with aliases from main
This commit is contained in:
Rob Ede 2024-07-21 19:39:45 +00:00 committed by GitHub
commit 3cd5492ef1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -87,7 +87,7 @@ impl ChatServer {
/// Send message to users in a room.
///
/// `skip` is used to prevent messages triggered by a connection also being received by it.
async fn send_system_message(&self, room: &str, skip: ConnId, msg: impl Into<String>) {
async fn send_system_message(&self, room: &str, skip: ConnId, msg: impl Into<Msg>) {
if let Some(sessions) = self.rooms.get(room) {
let msg = msg.into();
@ -106,7 +106,7 @@ impl ChatServer {
///
/// `conn` is used to find current room and prevent messages sent by a connection also being
/// received by it.
async fn send_message(&self, conn: ConnId, msg: impl Into<String>) {
async fn send_message(&self, conn: ConnId, msg: impl Into<Msg>) {
if let Some(room) = self
.rooms
.iter()
@ -124,7 +124,7 @@ impl ChatServer {
self.send_system_message("main", 0, "Someone joined").await;
// register session with random connection ID
let id = thread_rng().gen::<usize>();
let id = thread_rng().gen::<ConnId>();
self.sessions.insert(id, tx);
// auto join session to main room
@ -142,7 +142,7 @@ impl ChatServer {
async fn disconnect(&mut self, conn_id: ConnId) {
println!("Someone disconnected");
let mut rooms: Vec<String> = Vec::new();
let mut rooms: Vec<RoomId> = Vec::new();
// remove sender
if self.sessions.remove(&conn_id).is_some() {
@ -162,12 +162,12 @@ impl ChatServer {
}
/// Returns list of created room names.
fn list_rooms(&mut self) -> Vec<String> {
fn list_rooms(&mut self) -> Vec<RoomId> {
self.rooms.keys().cloned().collect()
}
/// Join room, send disconnect message to old room send join message to new room.
async fn join_room(&mut self, conn_id: ConnId, room: String) {
async fn join_room(&mut self, conn_id: ConnId, room: RoomId) {
let mut rooms = Vec::new();
// remove session from all rooms
@ -230,7 +230,7 @@ pub struct ChatServerHandle {
impl ChatServerHandle {
/// Register client message sender and obtain connection ID.
pub async fn connect(&self, conn_tx: mpsc::UnboundedSender<String>) -> ConnId {
pub async fn connect(&self, conn_tx: mpsc::UnboundedSender<Msg>) -> ConnId {
let (res_tx, res_rx) = oneshot::channel();
// unwrap: chat server should not have been dropped
@ -243,7 +243,7 @@ impl ChatServerHandle {
}
/// List all created rooms.
pub async fn list_rooms(&self) -> Vec<String> {
pub async fn list_rooms(&self) -> Vec<RoomId> {
let (res_tx, res_rx) = oneshot::channel();
// unwrap: chat server should not have been dropped
@ -254,7 +254,7 @@ impl ChatServerHandle {
}
/// Join `room`, creating it if it does not exist.
pub async fn join_room(&self, conn: ConnId, room: impl Into<String>) {
pub async fn join_room(&self, conn: ConnId, room: impl Into<RoomId>) {
let (res_tx, res_rx) = oneshot::channel();
// unwrap: chat server should not have been dropped
@ -271,7 +271,7 @@ impl ChatServerHandle {
}
/// Broadcast message to current room.
pub async fn send_message(&self, conn: ConnId, msg: impl Into<String>) {
pub async fn send_message(&self, conn: ConnId, msg: impl Into<Msg>) {
let (res_tx, res_rx) = oneshot::channel();
// unwrap: chat server should not have been dropped